feat(backend/voe.py): adding create voe endpoint

frbvoe/backend/voe.py

@@ -1,8 +1,52 @@
 """VOEvent Server Blueprint."""
 
-# Post at /voe
+from sanic import Blueprint
+from sanic.log import logger
+from sanic.request import Request
+from sanic.response import HTTPResponse
+from sanic.response import json as json_response
+from sanic_ext import openapi
+from pymongo.errors import PyMongoError
+
+from frbvoe.models.voe import VOEvent
+from frbvoe.utilities.senders import send_to_comet
+
+voe = Blueprint("voe", url_prefix="/")
 
+
+# Post at /voe
+@voe.post("voe")
+@openapi.response(201, description="Validates data from the telescope.")
 # Add the validated payload to the MongoDB Database
-# Database Name: frbvoe
-# Collection Name: voe
-# Document -> VOEvent Payload Dict.
+async def create_voe(request: Request, voe_event: VOEvent):
+
+    mongo = request.app.ctx.mongo
+    request.app.add_task(send_to_comet(voe_event.model_dump()))
+    # Database Name: frbvoe
+    # Collection Name: voe
+    # Document -> VOEvent Payload Dict.
+    try:
+        insert_result = await mongo["frbvoe"]["voe"].insert_one(voe_event.model_dump())
+    except (Exception, PyMongoError) as mongo_error:
+        logger.error(f"{mongo_error} on /voe")
+        return json_response({"message": mongo_error}, status=500)
+
+    return json_response({"id": insert_result.inserted_id}, status=201)
+
+
+async def delete_voe(request: Request):
+
+    # /voe?id=<id>
+    if "id" not in request.args.keys():
+        return json_response(
+            {"message": "Your request needs an ID."}, status=400
+        )
+    id = request.args["id"]
+    mongo = request.app.ctx.mongo
+    try:
+        delete_result = await mongo["frbvoe"]["voe"].delete_one({"id": id})
+    except (Exception, PyMongoError) as mongo_error:
+        logger.error(f"{mongo_error} on /voe")
+        return json_response({"message": mongo_error}, status=500)
+
+    return json_response({"message": delete_result.delete_count}, status=202)

frbvoe/server.py

@@ -10,6 +10,7 @@
 from sanic.worker.loader import AppLoader
 
 from frbvoe.models.voe import VOEvent
+from frbvoe.backend.voe import voe as voe_blueprint
 
 
 async def mongo(app: Sanic, loop: AbstractEventLoop) -> None:
@@ -70,7 +71,11 @@ def create(name: str = "frbvoe", debug: bool = False) -> Sanic:
     app.config.HEALTH = True
     app.config.HEALTH_ENDPOINT = True
     app.ctx.debug = debug
-    app.config.FALLBACK_ERROR_FORMAT = "json"
+    app.config.FALL
+    BACK_ERROR_FORMAT = "json"
+    # ? Blueprints
+    app.blueprint(voe_blueprint)
+    # ? Listeners
     app.register_listener(mongo, "before_server_start")
     return app
 

frbvoe/utilities/senders.py

@@ -0,0 +1,8 @@
+"""Function to send request to Comet."""
+
+import requests
+
+async def send_to_comet(payload) -> bool:
+    response = requests.post("http://comet:8098/", json=payload)
+    # ? Maybe you'll need to change this
+    return response.status_code == 200