database manager (school assignment)

3 files changed, 215 insertions, 4 deletions

diff --git a/api/passwords.py b/api/passwords.py
index 011400e..e2ae552 100644
--- a/api/passwords.py
+++ b/api/passwords.py
@@ -7,4 +7,4 @@ def check_password(password, password_hash):
     return bcrypt.checkpw(enc(password), password_hash)
 
 def password_hash(password):
-    return bcrypt.hashpw(enc(password), bcrypt.gensalt());
+    return bcrypt.hashpw(enc(password), bcrypt.gensalt())
diff --git a/database/database.py b/database/database.py
new file mode 100644
index 0000000..30d23bb
--- /dev/null
+++ b/database/database.py
@@ -0,0 +1,210 @@
+import sqlite3
+import bcrypt
+import time
+from random import random, choice
+from math import floor
+from uuid import uuid4
+
+connection = sqlite3.connect("./database.db")
+cursor = connection.cursor()
+
+def random_string():
+    return str(floor(random() * 1e16)) # 1e16 is wetenschappelijke notatie (1 * 10^16)
+
+def repeat_action(action, name):
+    def return_function():
+        count = int(input(f"Hoeveel {name} wil je aanmaken? "))
+        for i in range(count):
+            action()
+            print(f"{i+1}/{count} {name} toegevoegd!")
+        print(f"{count} {name} toegevoegd!")
+    return return_function
+
+# dit is om dit script te organiseren
+programma_acties = []
+class programma_actie:
+    def __init__(self, name, execute):
+        self.name = name
+        self.exec = execute
+
+# initialiseer een nieuwe (lege) database
+def init_db():
+    commands_text = open("./init_db.sql").read()
+    commands = commands_text.strip().split("\n"*2) # python sqlite3 kan maar 1 "statement" per cursor.execute doen, en de statements zijn gescheden door witregels in ./init_db.sql
+    for i, command in enumerate(commands):
+        cursor.execute(command)
+        print(f"SQL commando {i+1} van de {len(commands)} uitgevoerd...")
+    print("Klaar!")
+programma_acties.append(programma_actie("Maak tabellen aan", init_db))
+
+def destroy_tables():
+    for table_name in ["games", "users", "social"]:
+        # format strings + sql is gevaarlijk vanwege sql injectie, maar omdat de invoer hier niet door de gebruiker aangepast kan worden is het semi-ok
+        cursor.execute(f"drop table if exists {table_name}")
+        print(f"Tabel \"{table_name}\" verwijderd!")
+programma_acties.append(programma_actie("Verwijder alle tabellen", destroy_tables))
+
+def user_amount():
+    return len(cursor.execute("select user_id from users").fetchall())
+
+# pak een willekeurige user_id uit de users tabel (gebruikt voor insert_random::games)
+def random_player():
+    return choice(cursor.execute("select user_id from users").fetchall())[0]
+
+# class om alle functies die willekeurige data invoegen te organiseren (python heeft geen namespaces :( )
+class insert_random:
+    # omdat veel kolommen in onze database gebruikersinteracties opslaan laat ik ze hier leeg omdat ik geen tijd hem om gebruikersgedrag te simuleren.
+
+    def __init__(self):
+        pass
+
+    def users(self):
+        user_id = str(uuid4())
+        username = random_string()
+        email = f"{username}@example.com"
+        password_hash = bcrypt.hashpw(random_string().encode("utf-8"), bcrypt.gensalt())
+        registered = int( time.time() * 1000 )
+
+        cursor.execute("insert into users values (?, ?, ?, NULL, ?, ?, \"[]\", FALSE, \"user\", \"{}\", NULL, \"online\") ",
+                (user_id, username, email, password_hash, registered))
+
+    def games(self):
+        if user_amount() < 2:
+            print("Er zijn niet genoeg gebruikers om spellen aan te maken!")
+            return
+
+        game_id = str(uuid4())
+        timestamp = int( time.time() * 1000 )
+
+        cursor.execute("insert into games values (?, NULL, NULL, ?, ?, NULL, ?, NULL, NULL, NULL, FALSE, \"in_progress\", \"default\") ",
+                (game_id, random_player(), random_player(), timestamp))
+
+    def friends(self):
+        users = list(rij[0] for rij in cursor.execute("select user_id from users").fetchall()) # cursor.execute.fetchall() stuurt rijen altijd terug als tuples ookal vraag je 1 kolom op, dus dit zet het om naar een list van strings
+        # maak een nieuwe rij aan voor elke gebruiker in tabel users als deze nog niet bestaat
+        for user in users:
+            if cursor.execute("select user_id from social where user_id = ?", [user]).fetchone(): continue
+            cursor.execute("insert into social values (?, NULL, NULL, NULL)", [user])
+
+programma_acties.append(programma_actie("Maak nepgebruikers aan", repeat_action(insert_random().users, "nepgebruikers")))
+programma_acties.append(programma_actie("Maak voorbeeld spellen aan", repeat_action(insert_random().games, "spellen")))
+programma_acties.append(programma_actie("Vul social tabel met gebruikers", insert_random().friends))
+
+# Verifiëer een user_id of verkrijg een user_id uit een username
+def resolve(resolveable):
+    user_id = None
+    user_id = user_id or cursor.execute("select user_id from users where user_id = ?", [resolveable]).fetchone()
+    user_id = user_id or cursor.execute("select user_id from users where username = ?", [resolveable]).fetchone()
+    # stuur de eerste kolom van de rij terug omdat het weer een tuple is >:(
+    return user_id[0] if user_id else None
+
+def edit_username():
+    resolveable = input("Voer een gebruikers-id of gebruikersnaam in: ")
+    user_id = resolve(resolveable)
+    if not user_id:
+        print("Er is geen gebruiker met gebruikers-id of gebruikersnaam \"{resolveable}\"")
+        return
+    new_username = input("Voer een nieuwe gebruikersnaam in: ")
+    cursor.execute("update users set username = ? where user_id = ?", [new_username, user_id])
+    print("Nieuwe gebruikersnaam succesvol toegepast!")
+programma_acties.append(programma_actie("Pas de gebruikersnaam van een gebruiker aan", edit_username))
+
+def delete_user():
+    resolveable = input("Voer een gebruikers-id of gebruikersnaam in: ")
+    user_id = resolve(resolveable)
+    if not user_id:
+        print("Er is geen gebruiker met gebruikers-id of gebruikersnaam \"{resolveable}\"")
+        return
+    cursor.execute("delete from users where user_id = ?", [user_id])
+    print("Gebruiker verwijderd!")
+programma_acties.append(programma_actie("Verwijder een gebruiker", delete_user))
+
+def mark_game_done():
+    game_id = input("Voer een game-id in om als klaar te markeren: ")
+    if not cursor.execute("select game_id from games where game_id = ?", [game_id]).fetchone():
+        print("Kon geen spel vinden met die game_id!")
+        return
+
+    outcome = "w"
+    starttime = cursor.execute("select timestamp from games where game_id = ?", [game_id]).fetchone()[0]
+    duration = int( time.time() * 1000 ) - starttime
+    cursor.execute("update games set outcome = ?, duration = ?, status = \"finished\" where game_id = ?", [outcome, duration, game_id])
+    print("Spel gemarkeerd als afgelopen!")
+programma_acties.append(programma_actie("Spel markeren als afgelopen", mark_game_done))
+
+def delete_game():
+    game_id = input("Voer een game-id in om te verwijderen: ")
+    if not cursor.execute("select game_id from games where game_id = ?", [game_id]).fetchone():
+        print("Kon geen spel vinden met die game_id!")
+        return
+
+    cursor.execute("delete from games where game_id = ?", [game_id])
+    print("Spel verwijderd!")
+programma_acties.append(programma_actie("Spel verwijderen", delete_game))
+
+# maakt een vriendschap tussen 2 willekeurige user_id's
+def make_friendship():
+    # Dit is misschien de lelijkste functie die ik ooit heb geschreven, wees gewaarschuwd
+    user_1 = random_player()
+    user_2 = random_player()
+    while user_1 == user_2: # garandeer dat user_1 != user_2
+        user_2 = random_player()
+
+    # [v for v in ? if v] zorgt er voor dat lege list elementen worden verwijderd
+    # .fetchone()[0] or "" is omdat deze kolommen ook NULL kunnen zijn, maar ik wil altijd een string
+    # .split(",") is omdat het user_id's zijn die gescheden zijn door komma's
+    user_1_friend_list = [v for v in (cursor.execute("select friends from social where user_id = ?", [user_1]).fetchone()[0] or "").split(",") if v]
+    user_2_friend_list = [v for v in (cursor.execute("select friends from social where user_id = ?", [user_2]).fetchone()[0] or "").split(",") if v]
+
+    user_1_friend_list.append(user_2)
+    user_2_friend_list.append(user_1)
+
+    cursor.execute("update social set friends = ? where user_id = ?", [",".join(user_1_friend_list), user_1])
+    cursor.execute("update social set friends = ? where user_id = ?", [",".join(user_2_friend_list), user_2])
+
+    print(f"Nieuwe vriendschap gemaakt tussen {user_1} en {user_2}")
+programma_acties.append(programma_actie("Vriendschap maken", make_friendship))
+
+def reset_friendlist():
+    resolveable = input("Voer een gebruikers-id of gebruikersnaam in: ")
+    user_id = resolve(resolveable)
+    if not user_id:
+        print("Er is geen gebruiker met gebruikers-id of gebruikersnaam \"{resolveable}\"")
+        return
+    cursor.execute("delete from social where user_id = ?", [user_id])
+    cursor.execute("insert into social values (?, NULL, NULL, NULL)", [user_id])
+    print(f"Vriendenlijst van {user_id} geleegd!")
+programma_acties.append(programma_actie("Vriendenlijst legen", reset_friendlist))
+
+for table in ["users", "games", "social"]:
+    programma_acties.append(programma_actie(
+        f"Tabel '{table}' weergeven",
+        lambda: print(cursor.execute(f"select * from {table}").fetchall())
+        ))
+
+programma_acties.append(programma_actie("Sla alle wijzigingen op", connection.commit))
+programma_acties.append(programma_actie("Stop", exit))
+
+def main():
+    while True:
+        # print witregels
+        print("\n"*2, end="")
+
+        # print alle acties in programma_acties
+        for i, actie in enumerate(programma_acties):
+            print(f"#{i}: {actie.name}")
+
+        # vraag om keuze
+        keuze = int(input("Geef je keuze: "))
+        print("\n", end="")
+
+        # kijk of keuze bestaat
+        if keuze > len(programma_acties) - 1:
+            print("Die actie bestaat niet!")
+            continue # ga door naar de volgende iteratie van de while loop (voer keuze niet uit)
+
+        # voer keuze uit
+        programma_acties[keuze].exec()
+
+if __name__ == "__main__":
+    main()
diff --git a/database/init_db.sql b/database/init_db.sql
index 7603254..1bb36dd 100644
--- a/database/init_db.sql
+++ b/database/init_db.sql
@@ -1,6 +1,6 @@
 PRAGMA foreign_keys = ON;
 
-create table users (
+create table if not exists users (
 	user_id text primary key not null,
 	username varchar(35) not null,
 	email text not null,
@@ -15,7 +15,7 @@ create table users (
 	presence text
 );
 
-create table games (
+create table if not exists games (
 	game_id text primary key not null,
 	parent_game text,
 	moves text,
@@ -28,11 +28,12 @@ create table games (
 	rating_delta_player_2 integer,
 	ranked boolean,
 	status text not null,
+	ruleset text not null,
 	foreign key(player_1_id) references users(user_id),
 	foreign key(player_2_id) references users(user_id)
 );
 
-create table social (
+create table if not exists social (
 	user_id text not null,
 	friends text,
 	blocked text,