35 lines
980 B
Python
35 lines
980 B
Python
|
import sqlite3
|
||
|
|
|
||
|
|
#def __init__(self, path: str) -> None:
|
||
|
|
# self.db_connection = sqlite3.connect(path)
|
||
|
|
# cursor = self.db_connection.cursor()
|
||
|
|
# cursor.execute('''
|
||
|
|
# CREATE TABLE IF NOT EXISTS Tokens (
|
||
|
|
# TOKEN TEXT PRIMARY KEY NOT NULL,
|
||
|
|
# USER TEXT NOT NULL,
|
||
|
|
# NAME TEXT NOT NULL
|
||
|
|
# )
|
||
|
|
# ''')
|
||
|
|
#
|
||
|
|
# cursor.execute('''
|
||
|
|
# CREATE TABLE IF NOT EXISTS Domains (
|
||
|
|
# DOMAIN TEXT PRIMARY KEY NOT NULL,
|
||
|
|
# USER TEXT NOT NULL
|
||
|
|
# )
|
||
|
|
# ''')
|
||
|
|
#
|
||
|
|
# self.db_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.user = Domains.user
|
||
|
|
WHERE token = ?
|
||
|
|
AND domain = ?
|
||
|
|
''', (token,domain))
|
||
|
|
|
||
|
|
return cursor.fetchone()[0] == 1
|