fix None issue on db shutdown during failure

TISUnion · Dec 9, 2023 · 1e04b46 · 1e04b46
1 parent 5596266
commit 1e04b46
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/prime_backup/db/access.py b/prime_backup/db/access.py
@@ -33,9 +33,9 @@ def init(cls, auto_migrate: bool = True):
 
 	@classmethod
 	def shutdown(cls):
-		logger = db_logger.get()
-		for hdr in list(logger.handlers):
-			logger.removeHandler(hdr)
+		if (logger := db_logger.get()) is not None:
+			for hdr in list(logger.handlers):
+				logger.removeHandler(hdr)
 
 	@classmethod
 	def get_db_path(cls) -> Path: