Browse Source

Proactively create non-existent games

Getty Ritter 4 years ago
parent
commit
7c6ab3b6dd
1 changed files with 8 additions and 1 deletions
  1. 8 1
      parley.py

+ 8 - 1
parley.py

@@ -18,7 +18,14 @@ def get_game_id(game: str, c=None) -> int:
     if c is None:
         c = DB.cursor()
     c.execute('SELECT id FROM games WHERE name = ?', [game])
-    return c.fetchone()[0]
+    res = c.fetchone()
+    if res is not None:
+        return res[0]
+    else:
+        c.execute('INSERT INTO games (name) VALUES (?)', [game])
+        DB.commit()
+        c.execute('SELECT id FROM games WHERE name = ?', [game])
+        return c.fetchone()[0]
 
 def add_msg(user: str, content: str, game: str, c=None):
     if c is None: