mirror of
https://github.com/ArchiveBox/ArchiveBox
synced 2024-11-22 20:23:12 +00:00
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
__package__ = 'archivebox.core'
|
|
|
|
from typing import ClassVar
|
|
|
|
from statemachine import State
|
|
|
|
from core.models import Snapshot, ArchiveResult
|
|
from core.statemachines import SnapshotMachine, ArchiveResultMachine
|
|
from actors.actor import ActorType
|
|
|
|
|
|
class SnapshotActor(ActorType[Snapshot]):
|
|
Model = Snapshot
|
|
StateMachineClass = SnapshotMachine
|
|
|
|
ACTIVE_STATE: ClassVar[State] = SnapshotMachine.started
|
|
FINAL_STATES: ClassVar[list[State]] = SnapshotMachine.final_states
|
|
STATE_FIELD_NAME: ClassVar[str] = SnapshotMachine.state_field_name
|
|
|
|
MAX_CONCURRENT_ACTORS: ClassVar[int] = 3
|
|
MAX_TICK_TIME: ClassVar[int] = 10
|
|
CLAIM_FROM_TOP_N: ClassVar[int] = MAX_CONCURRENT_ACTORS * 10
|
|
|
|
|
|
|
|
class ArchiveResultActor(ActorType[ArchiveResult]):
|
|
Model = ArchiveResult
|
|
StateMachineClass = ArchiveResultMachine
|
|
|
|
ACTIVE_STATE: ClassVar[State] = ArchiveResultMachine.started
|
|
FINAL_STATES: ClassVar[list[State]] = ArchiveResultMachine.final_states
|
|
STATE_FIELD_NAME: ClassVar[str] = ArchiveResultMachine.state_field_name
|
|
|
|
MAX_CONCURRENT_ACTORS: ClassVar[int] = 6
|
|
MAX_TICK_TIME: ClassVar[int] = 60
|
|
CLAIM_FROM_TOP_N: ClassVar[int] = MAX_CONCURRENT_ACTORS * 10
|
|
|
|
# @classproperty
|
|
# def qs(cls) -> QuerySet[ModelType]:
|
|
# """Get the unfiltered and unsorted QuerySet of all objects that this Actor might care about."""
|
|
# return cls.Model.objects.filter(extractor='favicon')
|