impr(db): close connection when program exit using atexit hook

This commit is contained in:
283375 2023-06-27 23:27:10 +08:00
parent 47db4ca120
commit c1085c1f65

View File

@ -1,3 +1,4 @@
import atexit
import os
import sqlite3
from dataclasses import fields, is_dataclass
@ -30,6 +31,8 @@ class Database(metaclass=Singleton):
self.__conn.execute("PRAGMA journal_mode = WAL;")
self.__conn.execute("PRAGMA foreign_keys = ON;")
atexit.register(self.__conn.close)
self.__update_hooks = []
@property