53 lines
2.7 KiB
Python
53 lines
2.7 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import time
|
||
|
|
|
||
|
|
from .council import HUMAN, agent, read_claims
|
||
|
|
from .crons import cron_state, queue_rows, timer_rows
|
||
|
|
from .errors import ToolError
|
||
|
|
from .fields import single
|
||
|
|
from .lifetimes import edge_times, lifetime, session_gone, subagent_rows
|
||
|
|
from .names import PSEUDO, SESSION_ID, Directory, title_sightings
|
||
|
|
from .records import epoch, iso_epoch
|
||
|
|
from .register import REGISTER, read_register
|
||
|
|
from .tree import assign
|
||
|
|
|
||
|
|
|
||
|
|
class Sources:
|
||
|
|
def subagents(self, query: dict[str, list[str]]) -> dict:
|
||
|
|
session = single(query, "session") or ""
|
||
|
|
if not SESSION_ID.fullmatch(session):
|
||
|
|
raise ToolError("session must be a session id")
|
||
|
|
now = time.time()
|
||
|
|
beats = [claim["beat"] for claim in read_claims(self.council) if claim["session"] == session and claim["beat"] is not None]
|
||
|
|
try:
|
||
|
|
path = next(iter(sorted(self.transcripts.root.glob(f"*/{session}.jsonl"))), None)
|
||
|
|
except OSError:
|
||
|
|
path = None
|
||
|
|
times = edge_times(path) if path is not None else (None, None, None)
|
||
|
|
life = lifetime(times, session_gone(max(beats, default=None), times[2], now))
|
||
|
|
return {"session": session, **life, "subagents": subagent_rows(self.transcripts.root, session)}
|
||
|
|
|
||
|
|
def directory(self, claims: list[dict], news: list[dict], now: float) -> Directory:
|
||
|
|
claimed = {claim["session"] for claim in claims}
|
||
|
|
sightings = title_sightings(self.transcripts.root, claimed, now) if self.names.due(now) else []
|
||
|
|
sightings += [(claim["session"], claim["agent"], iso_epoch(claim["since"]), "claim") for claim in claims if claim["agent"] and claim["session"] != "-"]
|
||
|
|
self.names.observe(sorted(sightings, key=lambda sighting: (sighting[2] is None, sighting[2] or 0.0)), now)
|
||
|
|
known = {entry["name"] for entry in self.names.listed()}
|
||
|
|
radio = []
|
||
|
|
for row in news:
|
||
|
|
name, ts = agent(row.get("from")), epoch(row.get("ts"))
|
||
|
|
if name and name != HUMAN and name not in known and ts is not None:
|
||
|
|
radio.append((f"{PSEUDO}{name}", name, ts, "radio"))
|
||
|
|
known.add(name)
|
||
|
|
self.names.observe(radio, now)
|
||
|
|
return Directory(self.names.listed())
|
||
|
|
|
||
|
|
def repositories(self) -> dict:
|
||
|
|
tasks = read_register(self.council / REGISTER)
|
||
|
|
return assign(self.tree.get(), tasks, read_claims(self.council), self.reviews.listed())
|
||
|
|
|
||
|
|
def schedules(self, claims: list[dict], now: float) -> list[dict]:
|
||
|
|
beats = {claim["session"]: claim["beat"] for claim in claims if claim["beat"] is not None}
|
||
|
|
rows = [cron_state(cron, beats, now) for cron in self.transcripts.schedules()]
|
||
|
|
return rows + timer_rows(self.timers) + queue_rows(self.council)
|