Created database/prepare.py
This commit is contained in:
parent
1bde1a1739
commit
44bab0fb48
39
src/database/prepare.py
Normal file
39
src/database/prepare.py
Normal file
@ -0,0 +1,39 @@
|
||||
import mariadb
|
||||
|
||||
|
||||
def _create_tables(con: mariadb.Connection) -> None:
|
||||
cursor = con.cursor()
|
||||
cursor.execute(
|
||||
"""CREATE TABLE IF NOT EXISTS master_pass (user_id INT,
|
||||
salt BINARY(64),
|
||||
PASSWD BINARY(64),
|
||||
PRIMARY KEY(user_id)
|
||||
)"""
|
||||
)
|
||||
cursor.execute(
|
||||
"""CREATE TABLE IF NOT EXISTS accounts(user_id INT,
|
||||
acc_name VARCHAR(255),
|
||||
salt BINARY(64),
|
||||
enc_login BINARY(64),
|
||||
enc_pass BINARY(64),
|
||||
UNIQUE(acc_name, user_id)
|
||||
)"""
|
||||
)
|
||||
cursor.close()
|
||||
|
||||
|
||||
def _create_index(con: mariadb.Connection) -> None:
|
||||
cursor = con.cursor()
|
||||
cursor.execute(
|
||||
"""CREATE INDEX IF NOT EXISTS user_id_to_acc on accounts(user_id)
|
||||
"""
|
||||
)
|
||||
|
||||
|
||||
def prepare(host: int, user: str, passwd: str, database: str) -> None:
|
||||
con: mariadb.Connection = mariadb.connect(
|
||||
host=host, user=user, password=passwd, database=database
|
||||
)
|
||||
_create_tables(con)
|
||||
_create_index(con)
|
||||
con.close()
|
Reference in New Issue
Block a user