mirror of
https://github.com/ArchiveBox/ArchiveBox.git
synced 2026-04-06 07:47:53 +10:00
65 lines
2.6 KiB
Python
65 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
from asgiref.sync import sync_to_async
|
|
from abx_dl.events import BinaryEvent, BinaryInstalledEvent
|
|
from abx_dl.services.base import BaseService
|
|
|
|
|
|
class BinaryService(BaseService):
|
|
LISTENS_TO = [BinaryEvent, BinaryInstalledEvent]
|
|
EMITS = []
|
|
|
|
async def on_BinaryEvent(self, event: BinaryEvent) -> None:
|
|
await sync_to_async(self._project_binary, thread_sensitive=True)(event)
|
|
|
|
async def on_BinaryInstalledEvent(self, event: BinaryInstalledEvent) -> None:
|
|
await sync_to_async(self._project_installed_binary, thread_sensitive=True)(event)
|
|
|
|
def _project_binary(self, event: BinaryEvent) -> None:
|
|
from archivebox.machine.models import Binary, Machine
|
|
|
|
machine = Machine.current()
|
|
existing = Binary.objects.filter(machine=machine, name=event.name).first()
|
|
if existing and existing.status == Binary.StatusChoices.INSTALLED:
|
|
changed = False
|
|
if event.binproviders and existing.binproviders != event.binproviders:
|
|
existing.binproviders = event.binproviders
|
|
changed = True
|
|
if event.overrides and existing.overrides != event.overrides:
|
|
existing.overrides = event.overrides
|
|
changed = True
|
|
if changed:
|
|
existing.save(update_fields=["binproviders", "overrides", "modified_at"])
|
|
return
|
|
|
|
Binary.from_json(
|
|
{
|
|
"name": event.name,
|
|
"abspath": event.abspath,
|
|
"version": event.version,
|
|
"sha256": event.sha256,
|
|
"binproviders": event.binproviders,
|
|
"binprovider": event.binprovider,
|
|
"overrides": event.overrides or {},
|
|
},
|
|
)
|
|
|
|
def _project_installed_binary(self, event: BinaryInstalledEvent) -> None:
|
|
from archivebox.machine.models import Binary, Machine
|
|
|
|
machine = Machine.current()
|
|
binary, _ = Binary.objects.get_or_create(
|
|
machine=machine,
|
|
name=event.name,
|
|
defaults={
|
|
"status": Binary.StatusChoices.QUEUED,
|
|
},
|
|
)
|
|
binary.abspath = event.abspath or binary.abspath
|
|
binary.version = event.version or binary.version
|
|
binary.sha256 = event.sha256 or binary.sha256
|
|
binary.binprovider = event.binprovider or binary.binprovider
|
|
binary.status = Binary.StatusChoices.INSTALLED
|
|
binary.retry_at = None
|
|
binary.save(update_fields=["abspath", "version", "sha256", "binprovider", "status", "retry_at", "modified_at"])
|