mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-17 22:24:23 +01:00
43b749f417
The Builder's puerpose is to build data structures from the data obtained from the DB for both the Watcher and the Responder
40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
from pisa.responder import Job
|
|
from pisa.appointment import Appointment
|
|
|
|
|
|
class Builder:
|
|
|
|
@staticmethod
|
|
def build_appointments(appointments_data):
|
|
appointments = {}
|
|
locator_uuid_map = {}
|
|
|
|
for uuid, appointment_data in appointments_data.items():
|
|
appointment = Appointment.from_dict(appointment_data)
|
|
appointments[uuid] = appointment
|
|
|
|
if appointment.locator in locator_uuid_map:
|
|
locator_uuid_map[appointment.locator].append(uuid)
|
|
|
|
else:
|
|
locator_uuid_map[appointment.locator] = [uuid]
|
|
|
|
return appointments, locator_uuid_map
|
|
|
|
@staticmethod
|
|
def build_jobs(jobs_data):
|
|
jobs = {}
|
|
tx_job_map = {}
|
|
|
|
for uuid, job_data in jobs_data.items():
|
|
job = Job.from_dict(jobs_data)
|
|
jobs[uuid] = job
|
|
|
|
if job.justice_txid in tx_job_map:
|
|
tx_job_map[job.justice_txid].append(uuid)
|
|
|
|
else:
|
|
tx_job_map[job.justice_txid] = [uuid]
|
|
|
|
return jobs, tx_job_map
|