administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-14 23:25:25 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fixed typing errors

Browse Source
This commit is contained in:
George Tsiamasiotis 2024-09-17 13:59:49 +03:00
parent f5325e0567
commit 2445d706df
2 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/__main__.py
View File

@ -7,7 +7,7 @@ from flask import cli
from frigate.app import FrigateApp from frigate.app import FrigateApp
def main(): def main() -> None:
faulthandler.enable() faulthandler.enable()
logging.basicConfig(level=logging.INFO, handlers=[]) logging.basicConfig(level=logging.INFO, handlers=[])

18
frigate/log.py
View File

@ -6,7 +6,10 @@ import threading
from collections import deque from collections import deque
from contextlib import AbstractContextManager, ContextDecorator from contextlib import AbstractContextManager, ContextDecorator
from logging.handlers import QueueHandler, QueueListener from logging.handlers import QueueHandler, QueueListener
from typing import Deque from types import TracebackType
from typing import Deque, Optional
from typing_extensions import Self
from frigate.util.builtin import clean_camera_user_pass from frigate.util.builtin import clean_camera_user_pass
@ -30,7 +33,7 @@ class log_thread(AbstractContextManager, ContextDecorator):
self._handler = handler self._handler = handler
log_queue = mp.Queue() log_queue: mp.Queue = mp.Queue()
self._queue_handler = QueueHandler(log_queue) self._queue_handler = QueueHandler(log_queue)
self._log_listener = QueueListener( self._log_listener = QueueListener(
@ -41,10 +44,10 @@ class log_thread(AbstractContextManager, ContextDecorator):
def handler(self) -> logging.Handler: def handler(self) -> logging.Handler:
return self._handler return self._handler
def _stop_thread(self): def _stop_thread(self) -> None:
self._log_listener.stop() self._log_listener.stop()
def __enter__(self): def __enter__(self) -> Self:
logging.getLogger().addHandler(self._queue_handler) logging.getLogger().addHandler(self._queue_handler)
atexit.register(self._stop_thread) atexit.register(self._stop_thread)
@ -52,7 +55,12 @@ class log_thread(AbstractContextManager, ContextDecorator):
return self return self
def __exit__(self, *exc): def __exit__(
self,
exc_type: Optional[type[BaseException]],
exc_info: Optional[BaseException],
exc_tb: Optional[TracebackType],
) -> None:
logging.getLogger().removeHandler(self._queue_handler) logging.getLogger().removeHandler(self._queue_handler)
atexit.unregister(self._stop_thread) atexit.unregister(self._stop_thread)