mirror of
https://github.com/spalencsar/bearhub.git
synced 2026-07-07 00:04:15 +02:00
55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
import logging
|
|
import os
|
|
import time
|
|
from multiprocessing import Process
|
|
from threading import Thread
|
|
|
|
from bauh.api.abstract.context import ApplicationContext
|
|
from bauh.api.abstract.controller import SoftwareManager
|
|
|
|
from bauh.gems.arch import pacman, disk
|
|
|
|
URL_INDEX = 'https://aur.archlinux.org/packages.gz'
|
|
URL_INFO = 'https://aur.archlinux.org/rpc/?v=5&type=info&arg={}'
|
|
|
|
|
|
class AURIndexUpdater(Thread):
|
|
|
|
def __init__(self, context: ApplicationContext, man: SoftwareManager):
|
|
super(AURIndexUpdater, self).__init__(daemon=True)
|
|
self.http_client = context.http_client
|
|
self.logger = context.logger
|
|
self.man = man
|
|
|
|
def run(self):
|
|
while True:
|
|
self.logger.info('Pre-indexing AUR packages in memory')
|
|
res = self.http_client.get(URL_INDEX)
|
|
|
|
if res and res.text:
|
|
self.man.names_index = {n.replace('-', '').replace('_', '').replace('.', ''): n for n in res.text.split('\n') if n and not n.startswith('#')}
|
|
self.logger.info('Pre-indexed {} AUR package names in memory'.format(len(self.man.names_index)))
|
|
else:
|
|
self.logger.warning('No data returned from: {}'.format(URL_INDEX))
|
|
|
|
time.sleep(5 * 60) # updates every 5 minutes
|
|
|
|
|
|
class ArchDiskCacheUpdater(Thread if bool(os.getenv('BAUH_DEBUG', 0)) else Process):
|
|
|
|
def __init__(self, logger: logging.Logger, disk_cache: bool):
|
|
super(ArchDiskCacheUpdater, self).__init__(daemon=True)
|
|
self.logger = logger
|
|
self.disk_cache = disk_cache
|
|
|
|
def run(self):
|
|
if self.disk_cache:
|
|
self.logger.info('Pre-caching installed AUR packages data to disk')
|
|
installed = pacman.list_and_map_installed()
|
|
|
|
saved = 0
|
|
if installed and installed['not_signed']:
|
|
saved = disk.save_several({app for app in installed['not_signed']}, 'aur', overwrite=False)
|
|
|
|
self.logger.info('Pre-cached data of {} AUR packages to the disk'.format(saved))
|