Created database/prepare.py
This commit is contained in:
		
							
								
								
									
										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