Updated datamodel.py main.py model.py

Author: GorkiPower

backend/datamodel.py

@@ -20,3 +20,9 @@ class SchemaDefinition(BaseModel):
         None,
         description="Timestamp of the last update (optional)"
     )
+
+
+
+class UpdateSchema(BaseModel):
+    name: str | None = None
+    version: str | None = None

backend/main.py

@@ -1,16 +1,14 @@
-from datetime import datetime
 
+from datetime import datetime
 from bson import ObjectId
 from database import schemas_collection
-from datamodel import SchemaDefinition
+from datamodel import SchemaDefinition, UpdateSchema  # ✅ Updated import
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
-from model import UpdateSchema
-
-# https://github.com/BAMresearch/jsoned/tree/main/backend
 
 app = FastAPI()
 
+# Enable CORS for all origins
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -19,27 +17,49 @@
     allow_headers=["*"],
 )
 
-
-# Get all schemas
 @app.get("/schemas")
 async def get_all_schemas():
+    """
+    Retrieve all schemas from the database.
+
+    Returns:
+        list: A list of schema documents with stringified IDs.
+    """
     schemas = list(schemas_collection.find())
     for s in schemas:
         s["_id"] = str(s["_id"])
     return schemas
 
-
-# Add new schema
 @app.post("/schemas")
 async def add_schema(schema: SchemaDefinition):
+    """
+    Add a new schema to the database.
+
+    Args:
+        schema (SchemaDefinition): The schema data to insert.
+
+    Returns:
+        dict: The ID of the inserted schema.
+    """
     schema.updated_at = datetime.utcnow()
     result = schemas_collection.insert_one(schema.dict())
     return {"id": str(result.inserted_id)}
 
-
-# Update schema (PUT)
 @app.put("/schemas/{id}")
 async def update_schema(id: str, update: UpdateSchema):
+    """
+    Update an existing schema by ID.
+
+    Args:
+        id (str): The schema ID.
+        update (UpdateSchema): Fields to update.
+
+    Raises:
+        HTTPException: If the schema is not found.
+
+    Returns:
+        dict: Success message.
+    """
     result = schemas_collection.update_one(
         {"_id": ObjectId(id)},
         {"$set": {k: v for k, v in update.dict().items() if v is not None}},
@@ -48,10 +68,20 @@ async def update_schema(id: str, update: UpdateSchema):
         raise HTTPException(status_code=404, detail="Schema not found")
     return {"message": "Schema updated"}
 
-
-# Delete schema
 @app.delete("/schemas/{id}")
 async def delete_schema(id: str):
+    """
+    Delete a schema by ID.
+
+    Args:
+        id (str): The schema ID.
+
+    Raises:
+        HTTPException: If the schema is not found.
+
+    Returns:
+        dict: Success message.
+    """
     result = schemas_collection.delete_one({"_id": ObjectId(id)})
     if result.deleted_count == 0:
         raise HTTPException(status_code=404, detail="Schema not found")

backend/model.py

@@ -13,9 +13,15 @@
 ├── services/        # Business logic
 ├── utils/           # Helpers
 """
+from datetime import datetime
+from pydantic import BaseModel, Field
 
-from pydantic import BaseModel
-
+class SchemaDefinition(BaseModel):
+    id: str = Field(..., description="Unique identifier for the schema")
+    name: str = Field(..., description="Human-readable name of the schema", min_length=3)
+    version: str = Field("1.0.0", description="Version of the schema")
+    content: dict = Field(..., description="The actual schema content as a dictionary")
+    updated_at: datetime | None = Field(None, description="Timestamp of the last update (optional)")
 
 class UpdateSchema(BaseModel):
     name: str | None = None