Browse Source

Use better external interface for populate script

Getty Ritter 4 years ago
parent
commit
4fb84da5e3
1 changed files with 9 additions and 2 deletions
  1. 9 2
      scripts/populate.py

+ 9 - 2
scripts/populate.py

@@ -9,8 +9,15 @@ import lc.request as r
 
 def main():
     lc.config.DB.init("test.db")
-    u, _ = m.User.get_or_create(name="gdritter",)
-    print(u)
+    m.create_tables()
+
+    u = m.User.get_or_none(name="gdritter")
+    if not u:
+        u = m.User.from_request(r.User(
+            name="gdritter",
+            password="behest",
+        ))
+
     with open("scripts/aisamanra.json") as f:
         links = json.load(f)
     for l in links: