From 0ed8a66fb8130ec2cef4fa47b1b573eaf1985399 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <43494761+iLLiCiTiT@users.noreply.github.com>
Date: Wed, 6 Dec 2023 15:05:10 +0100
Subject: [PATCH] AYON Change of login should work

Co-authored-by: Libor Batek <112623825+LiborBatek@users.noreply.github.com>
---
 openpype/tools/tray/pype_tray.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/openpype/tools/tray/pype_tray.py b/openpype/tools/tray/pype_tray.py
index a5876ca721..db391b469a 100644
--- a/openpype/tools/tray/pype_tray.py
+++ b/openpype/tools/tray/pype_tray.py
@@ -632,6 +632,14 @@ class TrayManager:
             self.exit()
 
         elif result.restart or result.token_changed:
+            # Remove environment variables from current connection
+            # - keep develop, staging, headless values
+            for key in {
+                "AYON_SERVER_URL",
+                "AYON_API_KEY",
+                "AYON_BUNDLE_NAME",
+            }:
+                os.environ.pop(key, None)
             self.restart()
 
     def _on_restart_action(self):