283375/arcaea-offline
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline.git synced 2025-04-19 06:00:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
283375 2023-06-27 23:27:10 +08:00
parent 47db4ca120
commit c1085c1f65
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/arcaea_offline/database.py
View File

@ -1,3 +1,4 @@
import atexit
import os import os
import sqlite3 import sqlite3
from dataclasses import fields, is_dataclass 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 journal_mode = WAL;")
self.__conn.execute("PRAGMA foreign_keys = ON;") self.__conn.execute("PRAGMA foreign_keys = ON;")
atexit.register(self.__conn.close)
self.__update_hooks = [] self.__update_hooks = []
@property @property