75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
|
||
from sqlalchemy.orm import sessionmaker
|
||
from motor.motor_asyncio import AsyncIOMotorClient
|
||
from databases.model import Base
|
||
from utils.LogCon import setup_logger, load_config
|
||
|
||
# Загружаем конфигурацию
|
||
config = load_config()
|
||
|
||
# Настройки PostgreSQL
|
||
postgres_user = config['postgreSQL']['username']
|
||
postgres_password = config['postgreSQL']['password_DB']
|
||
postgres_host = "postgres" # Хост для PostgreSQL в Docker
|
||
POSTGRES_DSN = f"postgresql+asyncpg://{postgres_user}:{postgres_password}@{postgres_host}:5432/bot_db"
|
||
|
||
# Создание движка для PostgreSQL
|
||
postgres_engine = create_async_engine(POSTGRES_DSN, echo=False)
|
||
AsyncSessionLocal = sessionmaker(bind=postgres_engine, class_=AsyncSession, expire_on_commit=False)
|
||
|
||
# Настройки MongoDB
|
||
mongodb_user = config['mongodb']['mongodb_username']
|
||
mongodb_password = config['mongodb']['mongodb_password']
|
||
mongodb_host = "mongodb" # Хост для MongoDB в Docker
|
||
mongodb_uri = f"mongodb://{mongodb_user}:{mongodb_password}@{mongodb_host}:27017"
|
||
database_name = config['mongodb']['database_name']
|
||
|
||
# Создание клиента MongoDB
|
||
mongo_client = AsyncIOMotorClient(mongodb_uri)
|
||
mongo_db = mongo_client[database_name]
|
||
|
||
# Инициализация 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}")
|
||
|
||
# Инициализация MongoDB
|
||
async def init_mongodb():
|
||
"""
|
||
Проверка подключения к MongoDB.
|
||
"""
|
||
try:
|
||
# Проверяем подключение к MongoDB
|
||
await mongo_client.admin.command("ping")
|
||
print("MongoDB connected.")
|
||
except Exception as e:
|
||
print(f"Failed to connect to MongoDB: {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.")
|
||
|
||
# Закрытие MongoDB
|
||
mongo_client.close()
|
||
print("MongoDB connection closed.")
|