gem selector sketch

bauh/gems/arch/model.py

@@ -0,0 +1,109 @@
+import datetime
+
+from bauh.api.abstract.model import SoftwarePackage
+from bauh.api.constants import CACHE_PATH
+from bauh.commons import resource
+
+from bauh.gems.arch import ROOT_DIR
+
+
+class ArchPackage(SoftwarePackage):
+
+    def __init__(self, name: str = None, version: str = None, latest_version: str = None, description: str = None,
+                 package_base: str = None, votes: int = None, popularity: float = None,
+                 first_submitted: datetime.datetime = None, last_modified: datetime.datetime = None,
+                 maintainer: str = None, url_download: str = None, pkgbuild: str = None, mirror: str = None,
+                 desktop_entry: str = None, installed: bool = False):
+
+        super(ArchPackage, self).__init__(name=name, version=version, latest_version=latest_version, description=description, installed=installed)
+        self.package_base = package_base
+        self.votes = votes
+        self.popularity = popularity
+        self.maintainer = maintainer
+        self.url_download = url_download
+        self.first_submitted = first_submitted
+        self.last_modified = last_modified
+        self.pkgbuild = pkgbuild
+        self.mirror = mirror
+        self.command = None
+        self.icon_path = None
+        self.downgrade_enabled = False
+        self.desktop_entry = desktop_entry
+
+    @staticmethod
+    def disk_cache_path(pkgname: str, mirror: str):
+        return CACHE_PATH + '/arch/installed/' + ('aur' if mirror == 'aur' else 'mirror') + '/' + pkgname
+
+    def get_pkg_build_url(self):
+        if self.package_base:
+            return 'https://aur.archlinux.org/cgit/aur.git/plain/PKGBUILD?h=' + self.package_base
+
+    def has_history(self):
+        return self.installed
+
+    def has_info(self):
+        return True
+
+    def can_be_installed(self):
+        return super(ArchPackage, self).can_be_installed() and self.url_download
+
+    def can_be_downgraded(self):
+        return self.installed and self.downgrade_enabled
+
+    def get_type(self):
+        return 'aur' if self.mirror == 'aur' else 'arch'
+
+    def get_default_icon_path(self) -> str:
+        return self.get_type_icon_path()
+
+    def get_disk_icon_path(self) -> str:
+        return self.icon_path
+
+    def get_type_icon_path(self):
+        return resource.get_path('img/arch.png', ROOT_DIR)  # TODO change icon when from mirrors
+
+    def is_application(self):
+        return True
+
+    def supports_disk_cache(self):
+        return True
+
+    def get_disk_cache_path(self) -> str:
+        if self.name:
+            return self.disk_cache_path(self.name, self.mirror)
+
+    def get_data_to_cache(self) -> dict:
+        cache = {}
+
+        # required attrs to cache
+        for a in {'command', 'icon_path', 'mirror'}:
+            val = getattr(self, a)
+
+            if val:
+                cache[a] = val
+            else:
+                return None
+
+        if self.desktop_entry:
+            cache['desktop_entry'] = self.desktop_entry
+
+        return cache
+
+    def fill_cached_data(self, data: dict):
+        for a in {'command', 'icon_path', 'mirror', 'desktop_entry'}:
+            val = data.get(a)
+            if val:
+                setattr(self, a, val)
+
+                if a == 'icon_path':
+                    self.icon_url = val
+
+    def can_be_run(self) -> bool:
+        # only returns if there is a desktop entry set for the application to avoid running command-line applications
+        return bool(self.desktop_entry)
+
+    def get_command(self) -> str:
+        return self.command
+
+    def get_publisher(self):
+        return self.maintainer