57 lines
2.2 KiB
Python
57 lines
2.2 KiB
Python
import os
|
||
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
|
||
from sqlalchemy.orm import sessionmaker
|
||
from app.services.db_manager import DatabaseManager
|
||
from .model import Base
|
||
try:
|
||
# Настройки PostgreSQL из переменных окружения
|
||
POSTGRES_DSN = os.getenv("POSTGRES_URL")
|
||
BASE_URL_MARZBAN = os.getenv("BASE_URL_MARZBAN")
|
||
USERNAME_MARZBA = os.getenv('USERNAME_MARZBAN')
|
||
PASSWORD_MARZBAN = os.getenv('PASSWORD_MARZBAN')
|
||
BILLING_URL = os.getenv('BILLING_URL')
|
||
|
||
# Создание движка для PostgreSQL
|
||
if POSTGRES_DSN is None or BASE_URL_MARZBAN is None or USERNAME_MARZBA is None or PASSWORD_MARZBAN is None or BILLING_URL is None:
|
||
raise Exception
|
||
postgres_engine = create_async_engine(POSTGRES_DSN, echo=False)
|
||
except Exception as e:
|
||
print("Ошибки при инициализации сессии постгреса")
|
||
AsyncSessionLocal = sessionmaker(bind=postgres_engine, class_=AsyncSession, expire_on_commit=False)
|
||
|
||
# Инициализация PostgreSQL
|
||
async def init_postgresql():
|
||
"""
|
||
Инициализация подключения к PostgreSQL.
|
||
"""
|
||
try:
|
||
async with postgres_engine.begin() as conn:
|
||
await conn.run_sync(Base.metadata.create_all)
|
||
print("PostgreSQL connected.")
|
||
except Exception as e:
|
||
print(f"Failed to connect to PostgreSQL: {e}")
|
||
|
||
# Получение сессии PostgreSQL
|
||
async def get_postgres_session():
|
||
"""
|
||
Асинхронный генератор сессий PostgreSQL.
|
||
"""
|
||
async with AsyncSessionLocal() as session:
|
||
yield session
|
||
|
||
# Закрытие соединений
|
||
async def close_connections():
|
||
"""
|
||
Закрытие всех соединений с базами данных.
|
||
"""
|
||
# Закрытие PostgreSQL
|
||
await postgres_engine.dispose()
|
||
print("PostgreSQL connection closed.")
|
||
|
||
|
||
def get_database_manager() -> DatabaseManager:
|
||
"""
|
||
Функция-зависимость для получения экземпляра DatabaseManager.
|
||
"""
|
||
return DatabaseManager(get_postgres_session, USERNAME_MARZBA,PASSWORD_MARZBAN,BASE_URL_MARZBAN,BILLING_URL)
|