add: operations

src/db_models/BinaryLibrary.py

@@ -1,9 +1,8 @@
-from odmantic import Model
-
-from db_models.embedded_models.Codebase import Codebase
-
+from odmantic import Model, ObjectId
 
 class BinaryLibrary(Model):
-    codebase: Codebase
+    codebase_id: ObjectId
+    type: str="library"
     path: str
-    abstract: str
+    abstract: str
+    md5: str

src/db_models/BinaryTool.py

@@ -1,9 +1,9 @@
-from odmantic import Model
-
-from db_models.embedded_models.Codebase import Codebase
+from odmantic import Model, ObjectId
 
 
 class BinaryTool(Model):
-    codebase: Codebase
+    codebase_id: ObjectId
+    type: str="tool"
     path: str
     abstract: str
+    md5: str

src/db_models/CodeFile.py

@@ -1,13 +1,12 @@
-from odmantic import Model
+from odmantic import Model, ObjectId
 from typing import List
 
 from db_models.embedded_models.CodeSegment import CodeSegment
-from db_models.embedded_models.Codebase import Codebase
 
 
 class CodeFile(Model):
-    codebase: Codebase
-    type: str
+    codebase_id: ObjectId
+    type: str="code"
     path: str
     md5: str
     abstract: str

src/db_models/Codebase.py

@@ -0,0 +1,8 @@
+from odmantic import Model
+
+
+class Codebase(Model):
+    name: str
+    version: str
+    branch: str
+    repo: str

src/db_models/ConfigFile.py

@@ -1,11 +1,9 @@
-from odmantic import Model
-
-from db_models.embedded_models.Codebase import Codebase
+from odmantic import Model, ObjectId
 
 
 class ConfigFile(Model):
-    codebase: Codebase
-    type: str
+    codebase_id: ObjectId
+    type: str="config"
     path: str
     md5: str
     abstract: str

src/db_models/Directory.py

@@ -1,10 +1,9 @@
-from odmantic import Model
+from odmantic import Model, ObjectId
 
-from db_models.embedded_models.Codebase import Codebase
 
 
 class Directory(Model):
-    codebase: Codebase
+    codebase_id: ObjectId
     path: str
     md5: str
     abstract: str

src/db_models/Hotspot.py

@@ -1,9 +1,9 @@
-from odmantic import Model
+from odmantic import Model, ObjectId
 from typing import List
-from db_models.embedded_models.Codebase import Codebase
+from db_models.Codebase import Codebase
 
 
 class Hotspot(Model):
-    codebase: Codebase
+    codebase_id: ObjectId
     topic: str
     links: List[int]

src/db_models/IgnoreFile.py

@@ -1,9 +1,8 @@
-from odmantic import Model
-
-from db_models.embedded_models.Codebase import Codebase
+from odmantic import Model, ObjectId
 
 
 class IgnoreFile(Model):
-    codebase: Codebase
+    codebase_id: ObjectId
+    type:str="ignore"
     path: str
     md5: str

src/db_models/__init__.py

@@ -0,0 +1,6 @@
+from db_models.BinaryTool import BinaryTool
+from db_models.BinaryLibrary import BinaryLibrary
+from db_models.CodeFile import CodeFile
+from db_models.ConfigFile import ConfigFile
+from db_models.IgnoreFile import IgnoreFile
+from db_models.Codebase import Codebase

src/db_models/embedded_models/Codebase.py

@@ -1,9 +0,0 @@
-from odmantic import EmbeddedModel
-
-
-class Codebase(EmbeddedModel):
-    name: str
-    version: str
-    branch: str
-    path: str
-    repo: str

src/db_models/embedded_models/__init__.py

@@ -0,0 +1 @@
+from db_models.embedded_models.CodeSegment import CodeSegment