This commit is contained in:
Andrey
2024-07-06 20:53:33 +03:00
parent 99eb07f74e
commit 9788a763c4
2 changed files with 10 additions and 7 deletions

8
db.py
View File

@@ -311,14 +311,14 @@ class BotDB:
logger.error(f"Ошибка удаления пользователя с идентификатором {user_id} из таблицы blacklist. Ошибка: {str(error)}")
return False
def add_new_message_in_db(self, message_text, user_id, message_id, date, has_answer):
def add_new_message_in_db(self, message_text, user_id, message_id, date):
"""Добавляем сообщение юзера в базу"""
try:
self.cursor.execute(
"INSERT INTO `user_messages` (message_text, user_id, message_id, date, has_answer) "
"VALUES (?, ?, ?, ?, ?)",
(message_text, message_id, user_id, date, has_answer))
"INSERT INTO `user_messages` (message_text, user_id, message_id, date) "
"VALUES (?, ?, ?, ?)",
(message_text, message_id, user_id, date))
return self.conn.commit()
except sqlite3.Error as error:
print(error)

View File

@@ -6,17 +6,20 @@ from time import sleep
from enum import Enum
from typing import Any
from apscheduler.schedulers.background import BackgroundScheduler
import db
from db import BotDB
import telebot
import random
from datetime import datetime, timedelta
import time
from telebot import types
from telebot.apihelper import ApiTelegramException
import messages
import traceback
#TODO: Добавить проверку можно ли отвечать пользователю? Сейчас если у него скрыто лс, ему похоже не приходят сообщения
#TODO Подумать над реализацией функционала с поступлениями в колледжи
#TODO: Покрыть все логированием и ошибками корректными. Ерроры кидать в чат.
#TODO: Покрыть все тестами
# Настройки
config_path = os.path.join(sys.path[0], 'settings.ini')
config = configparser.ConfigParser()
@@ -366,7 +369,7 @@ class TelegramHelperBot:
)
current_date = datetime.now()
date = current_date.strftime("%Y-%m-%d %H:%M:%S")
BotDB.add_new_message_in_db(message.text, message.message_id + 1, message.from_user.id, date, 0)
BotDB.add_new_message_in_db(message.text, message.message_id + 1, message.from_user.id, date)
question = messages.get_message(self.__get_first_name(message), 'QUESTION')
markup = self.get_reply_keyboard(message)
self.bot.send_message(message.chat.id, question, parse_mode='html', disable_web_page_preview=not PREVIEW_LINK,