fix: sync module-level user_id after init_tracking generates UUID

Signed-off-by: Alexander Piskun <bigcat88@icloud.com>

Author: bigcat88

comfy_cli/tracking.py

@@ -97,6 +97,7 @@ def init_tracking(enable_tracking: bool):
     """
     Initialize the tracking system by setting the user identifier and tracking enabled status.
     """
+    global user_id
     logging.debug(f"Initializing tracking with enable_tracking: {enable_tracking}")
     config_manager.set(constants.CONFIG_KEY_ENABLE_TRACKING, str(enable_tracking))
     if not enable_tracking:
@@ -108,6 +109,7 @@ def init_tracking(enable_tracking: bool):
         curr_user_id = str(uuid.uuid4())
         config_manager.set(constants.CONFIG_KEY_USER_ID, curr_user_id)
         logging.debug(f'Setting user identifier for tracking user_id: {curr_user_id}."')
+    user_id = curr_user_id
 
     # Note: only called once when the user interacts with the CLI for the
     #  first time iff the permission is granted.

tests/comfy_cli/test_tracking.py

@@ -87,7 +87,11 @@ def test_disable_persists_as_parseable_bool(self, tracking_module):
     def test_enable_generates_user_id(self, tracking_module):
         assert tracking_module.config_manager.get(constants.CONFIG_KEY_USER_ID) is None
         tracking_module.init_tracking(True)
-        assert tracking_module.config_manager.get(constants.CONFIG_KEY_USER_ID) is not None
+        generated_user_id = tracking_module.config_manager.get(constants.CONFIG_KEY_USER_ID)
+        assert generated_user_id is not None
+        assert tracking_module.user_id == generated_user_id
+        _, kwargs = tracking_module.mp.track.call_args
+        assert kwargs["distinct_id"] == generated_user_id
 
     def test_disable_does_not_generate_user_id(self, tracking_module):
         tracking_module.init_tracking(False)