Init

core/__init__.py

@@ -0,0 +1 @@
+__version__ = '0.1.0'

core/controller.py

@@ -0,0 +1,22 @@
+from typing import List
+
+from core.model import FlatpakManager
+
+
+class FlatpakController:
+
+    def __init__(self, model: FlatpakManager):
+        self.model = model
+
+    def refresh(self) -> List[dict]:
+        return self.model.read_installed()
+
+    def update(self, package_refs: List[str]) -> List[dict]:
+        return self.model.update_packages(package_refs)
+
+    def check_installed(self) -> bool:
+        version = self.model.get_version()
+        return False if version is None else version
+
+    def get_version(self) -> str:
+        return self.model.get_version()

core/flatpak.py

@@ -0,0 +1,55 @@
+import subprocess
+from typing import List
+
+
+def _package_str_to_json(line: str, version: str) -> dict:
+
+    app_array = line.split('\t')
+
+    if version >= '1.4.0':
+        app = {'name': app_array[0], 'id': app_array[1], 'version': app_array[2], 'branch': app_array[3], 'runtime': 'runtime' in app_array[4]}
+        # TODO not working yet
+
+    elif '1.0' <= version < '1.1':
+        app = {'ref': app_array[0], 'options': app_array[1]}
+
+        ref_data = app['ref'].split('/')
+        app['id'] = ref_data[0]
+        app['arch'] = ref_data[1]
+        app['branch'] = ref_data[2]
+        app['name'] = ref_data[0].split('.')[-1]
+        app['runtime'] = 'runtime' in app['options']
+    else:
+        raise Exception('Unsupported version')
+
+    return app
+
+
+def _run_cmd(cmd: str) -> str:
+    res = subprocess.run(cmd, shell=True, stdout=subprocess.PIPE)
+    return res.stdout.decode() if res.returncode == 0 else None
+
+
+def get_version():
+    res = _run_cmd('flatpak --version')
+    return res.split(' ')[1].strip() if res else None
+
+
+def list_installed() -> List[str]:
+    apps_str = _run_cmd('flatpak list')
+
+    if apps_str:
+        version = get_version()
+        app_lines = apps_str.split('\n')
+        return [_package_str_to_json(line, version) for line in app_lines if line]
+
+    return None
+
+
+def check_update(ref: dict) -> bool:
+    res = _run_cmd('flatpak update ' + ref)
+    return 'Updating in system' in res
+
+
+def update(ref: str):
+    return bool(_run_cmd('flatpak update -y ' + ref))

core/model.py

@@ -0,0 +1,75 @@
+from typing import List
+
+import requests
+from PyQt5.QtCore import QUrl
+from PyQt5.QtGui import QPixmap, QIcon
+from PyQt5.QtNetwork import QNetworkAccessManager, QNetworkRequest
+
+from core import flatpak
+
+__FLATHUB_URL__ = 'https://flathub.org'
+__FLATHUB_API_URL__ = __FLATHUB_URL__ + '/api/v1'
+
+
+class FlatpakManager:
+
+    def __init__(self):
+        self.packages = []
+        self.packages_db = {}
+        self.http_session = requests.Session()
+        self._load_database()
+
+    def _load_database(self):
+
+        res = self.http_session.get(__FLATHUB_API_URL__ + '/apps')
+
+        if res.status_code == 200:
+            for app in res.json():
+                self.packages_db[app['flatpakAppId']] = app
+
+    def get_version(self):
+        return flatpak.get_version()
+
+    def read_installed(self) -> List[dict]:
+
+        packages = flatpak.list_installed()
+
+        if packages:
+            packages.sort(key=lambda p: p['name'].lower())
+
+            for pak in packages:
+
+                if self.packages_db:
+                    package_data = self.packages_db.get(pak['id'], None)
+                else:
+                    package_data = None
+
+                if not package_data:
+                    pak['latest_release'] = None
+                    pak['icon'] = None
+                else:
+                    pak['latest_release'] = package_data['currentReleaseVersion']
+                    pak['icon'] = __FLATHUB_URL__ + package_data['iconDesktopUrl']
+
+                pak['update'] = flatpak.check_update(pak['ref'])
+
+        self.packages = packages
+        return [*self.packages]
+
+    def update_packages(self, refs: List[str]) -> List[dict]:
+
+        if self.packages:
+
+            for ref in refs:
+                package_found = [pak for pak in self.packages if pak['ref'] == ref]
+
+                if package_found:
+                    package_found = package_found[0]
+                    updated = flatpak.update(ref)
+
+                    if updated:
+                        package_found['update'] = not updated
+
+            return [*self.packages]
+
+        return []

core/resource.py

@@ -0,0 +1,6 @@
+import os
+
+
+def get_path(resource_path):
+    cur_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+    return cur_dir + '/resources/' + resource_path

core/updater.py

@@ -0,0 +1,16 @@
+from model import flatpak
+
+
+def list_packages():
+    packages = flatpak.list_installed()
+
+    if packages:
+        packages = [pak for pak in packages if not pak['runtime']]
+
+        if packages:
+            packages.sort(key=lambda p: p['name'].lower())
+
+            for pak in packages:
+                pak['update'] = flatpak.check_update(pak['ref'])
+
+    return packages