35 lines
1014 B
Python
35 lines
1014 B
Python
import sqlite3
|
|
|
|
def init_db(path: str) -> None:
|
|
with sqlite3.connect(path) as connection:
|
|
cursor = connection.cursor()
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS Tokens (
|
|
TOKEN TEXT PRIMARY KEY NOT NULL,
|
|
USERID INTEGER NOT NULL,
|
|
NAME TEXT NOT NULL
|
|
)
|
|
''')
|
|
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS Domains (
|
|
DOMAIN TEXT PRIMARY KEY NOT NULL,
|
|
USERID INTEGER NOT NULL
|
|
)
|
|
''')
|
|
|
|
connection.commit()
|
|
cursor.close()
|
|
|
|
def check_token_domain(db_path: str, token: str, domain: str) -> bool:
|
|
with sqlite3.connect(db_path) as connection:
|
|
cursor = connection.cursor()
|
|
cursor.execute('''
|
|
SELECT COUNT(*) FROM Tokens
|
|
JOIN Domains ON Tokens.userid = Domains.userid
|
|
WHERE token = ?
|
|
AND domain = ?
|
|
''', (token,domain))
|
|
|
|
return cursor.fetchone()[0] == 1
|