From 62db7c56e3837d1b61b672a24b72fb2ac07e0c40 Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Fri, 28 Jul 2023 13:00:58 -0700
Subject: refactor: :loud_sound: potential telemetry fixes
---
continuedev/src/continuedev/core/sdk.py | 2 +-
continuedev/src/continuedev/libs/util/telemetry.py | 15 ++++++++-------
2 files changed, 9 insertions(+), 8 deletions(-)
(limited to 'continuedev')
diff --git a/continuedev/src/continuedev/core/sdk.py b/continuedev/src/continuedev/core/sdk.py
index 992bc1cf..264a6852 100644
--- a/continuedev/src/continuedev/core/sdk.py
+++ b/continuedev/src/continuedev/core/sdk.py
@@ -269,7 +269,7 @@ class ContinueSDK(AbstractContinueSDK):
# When the config is loaded, setup posthog logger
posthog_logger.setup(
- self.ide.unique_id, config.config.allow_anonymous_telemetry or True)
+ self.ide.unique_id, config.config.allow_anonymous_telemetry)
return config.config
except Exception as e:
diff --git a/continuedev/src/continuedev/libs/util/telemetry.py b/continuedev/src/continuedev/libs/util/telemetry.py
index a967828e..e1efb668 100644
--- a/continuedev/src/continuedev/libs/util/telemetry.py
+++ b/continuedev/src/continuedev/libs/util/telemetry.py
@@ -10,20 +10,21 @@ POSTHOG_API_KEY = 'phc_JS6XFROuNbhJtVCEdTSYk6gl5ArRrTNMpCcguAXlSPs'
class PostHogLogger:
+ unique_id: str = "NO_UNIQUE_ID"
+ allow_anonymous_telemetry: bool = True
+
def __init__(self, api_key: str):
self.api_key = api_key
- self.unique_id = None
- self.allow_anonymous_telemetry = True
-
- def setup(self, unique_id: str, allow_anonymous_telemetry: bool):
- self.unique_id = unique_id
- self.allow_anonymous_telemetry = allow_anonymous_telemetry
# The personal API key is necessary only if you want to use local evaluation of feature flags.
self.posthog = Posthog(self.api_key, host='https://app.posthog.com')
+ def setup(self, unique_id: str, allow_anonymous_telemetry: bool):
+ self.unique_id = unique_id or "NO_UNIQUE_ID"
+ self.allow_anonymous_telemetry = allow_anonymous_telemetry or True
+
def capture_event(self, event_name: str, event_properties: Any):
- if not self.allow_anonymous_telemetry or self.unique_id is None:
+ if not self.allow_anonymous_telemetry:
return
if in_codespaces:
--
cgit v1.2.3-70-g09d2