treewide: optimize imports

benchmark.py

@@ -1,11 +1,11 @@
-import os
-from statistics import mean
+import datetime
 import multiprocessing as mp
+from statistics import mean
+
 import numpy as np
-import datetime
+
 from frigate.config import DetectorTypeEnum
 from frigate.object_detection import (
-    LocalObjectDetector,
     ObjectDetectProcess,
     RemoteObjectDetector,
     load_labels,

docker/rootfs/usr/local/go2rtc/create_config.py

@@ -3,11 +3,14 @@
 import json
 import os
 import sys
+
 import yaml
 
 sys.path.insert(0, "/opt/frigate")
-from frigate.const import BIRDSEYE_PIPE, BTBN_PATH
-from frigate.ffmpeg_presets import parse_preset_hardware_acceleration_encode
+from frigate.const import BIRDSEYE_PIPE, BTBN_PATH  # noqa: E402
+from frigate.ffmpeg_presets import (  # noqa: E402
+    parse_preset_hardware_acceleration_encode,
+)
 
 sys.path.remove("/opt/frigate")
 

frigate/__main__.py

@@ -5,7 +5,6 @@
 
 from frigate.app import FrigateApp
 
-
 faulthandler.enable()
 
 threading.current_thread().name = "frigate"

frigate/app.py

@@ -1,16 +1,16 @@
 import logging
 import multiprocessing as mp
-from multiprocessing.queues import Queue
-from multiprocessing.synchronize import Event as MpEvent
 import os
 import shutil
 import signal
 import sys
-from typing import Optional
+import traceback
+from multiprocessing.queues import Queue
+from multiprocessing.synchronize import Event as MpEvent
 from types import FrameType
-import psutil
+from typing import Optional
 
-import traceback
+import psutil
 from peewee_migrate import Router
 from playhouse.sqlite_ext import SqliteExtDatabase
 from playhouse.sqliteq import SqliteQueueDatabase
@@ -27,13 +27,13 @@
     MODEL_CACHE_DIR,
     RECORD_DIR,
 )
-from frigate.object_detection import ObjectDetectProcess
 from frigate.events.cleanup import EventCleanup
 from frigate.events.external import ExternalEventProcessor
 from frigate.events.maintainer import EventProcessor
 from frigate.http import create_app
 from frigate.log import log_process, root_configurer
 from frigate.models import Event, Recordings, Timeline
+from frigate.object_detection import ObjectDetectProcess
 from frigate.object_processing import TrackedObjectProcessor
 from frigate.output import output_frames
 from frigate.plus import PlusApi
@@ -42,10 +42,10 @@
 from frigate.stats import StatsEmitter, stats_init
 from frigate.storage import StorageMaintainer
 from frigate.timeline import TimelineProcessor
+from frigate.types import CameraMetricsTypes, RecordMetricsTypes
 from frigate.version import VERSION
 from frigate.video import capture_camera, track_camera
 from frigate.watchdog import FrigateWatchdog
-from frigate.types import CameraMetricsTypes, RecordMetricsTypes
 
 logger = logging.getLogger(__name__)
 

frigate/comms/dispatcher.py

@@ -1,17 +1,14 @@
 """Handle communication between Frigate and other applications."""
 
 import logging
-
-from typing import Any, Callable
-
 from abc import ABC, abstractmethod
+from typing import Any, Callable
 
 from frigate.config import FrigateConfig
-from frigate.ptz import OnvifController, OnvifCommandEnum
+from frigate.ptz import OnvifCommandEnum, OnvifController
 from frigate.types import CameraMetricsTypes, RecordMetricsTypes
 from frigate.util import restart_frigate
 
-
 logger = logging.getLogger(__name__)
 
 

frigate/comms/mqtt.py

@@ -1,14 +1,12 @@
 import logging
 import threading
-
 from typing import Any, Callable
 
 import paho.mqtt.client as mqtt
 
 from frigate.comms.dispatcher import Communicator
 from frigate.config import FrigateConfig
 
-
 logger = logging.getLogger(__name__)
 
 

frigate/comms/ws.py

@@ -3,10 +3,9 @@
 import json
 import logging
 import threading
-
 from typing import Callable
-
 from wsgiref.simple_server import make_server
