UdoChudo
ccb47d527f
All checks were successful
Build and Push Docker Images / build (push) Successful in 1m28s
- Рефакторинг Telegram бота на модульную структуру для удобства поддержки и расширения - Создан общий RabbitMQ клиент для Flask и Telegram компонентов - Подготовлена базовая архитектура для будущего масштабирования и новых функций Signed-off-by: UdoChudo <stream@udochudo.ru>
63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
import pika
|
|
from config import RABBITMQ_HOST, RABBITMQ_PORT, RABBITMQ_LOGIN, RABBITMQ_PASS, RABBITMQ_QUEUE
|
|
import time
|
|
|
|
class RabbitMQClient:
|
|
def __init__(self):
|
|
self._connect()
|
|
|
|
def _connect(self):
|
|
credentials = pika.PlainCredentials(RABBITMQ_LOGIN, RABBITMQ_PASS)
|
|
parameters = pika.ConnectionParameters(
|
|
host=RABBITMQ_HOST,
|
|
port=RABBITMQ_PORT,
|
|
credentials=credentials,
|
|
heartbeat=600,
|
|
blocked_connection_timeout=300
|
|
)
|
|
self.connection = pika.BlockingConnection(parameters)
|
|
self.channel = self.connection.channel()
|
|
self.channel.queue_declare(queue=RABBITMQ_QUEUE, durable=True)
|
|
|
|
def publish_message(self, message: str, retry=1):
|
|
try:
|
|
if not self.connection or self.connection.is_closed or not self.channel or self.channel.is_closed:
|
|
self._connect()
|
|
self.channel.basic_publish(
|
|
exchange='',
|
|
routing_key=RABBITMQ_QUEUE,
|
|
body=message,
|
|
properties=pika.BasicProperties(
|
|
delivery_mode=2,
|
|
)
|
|
)
|
|
except (pika.exceptions.ChannelClosedByBroker,
|
|
pika.exceptions.ConnectionClosed,
|
|
pika.exceptions.AMQPConnectionError) as e:
|
|
if retry > 0:
|
|
# Короткая пауза перед переподключением
|
|
time.sleep(1)
|
|
self._connect()
|
|
self.publish_message(message, retry=retry-1)
|
|
else:
|
|
raise e
|
|
|
|
def close(self):
|
|
if self.connection and self.connection.is_open:
|
|
self.connection.close()
|
|
|
|
|
|
|
|
def close(self):
|
|
if self.connection and self.connection.is_open:
|
|
self.connection.close()
|
|
|
|
|
|
# Глобальная переменная
|
|
rabbitmq_client = None
|
|
|
|
def get_rabbitmq_client():
|
|
global rabbitmq_client
|
|
if rabbitmq_client is None:
|
|
rabbitmq_client = RabbitMQClient()
|
|
return rabbitmq_client |