Disledg
72d7fdd751
1. Оно перво на перво работает 2.Реализовано почти всё вроде из кнопок, осталось ток оплату подписок, выдачу URI и пополнение конечно. 3.Убрал .json конфиг и сделал всё через переменные окружения
66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
import os
|
||
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
|
||
|
||
# Настройки PostgreSQL из переменных окружения
|
||
POSTGRES_DSN = os.getenv("POSTGRES_URL")
|
||
|
||
# Создание движка для PostgreSQL
|
||
postgres_engine = create_async_engine(POSTGRES_DSN, echo=False)
|
||
AsyncSessionLocal = sessionmaker(bind=postgres_engine, class_=AsyncSession, expire_on_commit=False)
|
||
|
||
# Настройки MongoDB из переменных окружения
|
||
MONGO_URI = os.getenv("MONGO_URL")
|
||
DATABASE_NAME = os.getenv("DB_NAME")
|
||
|
||
# Создание клиента MongoDB
|
||
mongo_client = AsyncIOMotorClient(MONGO_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.")
|