refactor: move index to pydantic models

1 files changed, 46 insertions, 0 deletions

diff --git a/meta/model/index.py b/meta/model/index.py
new file mode 100644
index 0000000000..f7cdc36e0f
--- /dev/null
+++ b/meta/model/index.py
@@ -0,0 +1,46 @@
+from datetime import datetime
+from typing import Optional, List
+
+from pydantic import Field
+
+from meta.model import Dependency, MetaBase, Versioned, MetaVersion
+
+
+class MetaVersionIndexEntry(MetaBase):
+    version: str
+    type: Optional[str]
+    release_time: datetime = Field(alias="releaseTime")
+    requires: Optional[List[Dependency]]
+    conflicts: Optional[List[Dependency]]
+    recommended: Optional[bool]
+    volatile: Optional[bool]
+    sha256: str
+
+    @classmethod
+    def from_meta_version(cls, v: MetaVersion, recommended: bool, sha256: str):
+        return cls(
+            version=v.version,
+            type=v.type,
+            release_time=v.release_time,
+            requires=v.requires,
+            conflicts=v.conflicts,
+            recommended=recommended,
+            volatile=v.volatile,
+            sha256=sha256
+        )
+
+
+class MetaVersionIndex(Versioned):
+    name: str
+    uid: str
+    versions: List[MetaVersionIndexEntry] = Field([])
+
+
+class MetaPackageIndexEntry(MetaBase):
+    name: str
+    uid: str
+    sha256: str
+
+
+class MetaPackageIndex(Versioned):
+    packages: List[MetaPackageIndexEntry] = Field([])