mirror of
https://github.com/spalencsar/bearhub.git
synced 2026-07-06 21:44:16 +02:00
53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
import logging
|
|
import os
|
|
import time
|
|
from multiprocessing import Process
|
|
from threading import Thread
|
|
|
|
from bauh.api.http import HttpClient
|
|
|
|
|
|
class DatabaseUpdater(Thread if bool(int(os.getenv('BAUH_DEBUG', 0))) else Process):
|
|
|
|
URL_APPS = 'https://github.com/vinifmor/bauh-files/blob/master/appimage/apps.db?raw=true'
|
|
URL_RELEASES = 'https://github.com/vinifmor/bauh-files/blob/master/appimage/releases.db?raw=true'
|
|
APPS_PATH = '{}/.local/share/bauh/appimage/apps.db'
|
|
RELEASES_PATH = '{}/.local/share/bauh/appimage/releases.db'
|
|
|
|
def __init__(self, http_client: HttpClient, logger: logging.Logger):
|
|
super(DatabaseUpdater, self).__init__(daemon=True)
|
|
self.http_client = http_client
|
|
self.logger = logger
|
|
self.enabled = bool(int(os.getenv('BAUH_APPIMAGE_DB_UPDATER', 1)))
|
|
self.sleep = 60 * 20
|
|
|
|
def _download_file(self, url: str, output: str):
|
|
res = self.http_client.get(url, headers={'Authorization': 'token {}'.format(os.getenv('GITHUB_TOKEN'))})
|
|
|
|
if res:
|
|
with open(output, 'wb+') as f:
|
|
f.write(res.content)
|
|
|
|
self.logger.info("Database file saved at {}".format(output))
|
|
else:
|
|
self.logger.warning('Could not download a database file from {}'.format(url))
|
|
|
|
def run(self):
|
|
if self.enabled:
|
|
while True:
|
|
self.logger.info('Retrieving AppImage databases')
|
|
|
|
threads = [Thread(target=self._download_file, args=(self.URL_APPS, self.APPS_PATH)),
|
|
Thread(target=self._download_file, args=(self.URL_RELEASES, self.RELEASES_PATH))]
|
|
|
|
for t in threads:
|
|
t.start()
|
|
|
|
for t in threads:
|
|
t.join()
|
|
|
|
self.logger.info('Sleeping')
|
|
time.sleep(self.sleep)
|
|
else:
|
|
self.logger.warning('Disabled')
|