UzYfvVHi

1# -*- coding: utf-8 -*-
2import telebot
3import telebot
4import sqlite3
5from sqlite3 import Error
6from time import sleep, ctime
7
8
9def post_sql_query(sql_query):
10    with sqlite3.connect('my.db') as connection:
11        cursor = connection.cursor()
12        try:
13            cursor.execute(sql_query)
14        except Error:
15            pass
16        result = cursor.fetchall()
17        return result
18
19
20def create_tables():
21    users_query = '''CREATE TABLE IF NOT EXISTS USERS 
22                        (user_id INTEGER PRIMARY KEY NOT NULL,
23                        username TEXT,
24                        first_name TEXT,
25                        last_name TEXT,
26                        reg_date TEXT);'''
27    post_sql_query(users_query)
28
29
30def register_user(user, username, first_name, last_name):
31    user_check_query = f'SELECT * FROM USERS WHERE user_id = {user};'
32    user_check_data = post_sql_query(user_check_query)
33    if not user_check_data:
34        insert_to_db_query = f'INSERT INTO USERS (user_id, username, first_name,  last_name, reg_date) VALUES ({user}, "{username}", "{first_name}", "{last_name}", "{ctime()}");'
35        post_sql_query(insert_to_db_query)
36
37
38bot = telebot.TeleBot(token_test)
39
40create_tables()  # вызываем функцию создания таблицы users
41
42
43@bot.message_handler(commands=['start'])
44def start(message):
45    register_user(message.from_user.id, message.from_user.username,
46                  message.from_user.first_name, message.from_user.last_name)
47    bot.send_message(message.from_user.id, f'Welcome  {message.from_user.first_name}')
48
49
50if __name__ == "__main__":
51    try:
52        bot.polling(none_stop=True)
53    except Exception as e:
54        print(e)