aljaz/python-teos
1
0
Fork 0
mirror of https://github.com/aljazceru/python-teos.git synced 2025-12-17 22:24:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Updates Watcher to use new Cleaner functions and reformats filter_valid_breaches to return a list of valid and invalid breaches.

Browse Source
This commit is contained in:
Sergi Delgado Segura
2020-01-21 19:11:15 +01:00
parent 33f2ab35d8
commit 74a9dad071
1 changed files with 40 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

81
pisa/watcher.py
View File

@@ -124,7 +124,7 @@ class Watcher:
logger.info("Waking up") logger.info("Waking up")
self.db_manager.store_watcher_appointment(uuid, appointment.to_json()) self.db_manager.store_watcher_appointment(uuid, appointment.to_json())
self.db_manager.store_update_locator_map(appointment.locator, uuid) self.db_manager.create_append_locator_map(appointment.locator, uuid)
appointment_added = True appointment_added = True
signature = Cryptographer.sign(appointment.serialize(), self.signing_key) signature = Cryptographer.sign(appointment.serialize(), self.signing_key)
@@ -164,37 +164,38 @@ class Watcher:
if block["height"] > appointment_data.get("end_time") + self.config.get("EXPIRY_DELTA") if block["height"] > appointment_data.get("end_time") + self.config.get("EXPIRY_DELTA")
] ]
Cleaner.delete_expired_appointment( Cleaner.delete_expired_appointments(
expired_appointments, self.appointments, self.locator_uuid_map, self.db_manager expired_appointments, self.appointments, self.locator_uuid_map, self.db_manager
) )
filtered_breaches = self.filter_valid_breaches(self.get_breaches(txids)) valid_breaches, invalid_breaches = self.filter_valid_breaches(self.get_breaches(txids))
for uuid, filtered_breach in filtered_breaches.items(): for uuid, breach in valid_breaches.items():
# Errors decrypting the Blob will result in a None penalty_txid logger.info(
if filtered_breach["valid_breach"] is True: "Notifying responder and deleting appointment",
logger.info( penalty_txid=breach["penalty_txid"],
"Notifying responder and deleting appointment", locator=breach["locator"],
penalty_txid=filtered_breach["penalty_txid"], uuid=uuid,
locator=filtered_breach["locator"],
uuid=uuid,
)
self.responder.handle_breach(
uuid,
filtered_breach["locator"],
filtered_breach["dispute_txid"],
filtered_breach["penalty_txid"],
filtered_breach["penalty_rawtx"],
self.appointments[uuid].get("end_time"),
block_hash,
)
# Delete the appointment and update db
Cleaner.delete_completed_appointment(
uuid, self.appointments, self.locator_uuid_map, self.db_manager
) )
self.responder.handle_breach(
uuid,
breach["locator"],
breach["dispute_txid"],
breach["penalty_txid"],
breach["penalty_rawtx"],
self.appointments[uuid].get("end_time"),
block_hash,
)
Cleaner.flag_triggered_appointments(
list(valid_breaches.keys()), self.appointments, self.locator_uuid_map, self.db_manager
)
Cleaner.delete_completed_appointments(
invalid_breaches, self.appointments, self.locator_uuid_map, self.db_manager
)
# Register the last processed block for the watcher # Register the last processed block for the watcher
self.db_manager.store_last_block_hash_watcher(block_hash) self.db_manager.store_last_block_hash_watcher(block_hash)
@@ -248,7 +249,8 @@ class Watcher:
``{locator, dispute_txid, penalty_txid, penalty_rawtx, valid_breach}`` ``{locator, dispute_txid, penalty_txid, penalty_rawtx, valid_breach}``
""" """
filtered_breaches = {} valid_breaches = {}
invalid_breaches = []
for locator, dispute_txid in breaches.items(): for locator, dispute_txid in breaches.items():
for uuid in self.locator_uuid_map[locator]: for uuid in self.locator_uuid_map[locator]:
@@ -263,21 +265,18 @@ class Watcher:
penalty_tx = BlockProcessor.decode_raw_transaction(penalty_rawtx) penalty_tx = BlockProcessor.decode_raw_transaction(penalty_rawtx)
if penalty_tx is not None: if penalty_tx is not None:
penalty_txid = penalty_tx.get("txid") valid_breaches[uuid] = {
valid_breach = True "locator": locator,
"dispute_txid": dispute_txid,
"penalty_txid": penalty_tx.get("txid"),
"penalty_rawtx": penalty_rawtx,
}
logger.info("Breach found for locator", locator=locator, uuid=uuid, penalty_txid=penalty_txid) logger.info(
"Breach found for locator", locator=locator, uuid=uuid, penalty_txid=penalty_tx.get("txid")
)
else: else:
penalty_txid = None invalid_breaches.append(uuid)
valid_breach = False
filtered_breaches[uuid] = { return valid_breaches, invalid_breaches
"locator": locator,
"dispute_txid": dispute_txid,
"penalty_txid": penalty_txid,
"penalty_rawtx": penalty_rawtx,
"valid_breach": valid_breach,
}
return filtered_breaches