Default to current generation.

anybodys · Jun 1, 2024 · faf8b4d · faf8b4d
1 parent 1696955
commit faf8b4d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/votingapi/voting/api.py b/votingapi/voting/api.py
@@ -1,6 +1,6 @@
 from flask import Flask, jsonify, request
 
-from voting.storage import art_storage
+from voting.storage import art_storage, db
 
 
 def create_app():
@@ -19,6 +19,8 @@ def health_check():
 @app.get("/art")
 def art():
   gen = request.args.get('gen', -1)
+  if gen < 0:
+    gen = db.get_current_generation()
 
   # Return a list of all the requested generation's art metadata.
   return jsonify(art_storage.ArtStorage().get_art(gen))
diff --git a/votingapi/voting/storage/db.py b/votingapi/voting/storage/db.py
@@ -0,0 +1,3 @@
+def get_current_generation():
+  # TODO: Look up current gen in the db.
+  return 0