def __init__(self,
db: "MephistoDB",
db_id: str,
row: Optional[Mapping[str, Any]] = None):
self.db: "MephistoDB" = db
if row is None:
row = db.get_agent(db_id)
assert row is not None, f"Given db_id {db_id} did not exist in given db"
self.db_id: str = row["agent_id"]
self.db_status = row["status"]
self.worker_id = row["worker_id"]
self.unit_id = row["unit_id"]
self.task_type = row["task_type"]
self.provider_type = row["provider_type"]
self.pending_observations: List["Packet"] = []
self.pending_actions: List["Packet"] = []
self.has_action = threading.Event()
self.has_action.clear()
self.wants_action = threading.Event()
self.wants_action.clear()
self.has_updated_status = threading.Event()
self.assignment_id = row["assignment_id"]
self.task_run_id = row["task_run_id"]
self.task_id = row["task_id"]
self.did_submit = threading.Event()
# Deferred loading of related entities
self._worker: Optional["Worker"] = None
self._unit: Optional["Unit"] = None
self._assignment: Optional["Assignment"] = None
self._task_run: Optional["TaskRun"] = None
self._task: Optional["Task"] = None
# Follow-up initialization
self.state = AgentState(self) # type: ignore
def __init__(
self,
db: "MephistoDB",
db_id: str,
row: Optional[Mapping[str, Any]] = None,
_used_new_call: bool = False,
):
if not _used_new_call:
raise AssertionError(
"Direct OnboardingAgent and data model access via OnboardingAgent(db, id) is "
"now deprecated in favor of calling OnboardingAgent.get(db, id). "
)
self.db: "MephistoDB" = db
if row is None:
row = db.get_onboarding_agent(db_id)
assert row is not None, f"Given db_id {db_id} did not exist in given db"
self.db_id: str = row["onboarding_agent_id"]
self.db_status: str = row["status"]
self.worker_id: str = row["worker_id"]
self.task_type: str = row["task_type"]
self.pending_actions: "Queue[Dict[str, Any]]" = Queue()
self.has_live_update = threading.Event()
self.has_live_update.clear()
self.task_run_id: str = row["task_run_id"]
self.task_id: str = row["task_id"]
self.did_submit = threading.Event()
self.is_shutdown = False
# Deferred loading of related entities
self._worker: Optional["Worker"] = None
self._task_run: Optional["TaskRun"] = None
self._task: Optional["Task"] = None
# Related entity set by a live run
self._associated_live_run: Optional["LiveTaskRun"] = None
# Follow-up initialization
self.state = AgentState(self) # type: ignore
def state(self) -> "AgentState":
if self._state is None:
self._state = AgentState(self) # type: ignore
return cast("AgentState", self._state)
def state(self) -> "AgentState":
if self._state is None:
self._state = AgentState(self)
return self._state