WfJJ4dD3

1import mysql.connector
2
3
4# Connect to db
5def post_sql_query(sql_query, data=None):
6    with mysql.connector.connect(host="127.0.0.1", user="root", password="pass", database="bot_db") as connection:
7        cursor = connection.cursor()
8
9        if data:
10            cursor.execute(sql_query, data, multi=True)
11        else:
12            cursor.execute(sql_query, multi=True)
13        result = cursor.fetchall()
14
15        # save data and close connection
16        connection.commit()
17        cursor.close()
18        connection.close()
19
20        return result
21
22
23def create_users_table():
24    req = """CREATE TABLE IF NOT EXISTS users(
25            id              INTEGER AUTO_INCREMENT PRIMARY KEY,
26            user_id         INTEGER NOT NULL,
27            full_name       VARCHAR(255),
28            user_name       VARCHAR(255),
29            lang            VARCHAR(2) DEFAULT 'ru'
30    )"""
31    post_sql_query(req)
32
33
34# reg user
35def reg_user(user, name, user_name):
36    user_check_query = f'SELECT * FROM users WHERE user_id = %s'
37    data = (user,)
38    user_check_data = post_sql_query(user_check_query, data)
39    if not user_check_data:
40        req = "INSERT INTO users (user_id, full_name, user_name) VALUES (%s, %s, %s)"
41        data = (user, name, user_name,)
42        post_sql_query(req, data)
43