prompt: before_first_request: After creating the table, if the table …

…is empty, add three rows
sturdy-dev · Aug 24, 2022 · 1799234 · 1799234 · vercel · Aug 24, 2022
1 parent 6ca4b98
commit 1799234
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/app.py b/app.py
@@ -15,6 +15,15 @@ def create_tables():
         id INTEGER PRIMARY KEY AUTOINCREMENT,
         description text,
         done bool
+    c.execute("SELECT * FROM tasks")
+    tasks = c.fetchall()
+    if len(tasks) == 0:
+        c.execute("INSERT INTO tasks (description, done) VALUES (:description, :done)",
+                  {"description": "Task 1", "done": 0})
+        c.execute("INSERT INTO tasks (description, done) VALUES (:description, :done)",
+                  {"description": "Task 2", "done": 0})
+        c.execute("INSERT INTO tasks (description, done) VALUES (:description, :done)",
+                  {"description": "Task 3", "done": 0})
     )""")
     conn.commit()
     conn.close()