project renamed as 'bauh'

2026-07-09 02:54:16 +02:00 · 2019-08-12 11:50:24 -03:00
parent 6d891683da
commit 19a11434fe
69 changed files with 202 additions and 321 deletions
--- a/bauh/core/snap/worker.py
+++ b/bauh/core/snap/worker.py
@@ -0,0 +1,85 @@
+import time
+import traceback
+
+from colorama import Fore
+
+from bauh.core.controller import ApplicationManager
+from bauh.core.model import ApplicationStatus
+from bauh.core.snap import snap
+from bauh.core.snap.constants import SNAP_API_URL
+from bauh.core.snap.model import SnapApplication
+from bauh.core.worker import AsyncDataLoader
+from bauh.util.cache import Cache
+
+
+class SnapAsyncDataLoader(AsyncDataLoader):
+
+    def __init__(self, app: SnapApplication, manager: ApplicationManager, http_session, api_cache: Cache, download_icons: bool, attempts: int = 2, timeout: int = 30):
+        super(SnapAsyncDataLoader, self).__init__(app=app)
+        self.manager = manager
+        self.http_session = http_session
+        self.attempts = attempts
+        self.api_cache = api_cache
+        self.timeout = timeout
+        self.persist = False
+        self.download_icons = download_icons
+
+    def run(self):
+        if self.app:
+            self.app.status = ApplicationStatus.LOADING_DATA
+
+            for _ in range(0, self.attempts):
+                try:
+                    res = self.http_session.get('{}/search?q={}'.format(SNAP_API_URL, self.app.base_data.name), timeout=self.timeout)
+
+                    if res.status_code == 200 and res.text:
+
+                        try:
+                            snap_list = res.json()['_embedded']['clickindex:package']
+                        except:
+                            self.log_msg('Snap API response responded differently from expected for app: {}'.format(self.app.base_data.name))
+                            break
+
+                        if not snap_list:
+                            break
+
+                        snap_data = snap_list[0]
+
+                        api_data = {
+                            'confinement': snap_data.get('confinement'),
+                            'description': snap_data.get('description'),
+                            'icon_url': snap_data.get('icon_url') if self.download_icons else None
+                        }
+
+                        self.api_cache.add(self.app.base_data.id, api_data)
+                        self.app.confinement = api_data['confinement']
+                        self.app.base_data.icon_url = api_data['icon_url']
+
+                        if not api_data.get('description'):
+                            api_data['description'] = snap.get_info(self.app.base_data.name, ('description',)).get('description')
+
+                        self.app.base_data.description = api_data['description']
+
+                        self.app.status = ApplicationStatus.READY
+                        self.persist = self.app.supports_disk_cache()
+                        break
+                    else:
+                        self.log_msg("Could not retrieve app data for id '{}'. Server response: {}. Body: {}".format(self.app.base_data.id, res.status_code, res.content.decode()), Fore.RED)
+                except:
+                    self.log_msg("Could not retrieve app data for id '{}'".format(self.app.base_data.id), Fore.YELLOW)
+                    traceback.print_exc()
+                    time.sleep(0.5)
+
+            self.app.status = ApplicationStatus.READY
+
+            if self.persist:
+                self.manager.cache_to_disk(app=self.app, icon_bytes=None, only_icon=False)
+
+    def clone(self) -> "SnapAsyncDataLoader":
+        return SnapAsyncDataLoader(manager=self.manager,
+                                   api_cache=self.api_cache,
+                                   attempts=self.attempts,
+                                   http_session=self.http_session,
+                                   timeout=self.timeout,
+                                   app=self.app)
+