1 files changed, 160 insertions, 0 deletions
diff --git a/archinstall/lib/models/gen.py b/archinstall/lib/models/gen.py
new file mode 100644
index 00000000..fb7e5751
--- /dev/null
+++ b/archinstall/lib/models/gen.py
@@ -0,0 +1,160 @@
+from dataclasses import dataclass
+from typing import Optional, List, Dict, Any
+
+
+@dataclass
+class VersionDef:
+	version_string: str
+
+	@classmethod
+	def parse_version(cls) -> List[str]:
+		if '.' in cls.version_string:
+			versions = cls.version_string.split('.')
+		else:
+			versions = [cls.version_string]
+
+		return versions
+
+	@classmethod
+	def major(self) -> str:
+		return self.parse_version()[0]
+
+	@classmethod
+	def minor(cls) -> Optional[str]:
+		versions = cls.parse_version()
+		if len(versions) >= 2:
+			return versions[1]
+
+		return None
+
+	@classmethod
+	def patch(cls) -> Optional[str]:
+		versions = cls.parse_version()
+		if '-' in versions[-1]:
+			_, patch_version = versions[-1].split('-', 1)
+			return patch_version
+
+		return None
+
+	def __eq__(self, other) -> bool:
+		if other.major == self.major and \
+			other.minor == self.minor and \
+			other.patch == self.patch:
+
+			return True
+		return False
+
+	def __lt__(self, other) -> bool:
+		if self.major() > other.major():
+			return False
+		elif self.minor() and other.minor() and self.minor() > other.minor():
+			return False
+		elif self.patch() and other.patch() and self.patch() > other.patch():
+			return False
+
+		return True
+
+	def __str__(self) -> str:
+		return self.version_string
+
+
+@dataclass
+class PackageSearchResult:
+	pkgname: str
+	pkgbase: str
+	repo: str
+	arch: str
+	pkgver: str
+	pkgrel: str
+	epoch: int
+	pkgdesc: str
+	url: str
+	filename: str
+	compressed_size: int
+	installed_size: int
+	build_date: str
+	last_update: str
+	flag_date: Optional[str]
+	maintainers: List[str]
+	packager: str
+	groups: List[str]
+	licenses: List[str]
+	conflicts: List[str]
+	provides: List[str]
+	replaces: List[str]
+	depends: List[str]
+	optdepends: List[str]
+	makedepends: List[str]
+	checkdepends: List[str]
+
+	@staticmethod
+	def from_json(data: Dict[str, Any]) -> 'PackageSearchResult':
+		return PackageSearchResult(**data)
+
+	@property
+	def pkg_version(self) -> str:
+		return self.pkgver
+
+	def __eq__(self, other) -> bool:
+		return self.pkg_version == other.pkg_version
+
+	def __lt__(self, other) -> bool:
+		return self.pkg_version < other.pkg_version
+
+
+@dataclass
+class PackageSearch:
+	version: int
+	limit: int
+	valid: bool
+	num_pages: int
+	page: int
+	results: List[PackageSearchResult]
+
+	@staticmethod
+	def from_json(data: Dict[str, Any]) -> 'PackageSearch':
+		results = [PackageSearchResult.from_json(r) for r in data['results']]
+
+		return PackageSearch(
+			version=data['version'],
+			limit=data['limit'],
+			valid=data['valid'],
+			num_pages=data['num_pages'],
+			page=data['page'],
+			results=results
+		)
+
+
+@dataclass
+class LocalPackage:
+	name: str
+	version: str
+	description:str
+	architecture: str
+	url: str
+	licenses: str
+	groups: str
+	depends_on: str
+	optional_deps: str
+	required_by: str
+	optional_for: str
+	conflicts_with: str
+	replaces: str
+	installed_size: str
+	packager: str
+	build_date: str
+	install_date: str
+	install_reason: str
+	install_script: str
+	validated_by: str
+	provides: str
+
+	@property
+	def pkg_version(self) -> str:
+		return self.version
+
+	def __eq__(self, other) -> bool:
+		return self.pkg_version == other.pkg_version
+
+	def __lt__(self, other) -> bool:
+		return self.pkg_version < other.pkg_version