Fix /v0/cases handler and corresponding database function

app/persistence/db.py

@@ -19,12 +19,7 @@ def get_cases(
         conditions["lat"] = lat
     if lon is not None:
         conditions["lon"] = lon
-    last_case = mongo.db.cases.find_one({"uuid": uuid})
-    if last_case is None:
-        conditions["upload_timestamp"] = {"$gte": random_time_in_the_past()}
-    else:
-        conditions["upload_timestamp"] = {"$gte": last_case["upload_timestamp"]}
-    return (case for case in mongo.db.cases.find(conditions))
+    return list(case for case in mongo.db.cases.find(conditions, {"_id": False}))
 
 
 def random_time_in_the_past() -> datetime:

app/routes/v0/__init__.py

@@ -3,6 +3,7 @@
 from typing import Union, Optional, Generator
 from app.persistence.db import get_cases, insert_random_cases
 from app.model import ApiError
+import json
 
 cases = Blueprint("v0.cases", __name__, url_prefix="/v0/cases")
 
@@ -26,12 +27,12 @@ def index() -> Response:
 
     cases = get_cases(uuid, lat=lat, lon=lon)
 
-    def generate() -> Generator[str, None, None]:
-        for case in cases:
-            case_uuid = str(case["uuid"])
-            yield case_uuid + ","
+    # def generate() -> Generator[str, None, None]:
+    #    for case in cases:
+    #        case_uuid = str(case["uuid"])
+    #        yield case_uuid + ","
 
-    return Response(generate(), mimetype="application/octet-stream")
+    return Response(json.dumps(cases), mimetype="application/octet-stream")
 
 
 @cases.route("/insert/<int:n>", methods=["POST"])

pyproject.toml

@@ -9,6 +9,7 @@ python = "^3.8"
 Flask = "^1.1.2"
 Flask-PyMongo = "^2.3.0"
 python-dateutil = "^2.8.1"
+flask_pymongo = "^2.3.0"
 
 [tool.poetry.dev-dependencies]
 black = "^19.10b0"