more type hints

Author: Fallen-Breath

prime_backup/action/__init__.py

@@ -1,6 +1,7 @@
 """
 Actions for all kinds of DB accesses
 """
+import logging
 import threading
 from abc import ABC, abstractmethod
 from typing import TypeVar, Generic
@@ -14,8 +15,8 @@ def __init__(self):
 
 		from prime_backup import logger
 		from prime_backup.config.config import Config
-		self.logger = logger.get()
-		self.config = Config.get()
+		self.logger: logging.Logger = logger.get()
+		self.config: Config = Config.get()
 
 	@abstractmethod
 	def run(self) -> _T:

prime_backup/db/migrations/__init__.py

@@ -1,3 +1,4 @@
+import logging
 import os
 from abc import abstractmethod, ABC
 from pathlib import Path
@@ -12,7 +13,7 @@
 
 class MigrationImplBase(ABC):
 	def __init__(self, engine: Engine, temp_dir: Path, session: Session):
-		self.logger = logger.get()
+		self.logger: logging.Logger = logger.get()
 		self.engine = engine
 		self.temp_file_store = TempFileStore(temp_dir / f'migration_{os.getpid()}')
 		self.session = session

prime_backup/mcdr/crontab_job/basic_job.py

@@ -1,5 +1,6 @@
 import dataclasses
 import datetime
+import logging
 import threading
 from abc import ABC, abstractmethod
 from typing import TYPE_CHECKING, List, Optional, Any, Callable
@@ -34,8 +35,8 @@ class BasicCrontabJob(CrontabJob, TranslationContext, ABC):
 	def __init__(self, scheduler: BaseScheduler, task_manager: 'TaskManager'):
 		CrontabJob.__init__(self, scheduler, task_manager)
 		TranslationContext.__init__(self, f'job.{self.id.name}')
-		self._root_config = Config.get()
-		self.logger = logger.get()
+		self._root_config: Config = Config.get()
+		self.logger: logging.Logger = logger.get()
 		self.aps_job: Optional[Job] = None
 		self.abort_event = threading.Event()
 

prime_backup/mcdr/task/basic_task.py

@@ -1,3 +1,4 @@
+import logging
 import threading
 from abc import ABC
 from typing import Union, Optional, TypeVar
@@ -24,8 +25,8 @@ def __init__(self, source: CommandSource):
 
 		from prime_backup import logger
 		from prime_backup.config.config import Config
-		self.logger = logger.get()
-		self.config = Config.get()
+		self.logger: logging.Logger = logger.get()
+		self.config: Config = Config.get()
 
 		self.aborted_event = threading.Event()
 		self.plugin_unloaded_event = threading.Event()