fix: [API] Fixed an issue when storing the UUID of a scan.

api/crud.py

@@ -31,9 +31,9 @@ def create_item(db: Session, item: schemas.ItemCreate):
     return db_item
 
 
-def create_tst(db: Session, data: schemas.ItemCreate):
+def create_tst(db: Session, data: schemas.TimeStampTokenCreate):
     """Create a TimeStampToken."""
-    db_tst = models.TimeStampToken(tst=data)
+    db_tst = models.TimeStampToken(**data)
     db.add(db_tst)
     db.commit()
     db.refresh(db_tst)

api/main.py

@@ -6,7 +6,6 @@
 from fastapi import Depends
 from fastapi import FastAPI
 from fastapi import HTTPException
-from fastapi import Request
 from fastapi.openapi.utils import get_openapi
 from sqlalchemy.orm import Session
 
@@ -96,19 +95,25 @@ async def create_item(item: schemas.ScanDataCreate, db: Session = db_session):
 #
 
 
-@app.post("/tst/")
-async def create_tst(request: Request, db: Session = db_session):
+@app.post("/tst/", response_model=schemas.TimeStampTokenCreate)
+async def create_tst(
+    data: schemas.TimeStampTokenCreate, db: Session = db_session
+) -> schemas.TimeStampToken:
     """Insert a TimeStampToken."""
-    data: bytes = await request.body()
-    new_tst = crud.create_tst(db=db, data=data)
-    return {"message": new_tst.id}
+    dict_data = {
+        "scan_uuid": data.scan_uuid,
+        "tst": data.tst
+    }
+    new_tst = crud.create_tst(db=db, data=dict_data)
+    return new_tst
 
 
-@app.get("/tsts/", response_model=str)
-async def read_tst(skip: int = 0, limit: int = 100, db: Session = db_session) -> str:
+@app.get("/tsts/", response_model=list[schemas.TimeStampToken])
+async def read_tst(
+    skip: int = 0, limit: int = 100, db: Session = db_session
+) -> list[schemas.TimeStampToken]:
     tsts = crud.get_tst(db, skip=skip, limit=limit)
-    tst_list = [elem.tst for elem in tsts]
-    return str(tst_list)
+    return tsts
 
 
 #

api/models.py

@@ -1,4 +1,5 @@
 import uuid
+
 from sqlalchemy import Column
 from sqlalchemy import Integer
 from sqlalchemy import LargeBinary

api/schemas.py

@@ -1,3 +1,5 @@
+import uuid
+
 from pydantic import BaseModel
 
 
@@ -36,5 +38,15 @@ class ScanDataCreate(ScanData):
     pass
 
 
+#
+# Timestamp
+#
+
+
 class TimeStampToken(BaseModel):
+    scan_uuid: uuid.UUID
     tst: bytes
+
+
+class TimeStampTokenCreate(TimeStampToken):
+    pass

scandale/probe.py

@@ -4,6 +4,7 @@
 import json
 import subprocess
 import time
+import uuid
 
 import spade
 from spade.agent import Agent
@@ -51,12 +52,13 @@ async def run(self):
             )  # Set the language of the message content
 
             # Set the message content
+            scan_uuid = str(uuid.uuid4())
             msg.body = json.dumps(
                 {
                     "version": self.config["version"],
                     "format": self.config["format"],
                     "meta": {
-                        "uuid": self.config["uuid"],
+                        "uuid": scan_uuid,
                         "ts": int(time.time()),
                         "type": self.config["type"],
                     },