81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
|
"""
|
|||
|
|
Policy Service - Проверка полисов в MySQL БД
|
|||
|
|
"""
|
|||
|
|
import aiomysql
|
|||
|
|
from typing import Optional, Dict, Any
|
|||
|
|
from ..config import settings
|
|||
|
|
import logging
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PolicyService:
|
|||
|
|
"""Сервис для проверки полисов ERV"""
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
self.pool: Optional[aiomysql.Pool] = None
|
|||
|
|
|
|||
|
|
async def connect(self):
|
|||
|
|
"""Подключение к MySQL БД с полисами"""
|
|||
|
|
try:
|
|||
|
|
# Используем credentials из .env через settings
|
|||
|
|
self.pool = await aiomysql.create_pool(
|
|||
|
|
host=settings.mysql_host,
|
|||
|
|
port=settings.mysql_port,
|
|||
|
|
user=settings.mysql_user,
|
|||
|
|
password=settings.mysql_password,
|
|||
|
|
db=settings.mysql_db,
|
|||
|
|
autocommit=True,
|
|||
|
|
minsize=1,
|
|||
|
|
maxsize=5
|
|||
|
|
)
|
|||
|
|
logger.info(f"✅ MySQL Policy DB connected: {settings.mysql_host}/{settings.mysql_db}")
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"❌ MySQL Policy DB connection error: {e}")
|
|||
|
|
raise
|
|||
|
|
|
|||
|
|
async def check_policy(self, voucher: str) -> Optional[Dict[str, Any]]:
|
|||
|
|
"""
|
|||
|
|
Проверить полис в БД
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
voucher: Номер полиса вида E1000-302538524
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Dict с данными полиса или None если не найден
|
|||
|
|
"""
|
|||
|
|
if not self.pool:
|
|||
|
|
await self.connect()
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
async with self.pool.acquire() as conn:
|
|||
|
|
async with conn.cursor(aiomysql.DictCursor) as cursor:
|
|||
|
|
# Запрос поиска по номеру полиса в таблице lexrpiority
|
|||
|
|
query = "SELECT * FROM lexrpiority WHERE voucher = %s LIMIT 1"
|
|||
|
|
|
|||
|
|
await cursor.execute(query, [voucher])
|
|||
|
|
result = await cursor.fetchone()
|
|||
|
|
|
|||
|
|
if result:
|
|||
|
|
logger.info(f"✅ Policy found: {voucher}")
|
|||
|
|
return dict(result)
|
|||
|
|
else:
|
|||
|
|
logger.warning(f"⚠️ Policy not found: {voucher}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"Error checking policy: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
async def close(self):
|
|||
|
|
"""Закрыть пул подключений"""
|
|||
|
|
if self.pool:
|
|||
|
|
self.pool.close()
|
|||
|
|
await self.pool.wait_closed()
|
|||
|
|
logger.info("MySQL Policy DB pool closed")
|
|||
|
|
|
|||
|
|
|
|||
|
|
# Глобальный экземпляр
|
|||
|
|
policy_service = PolicyService()
|
|||
|
|
|