From ab6a90a2d253f4a8ad955606f12b616e32630b99 Mon Sep 17 00:00:00 2001 From: Dania Valladares Date: Mon, 31 Mar 2025 12:03:22 -0400 Subject: [PATCH 1/5] dev-env --- src/codegate/updates/client.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/codegate/updates/client.py b/src/codegate/updates/client.py index f899a43b..dca8a9c3 100644 --- a/src/codegate/updates/client.py +++ b/src/codegate/updates/client.py @@ -2,6 +2,7 @@ import requests import structlog +import os logger = structlog.get_logger("codegate") @@ -25,11 +26,15 @@ def get_latest_version(self, origin: Origin) -> str: """ Retrieves the latest version of CodeGate from updates.codegate.ai """ + user_agent = f"codegate/{self.__current_version} {origin.value}" + if os.environ.get("CODEGATE_DEV_ENV"): + user_agent += "-dev" + headers = { "X-Instance-ID": self.__instance_id, - "User-Agent": f"codegate/{self.__current_version} {origin.value}", + "User-Agent": user_agent, } - + try: response = requests.get(self.__update_url, headers=headers, timeout=10) # Throw if the request was not successful. From 506bcee86a1eb75ff0b71610085b5a2f1bd73a72 Mon Sep 17 00:00:00 2001 From: Dania Valladares Date: Mon, 31 Mar 2025 12:11:20 -0400 Subject: [PATCH 2/5] formatting --- src/codegate/updates/client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/codegate/updates/client.py b/src/codegate/updates/client.py index dca8a9c3..58af7da1 100644 --- a/src/codegate/updates/client.py +++ b/src/codegate/updates/client.py @@ -26,10 +26,10 @@ def get_latest_version(self, origin: Origin) -> str: """ Retrieves the latest version of CodeGate from updates.codegate.ai """ + user_agent = f"codegate/{self.__current_version} {origin.value}" if os.environ.get("CODEGATE_DEV_ENV"): user_agent += "-dev" - headers = { "X-Instance-ID": self.__instance_id, "User-Agent": user_agent, From 99d87bd487c7bb331edf595ee0435a81933deaeb Mon Sep 17 00:00:00 2001 From: Dania Valladares Date: Mon, 31 Mar 2025 12:19:18 -0400 Subject: [PATCH 3/5] black formatting --- src/codegate/updates/client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/codegate/updates/client.py b/src/codegate/updates/client.py index 58af7da1..29ecd2d4 100644 --- a/src/codegate/updates/client.py +++ b/src/codegate/updates/client.py @@ -34,7 +34,7 @@ def get_latest_version(self, origin: Origin) -> str: "X-Instance-ID": self.__instance_id, "User-Agent": user_agent, } - + try: response = requests.get(self.__update_url, headers=headers, timeout=10) # Throw if the request was not successful. From 0600a7e1159f60d7fa46729b26e32c3440ef6c9a Mon Sep 17 00:00:00 2001 From: Teodor Yanev Date: Mon, 31 Mar 2025 19:23:19 +0300 Subject: [PATCH 4/5] read the development environment flag once at module level --- src/codegate/updates/client.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/codegate/updates/client.py b/src/codegate/updates/client.py index 29ecd2d4..334c6220 100644 --- a/src/codegate/updates/client.py +++ b/src/codegate/updates/client.py @@ -9,6 +9,7 @@ __update_client_singleton = None +is_dev_env = bool(os.environ.get("CODEGATE_DEV_ENV")) # Enum representing whether the request is coming from the front-end or the back-end. class Origin(Enum): @@ -28,7 +29,7 @@ def get_latest_version(self, origin: Origin) -> str: """ user_agent = f"codegate/{self.__current_version} {origin.value}" - if os.environ.get("CODEGATE_DEV_ENV"): + if is_dev_env: user_agent += "-dev" headers = { "X-Instance-ID": self.__instance_id, From e2ae01c4a99c4ff13fdfc9d11e5d40dd9d5cd33a Mon Sep 17 00:00:00 2001 From: Teodor Yanev Date: Tue, 1 Apr 2025 12:02:37 +0300 Subject: [PATCH 5/5] lint with black --- src/codegate/updates/client.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/codegate/updates/client.py b/src/codegate/updates/client.py index 334c6220..77dd6ac0 100644 --- a/src/codegate/updates/client.py +++ b/src/codegate/updates/client.py @@ -11,6 +11,7 @@ is_dev_env = bool(os.environ.get("CODEGATE_DEV_ENV")) + # Enum representing whether the request is coming from the front-end or the back-end. class Origin(Enum): FrontEnd = "FE"