mirror of
https://github.com/LC044/WeChatMsg
synced 2025-04-08 05:28:03 +08:00
45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
@File : __init__.py.py
|
|
@Author : Shuaikang Zhou
|
|
@Time : 2023/1/5 0:10
|
|
@IDE : Pycharm
|
|
@Version : Python3.10
|
|
@comment : ···
|
|
"""
|
|
from .log import logger
|
|
from .model import Me, MessageType, Message, Person, Contact, TextMessage, ImageMessage
|
|
from .db_main import DataBaseInterface
|
|
from .manager_v4 import DataBaseV4
|
|
from .manager_v3 import DataBaseV3
|
|
|
|
__version__ = '3.0.0'
|
|
|
|
|
|
class DatabaseConnection:
|
|
def __init__(self, db_dir, db_version=4):
|
|
self.db_dir = db_dir
|
|
self.db_version = db_version
|
|
self.database_interface = self._initialize_database()
|
|
|
|
def _initialize_database(self) -> DataBaseInterface:
|
|
if self.db_version == 4:
|
|
database0 = DataBaseV4()
|
|
else:
|
|
database0 = DataBaseV3()
|
|
if database0.init_database(self.db_dir):
|
|
return database0
|
|
else:
|
|
logger.error(f'数据库初始化失败, 请检查路径或数据库版本是否正确, db_dir:{self.db_dir},db_version:{self.db_version}')
|
|
return None
|
|
|
|
def get_interface(self) -> DataBaseInterface:
|
|
return self._initialize_database()
|
|
|
|
|
|
"""
|
|
使用示例：
|
|
conn = DatabaseConnection(USER_DB_DIR, 4)
|
|
database: DataBaseInterface = conn.get_interface()
|
|
"""
|