79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
|
import sqlite3
|
||
|
|
|
||
|
|
|
||
|
|
def sqlite_connect():
|
||
|
|
conn = sqlite3.connect("db/database.db", check_same_thread=False)
|
||
|
|
conn.execute("pragma journal_mode=wal;")
|
||
|
|
return conn
|
||
|
|
|
||
|
|
|
||
|
|
def init_sqlite():
|
||
|
|
conn = sqlite_connect()
|
||
|
|
cursor = conn.cursor()
|
||
|
|
try:
|
||
|
|
# Начинаем транзакцию
|
||
|
|
cursor.execute('BEGIN')
|
||
|
|
# Выполняем операции
|
||
|
|
cursor.execute('''CREATE TABLE IF NOT EXISTS users (
|
||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
|
|
tg_id INTEGER NOT NULL UNIQUE ,
|
||
|
|
school_nik TEXT NOT NULL UNIQUE,
|
||
|
|
username TEXT UNIQUE,
|
||
|
|
number_quest INTEGER,
|
||
|
|
city TEXT)''')
|
||
|
|
|
||
|
|
cursor.execute('''CREATE INDEX IF NOT EXISTS usrid ON users (number_quest)''')
|
||
|
|
|
||
|
|
cursor.execute('''CREATE TABLE IF NOT EXISTS flags (
|
||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
|
|
flags TEXT NOT NULL UNIQUE)''')
|
||
|
|
|
||
|
|
cursor.execute('''CREATE TABLE IF NOT EXISTS token (
|
||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
|
|
token TEXT NOT NULL UNIQUE)''')
|
||
|
|
cursor.execute('COMMIT')
|
||
|
|
except Exception as e:
|
||
|
|
print(e)
|
||
|
|
cursor.execute('ROLLBACK')
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
def registration(user_id, school_nik, uname, number_quest, city, class_name, end_14):
|
||
|
|
conn = sqlite_connect()
|
||
|
|
cursor = conn.cursor()
|
||
|
|
try:
|
||
|
|
cursor.execute('BEGIN')
|
||
|
|
cursor.execute('INSERT INTO users (tg_id, school_nik, username, number_quest, city, class_name, end_14)'
|
||
|
|
' VALUES (?, ?, ?, ?, ?, ?, ?)',
|
||
|
|
(user_id, school_nik, uname, number_quest, city, class_name, end_14))
|
||
|
|
cursor.execute('COMMIT')
|
||
|
|
except Exception as e:
|
||
|
|
print(e)
|
||
|
|
cursor.execute('ROLLBACK')
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
|
||
|
|
def quest_1(tg_id, date_begin):
|
||
|
|
conn = sqlite_connect()
|
||
|
|
cursor = conn.cursor()
|
||
|
|
try:
|
||
|
|
cursor.execute('BEGIN')
|
||
|
|
cursor.execute(f'UPDATE users SET date_begin=(?) WHERE tg_id=(?)', (date_begin, tg_id,))
|
||
|
|
cursor.execute('COMMIT')
|
||
|
|
except Exception as e:
|
||
|
|
print(e)
|
||
|
|
cursor.execute('ROLLBACK')
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
def get_time(username):
|
||
|
|
conn = sqlite_connect()
|
||
|
|
cursor = conn.cursor()
|
||
|
|
try:
|
||
|
|
cursor.execute('BEGIN')
|
||
|
|
time = cursor.execute('SELECT date_begin, end_13 FROM users WHERE school_nik = (?)', (username,)).fetchone()
|
||
|
|
return time
|
||
|
|
except Exception as e:
|
||
|
|
print(e)
|
||
|
|
cursor.execute('ROLLBACK')
|
||
|
|
conn.close()
|
||
|
|
|