aboutsummaryrefslogtreecommitdiff
path: root/database
diff options
context:
space:
mode:
authorlonkaars <l.leblansch@gmail.com>2021-04-11 17:50:58 +0200
committerlonkaars <l.leblansch@gmail.com>2021-04-11 17:50:58 +0200
commit28f104de9ae9abe4b42abafbf3865ede5687996c (patch)
tree65e651f09d8fbf81380384692e45803cb4f9d61c /database
parent7b4859059b3bbabf4139ccdf3270a82c094f5d8e (diff)
dprint yapf python formatting
Diffstat (limited to 'database')
-rw-r--r--database/database.py418
1 files changed, 273 insertions, 145 deletions
diff --git a/database/database.py b/database/database.py
index 30d23bb..14f6cb0 100644
--- a/database/database.py
+++ b/database/database.py
@@ -8,203 +8,331 @@ 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)
+ 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
+ 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
+ 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!")
+ 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))
+ 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())
+ 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]
+ 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
+ # 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
+ )
+)
- 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
+# 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
- 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 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 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))
+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!")
-# 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))
+programma_acties.append(
+ programma_actie("Verwijder een gebruiker", delete_user)
+)
-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))
+ 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
+ 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!")
+
- 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)
+ # 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}")
- 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))
+ 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(
+ 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="")
+ 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}")
- # 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="")
- # 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)
- # 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()
- # voer keuze uit
- programma_acties[keuze].exec()
if __name__ == "__main__":
- main()
+ main()