diff --git a/evtg_publish.py b/evtg_publish.py index 23d0a6b..3c6cb4b 100644 --- a/evtg_publish.py +++ b/evtg_publish.py @@ -25,6 +25,7 @@ DESC_PREFIX = os.getenv('DESC_PREFIX') PZK_PREFIX = os.getenv('PZK_PREFIX') LOG_FILE = os.getenv('LOG_FILE', 'evtg_publisher.log') EVENT_POST_DELAY = int(os.getenv('EVENT_POST_DELAY', 0)) # Задержка в минутах +WORKMODE = os.getenv('WORKMODE', 'ZILANT') # Режим работы: ZILANT или VOLK # Параметры длины сообщений MAX_CAPTION_LENGTH = int(os.getenv('MAX_CAPTION_LENGTH', 1000)) @@ -230,23 +231,39 @@ async def tg_post_event_by_id(id_event): cursorclass=pymysql.cursors.DictCursor ) - # Вычисляем время, до которого должно быть изменено событие для публикации - modified_before = datetime.now(timezone.utc) - timedelta(minutes=EVENT_POST_DELAY) - + # Проверка времени изменения только в режиме ZILANT with conn.cursor() as cursor: - cursor.execute(""" - SELECT id_event, number, tags, unit_name, name, about, about_social_picture, tg_message_id - FROM events - WHERE id_event = %s - AND marked_to_publication = True - AND is_posted_tg = False - AND modified <= %s - """, (id_event, modified_before)) - event = cursor.fetchone() + if WORKMODE == "ZILANT": + # Вычисляем время, до которого должно быть изменено событие для публикации + modified_before = datetime.now(timezone.utc) - timedelta(minutes=EVENT_POST_DELAY) + + cursor.execute(""" + SELECT id_event, number, tags, unit_name, name, about, about_social_picture, tg_message_id + FROM events + WHERE id_event = %s + AND marked_to_publication = True + AND is_posted_tg = False + AND modified <= %s + """, (id_event, modified_before)) + event = cursor.fetchone() - if not event: - logger.info(f"Событие ID {id_event} не найдено, уже опубликовано или изменено менее чем {EVENT_POST_DELAY} минут назад") - return + if not event: + logger.info(f"Событие ID {id_event} не найдено, уже опубликовано или изменено менее чем {EVENT_POST_DELAY} минут назад") + return + else: + # В режиме VOLK проверка времени не выполняется + cursor.execute(""" + SELECT id_event, number, tags, unit_name, name, about, about_social_picture, tg_message_id + FROM events + WHERE id_event = %s + AND marked_to_publication = True + AND is_posted_tg = False + """, (id_event,)) + event = cursor.fetchone() + + if not event: + logger.info(f"Событие ID {id_event} не найдено или уже опубликовано") + return old_message_id = event.get('tg_message_id') # Сохраняем старый ID сообщения bot = Bot(token=BOT_TOKEN) @@ -299,18 +316,29 @@ async def tg_post_all_events(): cursorclass=pymysql.cursors.DictCursor ) - # Вычисляем время, до которого должно быть изменено событие для публикации - modified_before = datetime.now(timezone.utc) - timedelta(minutes=EVENT_POST_DELAY) - with conn.cursor() as cursor: - cursor.execute(""" - SELECT id_event - FROM events - WHERE marked_to_publication = True - AND is_posted_tg = False - AND modified <= %s - LIMIT 5 - """, (modified_before,)) + # Проверка времени изменения только в режиме ZILANT + if WORKMODE == "ZILANT": + # Вычисляем время, до которого должно быть изменено событие для публикации + modified_before = datetime.now(timezone.utc) - timedelta(minutes=EVENT_POST_DELAY) + + cursor.execute(""" + SELECT id_event + FROM events + WHERE marked_to_publication = True + AND is_posted_tg = False + AND modified <= %s + LIMIT 5 + """, (modified_before,)) + else: + # В режиме VOLK проверка времени не выполняется + cursor.execute(""" + SELECT id_event + FROM events + WHERE marked_to_publication = True + AND is_posted_tg = False + LIMIT 5 + """) events = cursor.fetchall() event_count = 0 diff --git a/volk_load.py b/volk_load.py index c1dd2c6..fe61e4c 100644 --- a/volk_load.py +++ b/volk_load.py @@ -172,18 +172,14 @@ def load_categories(): try: desc_url = VOLK_CATEGORY_DESC_PREFIX + category_id log_message(f"Запрос описания для категории {category_id} из {desc_url}") - log_message("отладочное сообщение для проверки вывода в лог-файл") - + desc_response = None try: # Добавляем таймаут для избежания зависаний (30 секунд) from urllib.request import Request req = Request(desc_url, headers={'User-Agent': 'Mozilla/5.0 (Telefaust)'}) log_message(f"request response = {req}", "DEBUG") - safe_log_direct("run urlopen", "DEBUG") response = urlopen(req, timeout=30) - safe_log_direct("done urlopen", "ERROR") - log_message("urlopen успешно выполнен") except (URLError, HTTPError, OSError, socket.gaierror, socket.timeout, ConnectionError) as e: # Используем безопасное логирование для критических ошибок safe_log_direct("ПЕРЕХВАЧЕНА СЕТЕВАЯ ОШИБКА urlopen", "ERROR") @@ -225,10 +221,8 @@ def load_categories(): with response: # Явно указываем кодировку UTF-8 при декодировании raw_data = response.read() - log_message("raw data read") try: decoded_data = raw_data.decode('utf-8') - log_message("data decoded") except UnicodeDecodeError as e: error_msg = f"Ошибка декодирования ответа для категории {category_id}: {e}. Попытка декодирования как latin-1." log_message(error_msg, "WARNING") @@ -236,7 +230,6 @@ def load_categories(): try: desc_response = json.loads(decoded_data) - log_message("json loaded") except json.JSONDecodeError as e: error_msg = f"Ошибка парсинга JSON для категории {category_id}: {e}. Первые 500 символов ответа: {decoded_data[:500]}" log_message(error_msg, "ERROR")