cyonjan/PyWxDump
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
xaoyaoo 2024-08-11 10:56:16 +08:00
parent 9a74e418ca
commit 472611153a
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pywxdump/db/dbbase.py
View File

@ -38,13 +38,13 @@ class DatabaseSingletonBase:
""" """
if not db_config: if not db_config:
raise ValueError("db_config 不能为空") raise ValueError("db_config 不能为空")
db_key = db_config["key"] db_key = db_config.get("key", "xaoyaoo_741852963")
db_type = db_config["type"] db_type = db_config.get("type", "sqlite")
if db_key in cls._db_pool and cls._db_pool[db_key] is not None: if db_key in cls._db_pool and cls._db_pool[db_key] is not None:
return cls._db_pool[db_key] return cls._db_pool[db_key]
if db_type == "sqlite": if db_type == "sqlite":
db_path = db_config["path"] db_path = db_config.get("path", "")
if not os.path.exists(db_path): if not os.path.exists(db_path):
raise FileNotFoundError(f"文件不存在: {db_path}") raise FileNotFoundError(f"文件不存在: {db_path}")
pool = PooledDB( pool = PooledDB(