51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
from DB.db_main import *
|
|||
|
|
import platform
|
|||
|
|
import logging
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SqliteDbMain(DbMain):
|
|||
|
|
def __init__(self, database_name):
|
|||
|
|
# logging.basicConfig()
|
|||
|
|
# logging.getLogger('sqlalchemy.engine').setLevel(logging.ERROR)
|
|||
|
|
# logging.getLogger('sqlalchemy.pool').setLevel(logging.ERROR)
|
|||
|
|
# logging.getLogger('sqlalchemy.dialects').setLevel(logging.ERROR)
|
|||
|
|
# logging.getLogger('sqlalchemy.orm').setLevel(logging.ERROR)
|
|||
|
|
self.database_name = database_name
|
|||
|
|
super().__init__()
|
|||
|
|
self.engine = self.__create_sqlite_engine()
|
|||
|
|
self.engine_path = self.__get_path()
|
|||
|
|
self.session = self.get_session()
|
|||
|
|
self.inspector = inspect(self.engine)
|
|||
|
|
|
|||
|
|
def __get_path(self):
|
|||
|
|
sys_platform = platform.platform().lower()
|
|||
|
|
print(f'当前操作系统：{platform.platform()}')
|
|||
|
|
__engine = ''
|
|||
|
|
if 'windows' in sys_platform.lower():
|
|||
|
|
__engine = f"E:\\sqlite_db\\stock_db\\{self.database_name}"
|
|||
|
|
elif 'macos' in sys_platform.lower():
|
|||
|
|
__engine = f"/Users/renmeng/Documents/sqlite_db/{self.database_name}"
|
|||
|
|
else:
|
|||
|
|
__engine = f"{self.database_name}"
|
|||
|
|
return __engine
|
|||
|
|
|
|||
|
|
def __create_sqlite_engine(self):
|
|||
|
|
sys_platform = platform.platform().lower()
|
|||
|
|
__engine = ''
|
|||
|
|
if 'windows' in sys_platform.lower():
|
|||
|
|
__engine = f"sqlite:///E:\\sqlite_db\\stock_db\\{self.database_name}"
|
|||
|
|
elif 'macos' in sys_platform.lower():
|
|||
|
|
__engine = f"sqlite:////Users/renmeng/Documents/sqlite_db/{self.database_name}"
|
|||
|
|
else:
|
|||
|
|
__engine = f"sqlite:///{self.database_name}"
|
|||
|
|
print(f"当前__engine是：{__engine}")
|
|||
|
|
return create_engine(__engine, pool_size=10, pool_timeout=10, echo=True)
|
|||
|
|
|
|||
|
|
def get_db_size(self):
|
|||
|
|
file_size = Path(self.engine_path).stat().st_size
|
|||
|
|
total = f"{file_size / (1024 * 1024):.2f} MB"
|
|||
|
|
print(f"文件大小: {file_size / (1024 * 1024):.2f} MB")
|
|||
|
|
return total
|