mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 12:05:54 +00:00
31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
import pip
|
|
import sqlalchemy.ext.asyncio as sqlalchemy_asyncio
|
|
import sys
|
|
|
|
from .. import database
|
|
|
|
|
|
@database.manager_class('postgresql')
|
|
class PostgreSQLDatabaseManager(database.BaseDatabaseManager):
|
|
"""PostgreSQL database manager"""
|
|
|
|
async def initialize(self) -> None:
|
|
|
|
# default to PostgreSQL with asyncpg driver
|
|
try:
|
|
__import__("asyncpg")
|
|
except ImportError:
|
|
print('以下依赖包未安装，将自动安装，请完成后重启程序：')
|
|
print(
|
|
'The dependence package asyncpg is missing, it will be installed automatically, please restart the program after completion:'
|
|
)
|
|
pip.main(['install', "asyncpg"])
|
|
print('已自动安装缺失的依赖包 asyncpg ，请重启程序。')
|
|
print('The missing dependence asyncpg have been installed automatically, please restart the program.')
|
|
sys.exit(0)
|
|
|
|
engine_url = self.ap.instance_config.data['system'].get('database', {}).get('engine_url', 'postgresql+asyncpg://root:***@127.0.0.1:5432/postgres')
|
|
self.engine = sqlalchemy_asyncio.create_async_engine(engine_url)
|