From 3d69805765a8d7a755ee911ff39885e2c4c30488 Mon Sep 17 00:00:00 2001
From: admin <admin@example.com>
Date: Thu, 16 May 2024 21:15:22 +0300
Subject: [PATCH] save users to db

---
 main.py | 63 +++++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 59 insertions(+), 4 deletions(-)

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