diff --git a/main.py b/main.py index e3971c6..49197d0 100644 --- a/main.py +++ b/main.py @@ -8,12 +8,59 @@ class Database: self.__create_table() def __create_table(self): sql = self.connect_db() + 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) + ) + """) 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} + 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() + print(info_users) + self.close(sql['connect'],sql['cursor']) + if info_users is None: + return {"status":False} + 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']) + def create_message(self,message: telebot.types.Message): + sql = self.connect_db() + date = datetime.datetime.now().strftime("%Y-%m-%d") + # TODO: specify params to query( you can get from check_user) + sql["cursor"].execute(""" + insert into messages (id_user, message_id, message_text, date_send) values (?,?,?,?) + """, ()) + self.close(sql['connect'],sql['cursor']) def close(self,connect,cursor): cursor.close() connect.close() @@ -24,20 +71,28 @@ class TelegramBot(Database): super().__init__(db_name) self.bot = telebot.TeleBot(token) self.router() + def router(self): @self.bot.message_handler(commands=['start']) def start(message): + 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}" self.bot.send_message( message.chat.id, - f"Добро пожаловать, {message.from_user.first_name}" + text ) @self.bot.message_handler(func=lambda message: True) def echo_all(message): - self.bot.reply_to(message,"Не понимаю") - self.bot.delete_message(chat_id=message.chat.id,message_id=message.message_id,timeout=1000) + + self.bot.reply_to(message,"Сообщение отправлено админу") + self.bot.polling() TelegramBot( db_name="tg.db", - token="" + token="7037219892:AAGoZ1bKjN79KMiEGqYCWtPOEOk8s_N0RI4" ) \ No newline at end of file