Merge pull request #292 from superannotateai/456

import mixp

Author: dshabin

src/superannotate/__init__.py

@@ -312,16 +312,19 @@
 local_version = parse(__version__)
 if local_version.is_prerelease:
     logging.info(constances.PACKAGE_VERSION_INFO_MESSAGE.format(__version__))
-req = requests.get('https://pypi.python.org/pypi/superannotate/json')
+req = requests.get("https://pypi.python.org/pypi/superannotate/json")
 if req.ok:
-    releases = req.json().get('releases', [])
-    pip_version = parse('0')
+    releases = req.json().get("releases", [])
+    pip_version = parse("0")
     for release in releases:
         ver = parse(release)
         if not ver.is_prerelease or local_version.is_prerelease:
             pip_version = max(pip_version, ver)
     if pip_version.major > local_version.major:
-        logging.warning(constances.PACKAGE_VERSION_MAJOR_UPGRADE.format(local_version, pip_version))
+        logging.warning(
+            constances.PACKAGE_VERSION_MAJOR_UPGRADE.format(local_version, pip_version)
+        )
     elif pip_version > local_version:
-        logging.warning(constances.PACKAGE_VERSION_UPGRADE.format(local_version, pip_version)
+        logging.warning(
+            constances.PACKAGE_VERSION_UPGRADE.format(local_version, pip_version)
         )

src/superannotate/lib/app/analytics/class_analytics.py

@@ -3,8 +3,8 @@
 
 import pandas as pd
 import plotly.express as px
+from lib.app.mixp.decorators import Trackable
 from superannotate.lib.app.exceptions import AppException
-from superannotate.lib.app.mixp.decorators import Trackable
 from superannotate.lib.core import DEPRICATED_DOCUMENT_VIDEO_MESSAGE
 
 from .common import aggregate_annotations_as_df

src/superannotate/lib/app/input_converters/df_converter.py

@@ -2,7 +2,7 @@
 from pathlib import Path
 
 import pandas as pd
-from superannotate.lib.app.mixp.decorators import Trackable
+from lib.app.mixp.decorators import Trackable
 
 
 @Trackable

src/superannotate/lib/app/input_converters/dicom_converter.py

@@ -2,8 +2,8 @@
 
 import numpy as np
 import pydicom
+from lib.app.mixp.decorators import Trackable
 from PIL import Image
-from superannotate.lib.app.mixp.decorators import Trackable
 
 
 @Trackable

src/superannotate/lib/app/interface/sdk_interface.py

@@ -2354,6 +2354,7 @@ def attach_image_urls_to_project(
         return uploaded, failed_images, duplications
     raise AppException(use_case.response.errors)
 
+
 @Trackable
 @validate_arguments
 def attach_video_urls_to_project(

src/superannotate/lib/app/mixp/decorators.py

@@ -8,6 +8,7 @@
 from .config import TOKEN
 from .utils import parsers
 
+controller = Controller.get_instance()
 mp = Mixpanel(TOKEN)
 
 
@@ -25,18 +26,32 @@ def get_default(team_name, user_id, project_name=None):
 
 class Trackable:
     TEAM_DATA = None
+    INITIAL_EVENT = {"event_name": "SDK init", "properties": {}}
+    INITIAL_LOGGED = False
 
-    def __init__(self, function):
+    def __init__(self, function, initial=False):
         self.function = function
         self._success = False
+        self._initial = initial
+        if initial:
+            self.track()
         functools.update_wrapper(self, function)
 
+    @property
+    def team(self):
+        return controller.get_team()
+
     def track(self, *args, **kwargs):
         try:
-            data = getattr(parsers, self.function.__name__)(*args, **kwargs)
+            if self._initial:
+                data = self.INITIAL_EVENT
+                Trackable.INITIAL_LOGGED = True
+                self._success = True
+            else:
+                data = getattr(parsers, self.function.__name__)(*args, **kwargs)
             event_name = data["event_name"]
             properties = data["properties"]
-            team_data = self.__class__.TEAM_DATA.data
+            team_data = self.team.data
             user_id = team_data.creator_id
             team_name = team_data.name
             properties["Success"] = self._success
@@ -55,16 +70,20 @@ def track(self, *args, **kwargs):
 
     def __call__(self, *args, **kwargs):
         try:
-            self.__class__.TEAM_DATA = Controller.get_instance().get_team()
-            ret = self.function(*args, **kwargs)
+            self.__class__.TEAM_DATA = controller.get_team()
+            result = self.function(*args, **kwargs)
             self._success = True
         except Exception as e:
             self._success = False
             raise e
         else:
-            return ret
+            return result
         finally:
             try:
                 self.track(*args, **kwargs)
             except Exception:
                 pass
+
+
+if __name__ == "lib.app.mixp.decorators" and not Trackable.INITIAL_LOGGED:
+    Trackable(None, initial=True)

src/superannotate/lib/app/mixp/utils/parsers.py

@@ -1519,6 +1519,6 @@ def attach_document_urls_to_project(*args, **kwargs):
         },
     }
 
+
 def delete_annotations(*args, **kwargs):
     return {"event_name": "delete_annotations", "properties": {}}
-

src/superannotate/lib/core/__init__.py

@@ -102,15 +102,21 @@
 )
 MOVE_PROJECT_LIMIT_ERROR_MESSAGE = "The number of items you want to move exceeds the limit of 500 000 items per project."
 
-PACKAGE_VERSION_INFO_MESSAGE = "Development version {} of SuperAnnotate SDK is being used."
+PACKAGE_VERSION_INFO_MESSAGE = (
+    "Development version {} of SuperAnnotate SDK is being used."
+)
 
-PACKAGE_VERSION_MAJOR_UPGRADE = "There is a major upgrade of SuperAnnotate Python SDK available on PyPI. " \
-                          "We recommend upgrading. Run 'pip install --upgrade superannotate' to " \
-                          "upgrade from your version {} to {}."
+PACKAGE_VERSION_MAJOR_UPGRADE = (
+    "There is a major upgrade of SuperAnnotate Python SDK available on PyPI. "
+    "We recommend upgrading. Run 'pip install --upgrade superannotate' to "
+    "upgrade from your version {} to {}."
+)
 
-PACKAGE_VERSION_UPGRADE = "There is a newer version of SuperAnnotate Python SDK available on PyPI." \
-                      " Run 'pip install --upgrade superannotate' to" \
-                      " upgrade from your version {} to {}"
+PACKAGE_VERSION_UPGRADE = (
+    "There is a newer version of SuperAnnotate Python SDK available on PyPI."
+    " Run 'pip install --upgrade superannotate' to"
+    " upgrade from your version {} to {}"
+)
 __alL__ = (
     ProjectType,
     UserRole,

src/superannotate/lib/core/usecases/annotations.py

@@ -356,7 +356,7 @@ def set_annotation_json(self):
                             constances.PIXEL_ANNOTATION_POSTFIX,
                             constances.ANNOTATION_MASK_POSTFIX,
                         ),
-                        "rb"
+                        "rb",
                     )
 
     def _is_valid_json(self, json_data: dict):