mirror of
https://github.com/LC044/WeChatMsg
synced 2025-04-20 11:08:03 +08:00
43 lines
1.1 KiB
Python
43 lines
1.1 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 .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:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def get_interface(self) -> DataBaseInterface:
|
|||
|
|
return self._initialize_database()
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
使用示例：
|
|||
|
|
conn = DatabaseConnection(USER_DB_DIR, 4)
|
|||
|
|
database: DataBaseInterface = conn.get_interface()
|
|||
|
|
"""
|