Merge pull request #364 from superannotateai/logger

Logger

Author: dshabin

src/superannotate/__init__.py

@@ -1,7 +1,6 @@
 import logging.config
 import os
 import sys
-from os.path import expanduser
 
 import requests
 import superannotate.lib.core as constances
@@ -107,6 +106,7 @@
     upload_videos_from_folder_to_project,
 )
 from superannotate.lib.app.interface.sdk_interface import validate_annotations
+from superannotate.logger import get_default_logger
 from superannotate.version import __version__
 
 __all__ = [
@@ -208,49 +208,13 @@
 WORKING_DIR = os.path.split(os.path.realpath(__file__))[0]
 sys.path.append(WORKING_DIR)
 logging.getLogger("botocore").setLevel(logging.CRITICAL)
-
-logging.config.dictConfig(
-    {
-        "version": 1,
-        "disable_existing_loggers": False,
-        "handlers": {
-            "console": {
-                "class": "logging.StreamHandler",
-                "level": "INFO",
-                "formatter": "consoleFormatter",
-                "stream": "ext://sys.stdout",
-            },
-            "fileHandler": {
-                "class": "logging.handlers.RotatingFileHandler",
-                "level": "DEBUG",
-                "formatter": "fileFormatter",
-                "filename": expanduser(constances.LOG_FILE_LOCATION),
-                "mode": "a",
-                "maxBytes": 5 * 1024 * 1024,
-                "backupCount": 5,
-            },
-        },
-        "formatters": {
-            "consoleFormatter": {
-                "format": "SA-PYTHON-SDK - %(levelname)s - %(message)s",
-            },
-            "fileFormatter": {
-                "format": "SA-PYTHON-SDK - %(levelname)s - %(asctime)s - %(message)s"
-            },
-        },
-        "root": {
-            # "level": "INFO",
-            "level": "DEBUG",
-            "handlers": ["console", "fileHandler"],
-        },
-    }
-)
+logger = get_default_logger()
 
 
 def log_version_info():
     local_version = parse(__version__)
     if local_version.is_prerelease:
-        logging.info(constances.PACKAGE_VERSION_INFO_MESSAGE.format(__version__))
+        logger.info(constances.PACKAGE_VERSION_INFO_MESSAGE.format(__version__))
     req = requests.get("https://pypi.python.org/pypi/superannotate/json")
     if req.ok:
         releases = req.json().get("releases", [])
@@ -260,13 +224,13 @@ def log_version_info():
             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(
+            logger.warning(
                 constances.PACKAGE_VERSION_MAJOR_UPGRADE.format(
                     local_version, pip_version
                 )
             )
         elif pip_version > local_version:
-            logging.warning(
+            logger.warning(
                 constances.PACKAGE_VERSION_UPGRADE.format(local_version, pip_version)
             )
 

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

@@ -1,20 +1,20 @@
 import copy
 import json
-import logging
+from dataclasses import dataclass
 from pathlib import Path
 from typing import List
 from typing import Optional
 from typing import Union
 
 import lib.core as constances
 import pandas as pd
-from dataclasses import dataclass
 from lib.app.exceptions import AppException
 from lib.core import ATTACHED_VIDEO_ANNOTATION_POSTFIX
 from lib.core import PIXEL_ANNOTATION_POSTFIX
 from lib.core import VECTOR_ANNOTATION_POSTFIX
+from superannotate.logger import get_default_logger
 
-logger = logging.getLogger("root")
+logger = get_default_logger()
 
 
 @dataclass

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

@@ -1,15 +1,15 @@
-import logging
 from pathlib import Path
 
 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.core import DEPRICATED_DOCUMENT_VIDEO_MESSAGE
+from superannotate.logger import get_default_logger
 
 from .common import aggregate_image_annotations_as_df
 
-logger = logging.getLogger()
+logger = get_default_logger()
 
 
 @Trackable

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

@@ -1,13 +1,14 @@
 import json
-import logging
 from pathlib import Path
 
 import pandas as pd
 import plotly.express as px
 from lib.app.exceptions import AppException
 from lib.core import DEPRICATED_DOCUMENT_VIDEO_MESSAGE
+from superannotate.logger import get_default_logger
 
-logger = logging.getLogger("root")
+
+logger = get_default_logger()
 
 
 def aggregate_image_annotations_as_df(

src/superannotate/lib/app/bin/superannotate.py

@@ -1,10 +1,9 @@
 #!/usr/bin/env python3
-import logging
-
 import fire
 from lib.app.interface.cli_interface import CLIFacade
+from superannotate.logger import get_default_logger
 
-logger = logging.getLogger()
+logger = get_default_logger()
 
 
 def main():

src/superannotate/lib/app/common.py

@@ -1,14 +1,14 @@
 import json
-import logging
 import os
 import sys
 import time
 
 import numpy as np
 from PIL import Image
+from superannotate.logger import get_default_logger
 from tqdm import tqdm
 
-logger = logging.getLogger()
+logger = get_default_logger()
 
 _PROJECT_TYPES = {"Vector": 1, "Pixel": 2}
 

src/superannotate/lib/app/input_converters/converters/baseStrategy.py

@@ -1,6 +1,6 @@
 """
 """
-import logging
+from superannotate.logger import get_default_logger
 
 from .coco_converters.coco_to_sa_pixel import coco_instance_segmentation_to_sa_pixel
 from .coco_converters.coco_to_sa_pixel import coco_panoptic_segmentation_to_sa_pixel
@@ -38,7 +38,7 @@
 from .vott_converters.vott_to_sa_vector import vott_to_sa
 from .yolo_converters.yolo_to_sa_vector import yolo_object_detection_to_sa_vector
 
-logger = logging.getLogger()
+logger = get_default_logger()
 
 CONVERSION_ALGORITHMS = {
     "from": {

src/superannotate/lib/app/input_converters/converters/coco_converters/coco_converter.py

@@ -1,20 +1,20 @@
 """
 """
 import json
-import logging
 from collections import namedtuple
 from datetime import datetime
 from pathlib import Path
 
 import cv2
 import numpy as np
 from PIL import Image
+from superannotate.logger import get_default_logger
 
 from ....common import id2rgb
 from ....common import write_to_json
 from ..baseStrategy import baseStrategy
 
-logger = logging.getLogger()
+logger = get_default_logger()
 
 
 class CocoBaseStrategy(baseStrategy):

src/superannotate/lib/app/input_converters/converters/coco_converters/coco_strategies.py

@@ -1,17 +1,17 @@
 """
 """
-import logging
 import threading
 from pathlib import Path
 
 from PIL import Image
+from superannotate.logger import get_default_logger
 
 from ....common import id2rgb
 from ....common import tqdm_converter
 from ....common import write_to_json
 from .coco_converter import CocoBaseStrategy
 
-logger = logging.getLogger()
+logger = get_default_logger()
 
 
 class CocoPanopticConverterStrategy(CocoBaseStrategy):

src/superannotate/lib/app/input_converters/converters/coco_converters/coco_to_sa_pixel.py

@@ -2,12 +2,12 @@
 COCO to SA conversion method
 """
 import json
-import logging
 import threading
 from pathlib import Path
 
 import cv2
 import numpy as np
+from superannotate.logger import get_default_logger
 
 from ....common import blue_color_generator
 from ....common import hex_to_rgb
@@ -19,7 +19,7 @@
 from .coco_api import _maskfrRLE
 from .coco_api import decode
 
-logger = logging.getLogger()
+logger = get_default_logger()
 
 
 def annot_to_bitmask(annot):