+
 from ws4py.server.wsgirefserver import (
     WebSocketWSGIHandler,
     WebSocketWSGIRequestHandler,
@@ -18,7 +17,6 @@
 from frigate.comms.dispatcher import Communicator
 from frigate.config import FrigateConfig
 
-
 logger = logging.getLogger(__name__)
 
 

frigate/config.py

@@ -8,37 +8,30 @@
 
 import matplotlib.pyplot as plt
 import numpy as np
-from pydantic import BaseModel, Extra, Field, validator, parse_obj_as
+from pydantic import BaseModel, Extra, Field, parse_obj_as, validator
 from pydantic.fields import PrivateAttr
 
-from frigate.const import (
-    CACHE_DIR,
-    DEFAULT_DB_PATH,
-    REGEX_CAMERA_NAME,
-    YAML_EXT,
-)
+from frigate.const import CACHE_DIR, DEFAULT_DB_PATH, REGEX_CAMERA_NAME, YAML_EXT
+from frigate.detectors import DetectorConfig, ModelConfig
+from frigate.detectors.detector_config import InputTensorEnum  # noqa: F401
+from frigate.detectors.detector_config import PixelFormatEnum  # noqa: F401
 from frigate.detectors.detector_config import BaseDetectorConfig
-from frigate.plus import PlusApi
-from frigate.util import (
-    create_mask,
-    deep_merge,
-    get_ffmpeg_arg_list,
-    escape_special_characters,
-    load_config_with_no_duplicates,
-)
 from frigate.ffmpeg_presets import (
     parse_preset_hardware_acceleration_decode,
     parse_preset_hardware_acceleration_scale,
     parse_preset_input,
     parse_preset_output_record,
     parse_preset_output_rtmp,
 )
-from frigate.detectors import (
-    ModelConfig,
-    DetectorConfig,
+from frigate.plus import PlusApi
+from frigate.util import (
+    create_mask,
+    deep_merge,
+    escape_special_characters,
+    get_ffmpeg_arg_list,
+    load_config_with_no_duplicates,
 )
 
-
 logger = logging.getLogger(__name__)
 
 # TODO: Identify what the default format to display timestamps is

frigate/detectors/__init__.py

@@ -1,7 +1,7 @@
 import logging
 
-from .detector_types import DetectorTypeEnum, api_types, DetectorConfig
-
+from .detector_config import InputTensorEnum, ModelConfig, PixelFormatEnum  # noqa: F401
+from .detector_types import DetectorConfig, DetectorTypeEnum, api_types  # noqa: F401
 
 logger = logging.getLogger(__name__)
 

frigate/detectors/detection_api.py

@@ -1,7 +1,6 @@
 import logging
 from abc import ABC, abstractmethod
 
-
 logger = logging.getLogger(__name__)
 
 

frigate/detectors/detector_config.py

@@ -1,20 +1,18 @@
 import hashlib
 import json
 import logging
-from enum import Enum
 import os
+from enum import Enum
 from typing import Dict, Optional, Tuple
 
-
-import requests
 import matplotlib.pyplot as plt
+import requests
 from pydantic import BaseModel, Extra, Field
 from pydantic.fields import PrivateAttr
-from frigate.plus import PlusApi
 
+from frigate.plus import PlusApi
 from frigate.util import load_labels
 
-
 logger = logging.getLogger(__name__)
 
 

frigate/detectors/detector_types.py

@@ -1,16 +1,16 @@
-import logging
 import importlib
+import logging
 import pkgutil
-from typing import Union
-from typing_extensions import Annotated
 from enum import Enum
+from typing import Union
+
 from pydantic import Field
+from typing_extensions import Annotated
 
 from . import plugins
 from .detection_api import DetectionApi
 from .detector_config import BaseDetectorConfig
 
-
 logger = logging.getLogger(__name__)
 
 

frigate/detectors/plugins/cpu_tfl.py

@@ -1,12 +1,12 @@
 import logging
+
 import numpy as np
+from pydantic import Field
+from typing_extensions import Literal
 
 from frigate.detectors.detection_api import DetectionApi
 from frigate.detectors.detector_config import BaseDetectorConfig
 
-from typing_extensions import Literal
-from pydantic import Field
-
 try:
     from tflite_runtime.interpreter import Interpreter
 except ModuleNotFoundError:

frigate/detectors/plugins/deepstack.py

@@ -1,14 +1,14 @@
+import io
 import logging
+
 import numpy as np
 import requests
-import io
+from PIL import Image
+from pydantic import Field
+from typing_extensions import Literal
 
 from frigate.detectors.detection_api import DetectionApi
 from frigate.detectors.detector_config import BaseDetectorConfig
-from typing_extensions import Literal
-from pydantic import Field
-from PIL import Image
-
 
 logger = logging.getLogger(__name__)
 

frigate/detectors/plugins/edgetpu_tfl.py

@@ -1,10 +1,11 @@
 import logging
+
 import numpy as np
+from pydantic import Field
+from typing_extensions import Literal
 
 from frigate.detectors.detection_api import DetectionApi
 from frigate.detectors.detector_config import BaseDetectorConfig
-from typing_extensions import Literal
-from pydantic import Field
 
 try:
     from tflite_runtime.interpreter import Interpreter, load_delegate

frigate/detectors/plugins/openvino.py

@@ -1,12 +1,12 @@
 import logging
+
 import numpy as np
 import openvino.runtime as ov
+from pydantic import Field
+from typing_extensions import Literal
 
 from frigate.detectors.detection_api import DetectionApi
 from frigate.detectors.detector_config import BaseDetectorConfig, ModelTypeEnum
-from typing_extensions import Literal
-from pydantic import Field
-
 
 logger = logging.getLogger(__name__)
 

frigate/detectors/plugins/tensorrt.py

@@ -1,6 +1,6 @@
+import ctypes
 import logging
 
-import ctypes
 import numpy as np
 
 try:
@@ -11,10 +11,11 @@
 except ModuleNotFoundError:
     TRT_SUPPORT = False
 
+from pydantic import Field
+from typing_extensions import Literal
+
 from frigate.detectors.detection_api import DetectionApi
 from frigate.detectors.detector_config import BaseDetectorConfig
-from typing_extensions import Literal
-from pydantic import Field
 
 logger = logging.getLogger(__name__)
 

frigate/events/cleanup.py

@@ -4,16 +4,13 @@
 import logging
 import os
 import threading
-
+from multiprocessing.synchronize import Event as MpEvent
 from pathlib import Path
 
-
 from frigate.config import FrigateConfig
 from frigate.const import CLIPS_DIR
 from frigate.models import Event
 
-from multiprocessing.synchronize import Event as MpEvent
-
 logger = logging.getLogger(__name__)
 
 

frigate/events/external.py

@@ -1,16 +1,15 @@
 """Handle external events created by the user."""
 
 import base64
-import cv2
 import datetime
 import logging
 import os
 import random
 import string
-
+from multiprocessing.queues import Queue
 from typing import Optional
 
-from multiprocessing.queues import Queue
+import cv2
 
 from frigate.config import CameraConfig, FrigateConfig
 from frigate.const import CLIPS_DIR