lifegpc/game-backuper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

Fix bug

Browse Source
This commit is contained in:
lifegpc
2024-08-06 15:20:06 +08:00
parent b5bc5cbf39
commit 4e816cbf61
1 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
game_backuper/db.py
View File

@@ -53,6 +53,7 @@ def getpass(prompt, cfg: Config) -> str:
class Db: class Db:
VERSION = [1, 0, 0, 2] VERSION = [1, 0, 0, 2]
fn = None
def __check_database(self) -> bool: def __check_database(self) -> bool:
self.__updateExistsTable() self.__updateExistsTable()
@@ -85,9 +86,10 @@ class Db:
def __init__(self, config: Config, opts: Opts): def __init__(self, config: Config, opts: Opts):
self._cfg = config self._cfg = config
self._opt = opts self._opt = opts
fn = config.db_path if config.db_path else join(config.dest, "data.db") self.fn = config.db_path if config.db_path else join(
hydrate_file_if_needed(fn) config.dest, "data.db")
self.db = connect(fn, check_same_thread=False) hydrate_file_if_needed(self.fn)
self.db = connect(self.fn, check_same_thread=False)
if config.encrypt_db: if config.encrypt_db:
passpharse = getpass('Please input the password of the database:', config) # noqa: E501 passpharse = getpass('Please input the password of the database:', config) # noqa: E501
if not self.encrypted: if not self.encrypted:
@@ -100,8 +102,8 @@ class Db:
db.execute(q) db.execute(q)
self.db.close() self.db.close()
db.close() db.close()
move(tfn, fn) move(tfn, self.fn)
self.db = connect(fn, check_same_thread=False) self.db = connect(self.fn, check_same_thread=False)
elif opts.change_key: elif opts.change_key:
self.__set_encrypt_key(passpharse) self.__set_encrypt_key(passpharse)
passpharse = getpass('Please input new password of the database:', config) # noqa: E501 passpharse = getpass('Please input new password of the database:', config) # noqa: E501
@@ -114,8 +116,8 @@ class Db:
db.execute(q) db.execute(q)
self.db.close() self.db.close()
db.close() db.close()
move(tfn, fn) move(tfn, self.fn)
self.db = connect(fn, check_same_thread=False) self.db = connect(self.fn, check_same_thread=False)
self.__set_encrypt_key(passpharse) self.__set_encrypt_key(passpharse)
else: else:
if self.encrypted: if self.encrypted:
@@ -129,8 +131,8 @@ class Db:
db.execute(q) db.execute(q)
self.db.close() self.db.close()
db.close() db.close()
move(tfn, fn) move(tfn, self.fn)
self.db = connect(fn, check_same_thread=False) self.db = connect(self.fn, check_same_thread=False)
if opts.optimize_db: if opts.optimize_db:
self.db.execute('VACUUM;') self.db.execute('VACUUM;')
self.db.commit() self.db.commit()
@@ -194,7 +196,7 @@ class Db:
@property @property
def encrypted(self): def encrypted(self):
try: try:
con = connect(join(self._cfg.dest, 'data.db')) con = connect(self.fn)
con.execute('SELECT count(*) FROM sqlite_master;') con.execute('SELECT count(*) FROM sqlite_master;')
con.close() con.close()
return False return False