Удалил мусор и сертификаты вроде добавил
This commit is contained in:
@@ -132,80 +132,75 @@ class DatabaseManager:
|
||||
self.logger.error(f"Тарифный план {plan_id} не найден.")
|
||||
return "ERROR"
|
||||
|
||||
# Проверка достаточности средств для покупки подписки
|
||||
# Проверка достаточности средств
|
||||
cost = int(plan["price"])
|
||||
if result.balance >= cost:
|
||||
result.balance -= cost
|
||||
await session.commit()
|
||||
|
||||
# Создание подписки для пользователя
|
||||
expiry_date = datetime.utcnow() + relativedelta(months=plan["duration_months"])
|
||||
server = await self.mongo_repo.get_server_with_least_clients()
|
||||
self.logger.info(f"{server}")
|
||||
new_subscription = Subscription(user_id=result.id, vpn_server_id=str(server['server']["name"]), plan=plan_id, expiry_date=expiry_date)
|
||||
session.add(new_subscription)
|
||||
await session.commit()
|
||||
|
||||
self.logger.info(f"Подписка успешно оформлена для пользователя {telegram_id} на план {plan_id}.")
|
||||
return "OK"
|
||||
else:
|
||||
if result.balance < cost:
|
||||
self.logger.error(f"Недостаточно средств у пользователя {telegram_id} для покупки плана {plan_id}.")
|
||||
return "INSUFFICIENT_FUNDS"
|
||||
except SQLAlchemyError as e:
|
||||
self.logger.error(f"Ошибка при покупке подписки {plan_id} для пользователя {telegram_id}: {e}")
|
||||
await session.rollback()
|
||||
return "ERROR"
|
||||
|
||||
async def add_to_server(self, telegram_id: int):
|
||||
"""
|
||||
Метод для добавления пользователя на сервер.
|
||||
"""
|
||||
async for session in self.session_generator():
|
||||
try:
|
||||
# Получаем подписку пользователя по telegram_id
|
||||
result = await session.execute(select(Subscription).join(User).where(User.telegram_id == int(telegram_id)))
|
||||
user_sub = result.scalars().first()
|
||||
# Списываем средства
|
||||
result.balance -= cost
|
||||
|
||||
if not user_sub:
|
||||
self.logger.error(f"Не удалось найти подписку для пользователя с Telegram ID {telegram_id}.")
|
||||
return "ERROR"
|
||||
# Создаем подписку
|
||||
expiry_date = datetime.utcnow() + relativedelta(months=plan["duration_months"])
|
||||
server = await self.mongo_repo.get_server_with_least_clients()
|
||||
self.logger.info(f"Выбран сервер для подписки: {server}")
|
||||
new_subscription = Subscription(
|
||||
user_id=result.id,
|
||||
vpn_server_id=str(server['server']["name"]),
|
||||
plan=plan_id,
|
||||
expiry_date=expiry_date
|
||||
)
|
||||
session.add(new_subscription)
|
||||
|
||||
# Попытка добавить пользователя на сервер
|
||||
# Получаем информацию о пользователе
|
||||
user_result = await session.execute(select(User).where(User.telegram_id == telegram_id))
|
||||
user = user_result.scalars().first()
|
||||
|
||||
# Получаем сервер с MongoDB
|
||||
server = await self.mongo_repo.get_server(user_sub.vpn_server_id)
|
||||
|
||||
if not server:
|
||||
self.logger.error(f"Не удалось найти сервер с ID {user_sub.vpn_server_id}.")
|
||||
user = result # так как result уже содержит пользователя
|
||||
if not user:
|
||||
self.logger.error(f"Не удалось найти пользователя для добавления на сервер.")
|
||||
await session.rollback()
|
||||
return "ERROR"
|
||||
|
||||
# Доступ к данным сервера для добавления клиента
|
||||
server_info = server['server']
|
||||
# Получаем сервер из MongoDB
|
||||
server_data = await self.mongo_repo.get_server(new_subscription.vpn_server_id)
|
||||
if not server_data:
|
||||
self.logger.error(f"Не удалось найти сервер с ID {new_subscription.vpn_server_id}.")
|
||||
await session.rollback()
|
||||
return "ERROR"
|
||||
|
||||
server_info = server_data['server']
|
||||
url_base = f"https://{server_info['ip']}:{server_info['port']}/{server_info['secretKey']}"
|
||||
login_data = {
|
||||
'username': server_info['login'],
|
||||
'password': server_info['password'],
|
||||
}
|
||||
|
||||
# Инициализируем взаимодействие с панелью управления
|
||||
panel = PanelInteraction(url_base, login_data, self.logger)
|
||||
panel = PanelInteraction(url_base, login_data, self.logger,)
|
||||
expiry_date_iso = new_subscription.expiry_date.isoformat()
|
||||
|
||||
expiry_date_iso = user_sub.expiry_date.isoformat()
|
||||
# Добавляем на сервер
|
||||
response = await panel.add_client(user.id, expiry_date_iso, user.username)
|
||||
|
||||
# Логируем результат
|
||||
if response == "OK":
|
||||
self.logger.info(f"Клиент {telegram_id} успешно добавлен на сервер.")
|
||||
return "OK"
|
||||
else:
|
||||
if response != "OK":
|
||||
self.logger.error(f"Ошибка при добавлении клиента {telegram_id} на сервер: {response}")
|
||||
# Если не получилось добавить на сервер, откатываем транзакцию
|
||||
await session.rollback()
|
||||
return "ERROR"
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при установке на сервер для пользователя {telegram_id}: {e}")
|
||||
# Если мы здесь - значит и подписка, и добавление на сервер успешны
|
||||
await session.commit()
|
||||
self.logger.info(f"Подписка успешно оформлена для пользователя {telegram_id} на план {plan_id} и клиент добавлен на сервер.")
|
||||
return "OK"
|
||||
|
||||
except SQLAlchemyError as e:
|
||||
self.logger.error(f"Ошибка при покупке подписки {plan_id} для пользователя {telegram_id}: {e}")
|
||||
await session.rollback()
|
||||
return "ERROR"
|
||||
except Exception as e:
|
||||
self.logger.error(f"Непредвиденная ошибка: {e}")
|
||||
await session.rollback()
|
||||
return "ERROR"
|
||||
|
||||
|
||||
@staticmethod
|
||||
def generate_string(length):
|
||||
|
||||
Reference in New Issue
Block a user