feat: added support for bucket data object and updating bucket metadata (#128)

ErikBjare · web-flow · commit 8fecb9856dbe · 2023-05-15T10:06:41.000+02:00
diff --git a/aw_server/api.py b/aw_server/api.py
@@ -139,6 +139,7 @@ def create_bucket(
         client: str,
         hostname: str,
         created: Optional[datetime] = None,
+        data: Optional[Dict[str, Any]] = None,
     ) -> bool:
         """
         Create bucket.
@@ -154,9 +155,29 @@ def create_bucket(
             client=client,
             hostname=hostname,
             created=created,
+            data=data,
         )
         return True
 
+    @check_bucket_exists
+    def update_bucket(
+        self,
+        bucket_id: str,
+        event_type: Optional[str] = None,
+        client: Optional[str] = None,
+        hostname: Optional[str] = None,
+        data: Optional[Dict[str, Any]] = None,
+    ) -> None:
+        """Update bucket metadata"""
+        self.db.update_bucket(
+            bucket_id,
+            type=event_type,
+            client=client,
+            hostname=hostname,
+            data=data,
+        )
+        return None
+
     @check_bucket_exists
     def delete_bucket(self, bucket_id: str) -> None:
         """Delete a bucket"""
diff --git a/aw_server/rest.py b/aw_server/rest.py
@@ -104,6 +104,16 @@ def format(self, value):
     },
 )
 
+update_bucket = api.model(
+    "UpdateBucket",
+    {
+        "client": fields.String(required=False),
+        "type": fields.String(required=False),
+        "hostname": fields.String(required=False),
+        "data": fields.String(required=False),
+    },
+)
+
 query = api.model(
     "Query",
     {
@@ -173,6 +183,19 @@ def post(self, bucket_id):
         else:
             return {}, 304
 
+    @api.expect(update_bucket)
+    @copy_doc(ServerAPI.update_bucket)
+    def put(self, bucket_id):
+        data = request.get_json()
+        current_app.api.update_bucket(
+            bucket_id,
+            event_type=data["type"],
+            client=data["client"],
+            hostname=data["hostname"],
+            data=data["data"],
+        )
+        return {}, 200
+
     @copy_doc(ServerAPI.delete_bucket)
     @api.param("force", "Needs to be =1 to delete a bucket it non-testing mode")
     def delete(self, bucket_id):
diff --git a/poetry.lock b/poetry.lock
