UUID to ObjectId conversion in apiserver

Author: p4vook

src/apiserver/api_models.py

@@ -1,5 +1,4 @@
 from __future__ import annotations
-import uuid
 import pydantic
 import typing
 import db_models
@@ -11,17 +10,13 @@ class ApiVersion(pydantic.BaseModel):
 
 
 class Run(pydantic.BaseModel):
-    id: uuid.UUID
-    """
-    Run identifier, unique in the system.
-    """
     toolchain_name: str
     problem_name: str
-    user_id: uuid.UUID
+    user_id: db_models.Id
     contest_name: str
     status: typing.Mapping[str, str] = pydantic.Field(default_factory=dict)
 
     @staticmethod
     def from_db(doc: db_models.RunMainProj) -> Run:
-        return Run(id=doc['id'], toolchain_name=doc['toolchain_name'],
+        return Run(toolchain_name=doc['toolchain_name'],
                    user_id=doc['user_id'], contest_name=doc['contest_name'], problem_name=doc['problem_name'], status=doc['status'])

src/apiserver/db_models.py

@@ -1,10 +1,29 @@
-import uuid
+from bson import ObjectId
 from enum import Enum
-import time
 import typing
 from pydantic import BaseModel, Field
 
 
+class Id(ObjectId):
+    @classmethod
+    def __get_validators__(cls):
+        yield cls.validate
+
+    @classmethod
+    def validate(cls, v):
+        print(type(v))
+        if not isinstance(v, ObjectId):
+            raise TypeError('ObjectId required')
+        return str(v)
+
+    @classmethod
+    def __modify_schema__(cls, schema):
+        schema.update({
+            'Title': 'MongoDB ObjectID',
+            'type': 'string'
+        })
+
+
 class RunPhase(Enum):
     """
     # QUEUED
@@ -24,10 +43,9 @@ class RunPhase(Enum):
 
 
 class RunMainProj(BaseModel):
-    id: uuid.UUID
     toolchain_name: str
     problem_name: str
-    user_id: uuid.UUID
+    user_id: Id
     contest_name: str
     phase: str  # RunPhase
     status: typing.Mapping[str, str] = Field(default_factory=dict)
@@ -37,7 +55,7 @@ class RunMainProj(BaseModel):
     """
 
 
-RunMainProj.FIELDS = ['id', 'toolchain_name',
+RunMainProj.FIELDS = ['toolchain_name',
                       'problem_name', 'user_id', 'contest_name', 'status']
 
 

src/apiserver/routes.py

@@ -1,10 +1,10 @@
 import fastapi
 import db_models
 import api_models
-import uuid
 import typing
 import base64
 import pymongo
+from bson import ObjectId
 import pydantic
 
 
@@ -97,15 +97,15 @@ def route_submit(params: RunSubmitSimpleParams, db: pymongo.database.Database =
         fields of request body; `id` will be real id of this run.
         """
 
-        run_uuid = uuid.uuid4()
-        user_id = uuid.UUID('12345678123456781234567812345678')
-        doc_main = db_models.RunMainProj(id=run_uuid, toolchain_name=params.toolchain,
+        user_id = ObjectId('507f1f77bcf86cd799439011')
+        doc_main = db_models.RunMainProj(toolchain_name=params.toolchain,
                                          problem_name=params.problem, user_id=user_id, contest_name=params.contest, phase=str(db_models.RunPhase.QUEUED))
         doc_source = db_models.RunSourceProj(
             source=base64.b64decode(params.code))
         doc = {**dict(doc_main), **dict(doc_source)}
         db.runs.insert_one(doc)
-        return api_models.Run(id=run_uuid, toolchain_name=params.toolchain, problem_name=params.problem, user_id=user_id, contest_name=params.contest)
+        print(type(user_id))
+        return api_models.Run(toolchain_name=params.toolchain, problem_name=params.problem, user_id=user_id, contest_name=params.contest)
 
     @app.get('/runs', response_model=typing.List[api_models.Run],
              operation_id='listRuns')
@@ -122,7 +122,7 @@ def route_list_runs(db: pymongo.database.Database = fastapi.Depends(db_connect))
         return runs
 
     @app.get('/runs/{run_id}', response_model=api_models.Run, operation_id='getRun')
-    def route_get_run(run_id: uuid.UUID, db: pymongo.database.Database = fastapi.Depends(db_connect)):
+    def route_get_run(run_id: db_models.Id, db: pymongo.database.Database = fastapi.Depends(db_connect)):
         """
         Loads run by id
         """
@@ -139,7 +139,7 @@ def route_get_run(run_id: uuid.UUID, db: pymongo.database.Database = fastapi.Dep
             'description': "Run source is not available"
         }
     })
-    def route_get_run_source(run_id: uuid.UUID, db: pymongo.database.Database = fastapi.Depends(db_connect)):
+    def route_get_run_source(run_id: db_models.Id, db: pymongo.database.Database = fastapi.Depends(db_connect)):
         """
         Returns run source as base64-encoded JSON string
         """
@@ -154,7 +154,7 @@ def route_get_run_source(run_id: uuid.UUID, db: pymongo.database.Database = fast
         return base64.b64encode(doc['source'])
 
     @app.patch('/runs/{run_id}', response_model=api_models.Run, operation_id='patchRun')
-    def route_run_patch(run_id: uuid.UUID, patch: RunPatch, db: pymongo.database.Database = fastapi.Depends(db_connect)):
+    def route_run_patch(run_id: db_models.Id, patch: RunPatch, db: pymongo.database.Database = fastapi.Depends(db_connect)):
         """
         Modifies existing run