telegrambot/main.py

132 lines
5.4 KiB
Python
Raw Normal View History

2024-05-14 21:21:27 +03:00
import telebot
import datetime
import sqlite3
2024-05-21 21:13:08 +03:00
import re
2024-05-14 21:21:27 +03:00
class Database:
def __init__(self,db_name) -> None:
self.db_name = db_name
self.__create_table()
def __create_table(self):
sql = self.connect_db()
2024-05-16 21:15:22 +03:00
sql["cursor"].execute("""
create table if not exists users (
id integer primary key autoincrement,
id_telegram integer not null unique,
username text,
first_name text,
last_name text,
date_register date,
access boolean default 1
)
""")
sql["cursor"].execute("""
create table if not exists messages (
id integer primary key autoincrement,
id_user integer not null,
message_id integer not null,
message_text text not null,
date_send date,
status boolean default 0 check(status in (0,1)),
foreign key (id_user) references users(id)
)
""")
2024-05-14 21:21:27 +03:00
self.close(sql['connect'],sql['cursor'])
def connect_db(self):
with sqlite3.connect(self.db_name) as connect:
cursor = connect.cursor()
return {"connect":connect,"cursor":cursor}
2024-05-16 21:15:22 +03:00
def check_user(self,user_id: int):
sql = self.connect_db()
sql["cursor"].execute("select * from users where id_telegram = ?", (user_id,))
info_users = sql["cursor"].fetchone()
self.close(sql['connect'],sql['cursor'])
if info_users is None:
2024-05-21 21:13:08 +03:00
return {"status":False,"info_users":info_users}
2024-05-16 21:15:22 +03:00
return {"status":True,"info_user":info_users}
def create_user(self, message: telebot.types.Message):
sql = self.connect_db()
date = datetime.datetime.now().strftime("%Y-%m-%d")
sql["cursor"].execute("""
insert into users (id_telegram, username, first_name, last_name, date_register) values (?,?,?,?,?)
""", (message.from_user.id, message.from_user.username,message.from_user.first_name,message.from_user.last_name,date))
sql["connect"].commit()
self.close(sql['connect'],sql['cursor'])
2024-05-21 21:13:08 +03:00
def create_message(self,message: telebot.types.Message,info_user):
2024-05-16 21:15:22 +03:00
sql = self.connect_db()
date = datetime.datetime.now().strftime("%Y-%m-%d")
sql["cursor"].execute("""
insert into messages (id_user, message_id, message_text, date_send) values (?,?,?,?)
2024-05-21 21:13:08 +03:00
""", (info_user[0],message.message_id,message.text,date))
2024-05-18 17:36:50 +03:00
sql["connect"].commit()
2024-05-21 21:13:08 +03:00
2024-05-16 21:15:22 +03:00
self.close(sql['connect'],sql['cursor'])
2024-05-21 21:13:08 +03:00
return sql["cursor"].lastrowid
def check_application(self,id_application) -> int:
sql = self.connect_db()
sql["cursor"].execute("""
select message_id from messages where id = ?
""",(id_application,))
data_message = sql["cursor"].fetchone()
self.close(sql["connect"],sql["cursor"])
return data_message[0]
2024-05-14 21:21:27 +03:00
def close(self,connect,cursor):
cursor.close()
connect.close()
class TelegramBot(Database):
def __init__(self,db_name,token):
super().__init__(db_name)
self.bot = telebot.TeleBot(token)
2024-05-21 21:13:08 +03:00
self.admin_chat_id = -4235588549
2024-05-14 21:21:27 +03:00
self.router()
2024-05-16 21:15:22 +03:00
2024-05-14 21:21:27 +03:00
def router(self):
@self.bot.message_handler(commands=['start'])
2024-05-21 21:13:08 +03:00
def start(message: telebot.types.Message):
2024-05-16 21:15:22 +03:00
text = ""
if self.check_user(message.from_user.id)["status"]:
text += f"С возвращением, {message.from_user.first_name}"
else:
self.create_user(message)
text += f"Добро пожаловать, {message.from_user.first_name}"
2024-05-14 21:21:27 +03:00
self.bot.send_message(
message.chat.id,
2024-05-16 21:15:22 +03:00
text
2024-05-14 21:21:27 +03:00
)
@self.bot.message_handler(func=lambda message: True)
2024-05-21 21:13:08 +03:00
def echo_all(message: telebot.types.Message):
if message.chat.id == self.admin_chat_id:
pass
if message.chat.id == self.admin_chat_id and message.reply_to_message != None:
reply_message = message.reply_to_message.text
id_application = re.search(r"Номер заявки: (\d+)",reply_message).group(1)
id_user = re.search(r"ID пользователя: (\d+)",reply_message).group(1)
current_message = self.check_application(id_application)
print(current_message)
2024-05-21 21:13:08 +03:00
self.bot.send_message(
id_user,
f"Ответ от админа: {message.text}",
reply_to_message_id=current_message
2024-05-21 21:13:08 +03:00
)
else:
check = self.check_user(message.from_user.id)
if not check["status"]:
self.create_user(message)
check = self.check_user(message.from_user.id)
id_message = self.create_message(message,check["info_user"])
self.bot.reply_to(message,"Сообщение отправлено админу")
text = f'''
Номер заявки: {id_message}
ID пользователя: {message.from_user.id}
Сообщение: {message.text}
'''
self.bot.send_message(self.admin_chat_id, text)
2024-05-16 21:15:22 +03:00
2024-05-14 21:21:27 +03:00
self.bot.polling()
TelegramBot(
db_name="tg.db",
token="7037219892:AAGAiUsOL_Ia6lz3iZSyAN0PMlmZFDXbDL0"
2024-05-14 21:21:27 +03:00
)