KevinMidboe/mktxp-no-cli
1
0
Fork 0
mirror of https://github.com/KevinMidboe/mktxp-no-cli.git synced 2025-10-29 17:50:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #43 from M0r13n/main

Browse Source 
use dict.get() to prevent a KeyError when the key "mac_address" does not exists
This commit is contained in:
Arseniy Kuznetsov
2022-10-23 03:30:21 -05:00
committed by GitHub
parent f6c9778eda 6ea940d951
commit ef8e2a82aa
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mktxp/flow/processor/output.py
View File

@@ -33,11 +33,11 @@ class BaseOutputProcessor:
@staticmethod @staticmethod
def augment_record(router_entry, registration_record, dhcp_lease_records): def augment_record(router_entry, registration_record, dhcp_lease_records):
try: try:
dhcp_lease_record = next((dhcp_lease_record for dhcp_lease_record in dhcp_lease_records if dhcp_lease_record['mac_address']==registration_record['mac_address'])) dhcp_lease_record = next((dhcp_lease_record for dhcp_lease_record in dhcp_lease_records if dhcp_lease_record.get('mac_address')==registration_record.get('mac_address')))
dhcp_name = BaseOutputProcessor.dhcp_name(router_entry, dhcp_lease_record) dhcp_name = BaseOutputProcessor.dhcp_name(router_entry, dhcp_lease_record)
dhcp_address = dhcp_lease_record.get('address', '') dhcp_address = dhcp_lease_record.get('address', '')
except StopIteration: except StopIteration:
dhcp_name = registration_record['mac_address'] dhcp_name = registration_record.get('mac_address')
dhcp_address = 'No DHCP Record' dhcp_address = 'No DHCP Record'
registration_record['dhcp_name'] = dhcp_name registration_record['dhcp_name'] = dhcp_name