Additional bug-fixing

This commit is contained in:
Sergi Delgado Segura
2019-10-27 13:04:43 -07:00
parent dfaec51a3b
commit 15a66751b2
2 changed files with 14 additions and 11 deletions

View File

@@ -9,8 +9,8 @@ class Builder:
appointments = {}
locator_uuid_map = {}
for uuid, appointment_data in appointments_data.items():
appointment = Appointment.from_dict(appointment_data)
for uuid, data in appointments_data.items():
appointment = Appointment.from_dict(data)
appointments[uuid] = appointment
if appointment.locator in locator_uuid_map:
@@ -26,8 +26,8 @@ class Builder:
jobs = {}
tx_job_map = {}
for uuid, job_data in jobs_data.items():
job = Job.from_dict(jobs_data)
for uuid, data in jobs_data.items():
job = Job.from_dict(data)
jobs[uuid] = job
if job.justice_txid in tx_job_map:

View File

@@ -29,20 +29,23 @@ class Job:
self.locator = sha256(unhexlify(dispute_txid)).hexdigest()
@classmethod
def from_dict(cls, jobs_data):
dispute_txid = jobs_data.get("dispute_txid")
justice_txid = jobs_data.get("justice_txid")
justice_rawtx = jobs_data.get("justice_rawtx")
appointment_end = jobs_data.get("appointment_end")
def from_dict(cls, job_data):
dispute_txid = job_data.get("dispute_txid")
justice_txid = job_data.get("justice_txid")
justice_rawtx = job_data.get("justice_rawtx")
appointment_end = job_data.get("appointment_end")
if all([dispute_txid, justice_txid, justice_rawtx, appointment_end]) is not None:
if all(v is not None for v in [dispute_txid, justice_txid, justice_rawtx, appointment_end]):
job = cls(dispute_txid, justice_txid, justice_rawtx, appointment_end)
else:
raise ValueError("Wrong job data, some fields are missing")
return job
def to_dict(self):
job = {"locator": self.locator, "justice_rawtx": self.justice_rawtx, "appointment_end": self.appointment_end}
job = {"locator": self.locator, "dispute_txid": self.dispute_txid, "justice_txid": self.justice_txid,
"justice_rawtx": self.justice_rawtx, "appointment_end": self.appointment_end}
return job