api: rewrite to use async

1 files changed, 13 insertions, 5 deletions

diff --git a/api.py b/api.py
index 2f96704..c679df9 100644
--- a/api.py
+++ b/api.py
@@ -1,10 +1,18 @@
 #!/usr/bin/python
 
+import asyncio
 import database
 import crawler
 
-db = database.Database("dbname=vgstats user=vgstats")
-api = crawler.Crawler()
-#db.upsert("na", api.matches(params={"filter[playerNames]": "fallingcomets"}), True)
-db.upsert("na", api.matches(), True)
-print(db.select("SELECT data->'attributes'->>'name' as name FROM player WHERE CAST(data->'attributes'->'stats'->>'winStreak' AS integer) > 10"))
+
+async def main():
+    db = database.Database()
+    await db.connect("postgres://vgstats@localhost/vgstats")
+    api = crawler.Crawler()
+    matches = await api.matches()
+    await db.upsert("na", matches, True)
+    print(await db.select("SELECT data->'attributes'->>'name' as name FROM player WHERE CAST(data->'attributes'->'stats'->>'winStreak' AS integer) > 3"))
+
+loop = asyncio.get_event_loop()
+loop.run_until_complete(main())
+loop.close()