tools: fix style check babletrace file

python3 -m black tools/frr_babeltrace.py

Signed-off-by: Chirag Shah <chirag@nvidia.com>
This commit is contained in:
Chirag Shah 2024-11-25 08:58:25 -08:00
parent 91119dab9b
commit 814a89f6e5

View file

@ -18,6 +18,7 @@ import sys
import babeltrace import babeltrace
########################### common parsers - start ############################ ########################### common parsers - start ############################
def print_ip_addr(field_val): def print_ip_addr(field_val):
""" """
@ -48,24 +49,28 @@ def print_mac(field_val):
""" """
return ":".join("%02x" % fb for fb in field_val) return ":".join("%02x" % fb for fb in field_val)
def print_net_ipv4_addr(field_val): def print_net_ipv4_addr(field_val):
""" """
pretty print ctf_integer_network ipv4 pretty print ctf_integer_network ipv4
""" """
return str(ipaddress.IPv4Address(field_val)) return str(ipaddress.IPv4Address(field_val))
def print_esi(field_val): def print_esi(field_val):
""" """
pretty print ethernet segment id, esi_t pretty print ethernet segment id, esi_t
""" """
return ":".join("%02x" % fb for fb in field_val) return ":".join("%02x" % fb for fb in field_val)
def get_field_list(event): def get_field_list(event):
""" """
only fetch fields added via the TP, skip metadata etc. only fetch fields added via the TP, skip metadata etc.
""" """
return event.field_list_with_scope(babeltrace.CTFScope.EVENT_FIELDS) return event.field_list_with_scope(babeltrace.CTFScope.EVENT_FIELDS)
def parse_event(event, field_parsers): def parse_event(event, field_parsers):
""" """
Wild card event parser; doesn't make things any prettier Wild card event parser; doesn't make things any prettier
@ -103,6 +108,7 @@ def print_family_str(field_val):
############################ common parsers - end ############################# ############################ common parsers - end #############################
############################ evpn parsers - start ############################# ############################ evpn parsers - start #############################
def parse_frr_bgp_evpn_mac_ip_zsend(event): def parse_frr_bgp_evpn_mac_ip_zsend(event):
""" """
@ -114,13 +120,16 @@ def parse_frr_bgp_evpn_mac_ip_zsend(event):
ctf_integer_network_hex(unsigned int, vtep, vtep.s_addr) ctf_integer_network_hex(unsigned int, vtep, vtep.s_addr)
ctf_array(unsigned char, esi, esi, sizeof(esi_t)) ctf_array(unsigned char, esi, esi, sizeof(esi_t))
""" """
field_parsers = {"ip": print_ip_addr, field_parsers = {
"mac": print_mac, "ip": print_ip_addr,
"esi": print_esi, "mac": print_mac,
"vtep": print_net_ipv4_addr} "esi": print_esi,
"vtep": print_net_ipv4_addr,
}
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_bum_vtep_zsend(event): def parse_frr_bgp_evpn_bum_vtep_zsend(event):
""" """
bgp evpn bum-vtep parser; raw format - bgp evpn bum-vtep parser; raw format -
@ -132,6 +141,7 @@ def parse_frr_bgp_evpn_bum_vtep_zsend(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_mh_nh_rmac_send(event): def parse_frr_bgp_evpn_mh_nh_rmac_send(event):
""" """
bgp evpn nh-rmac parser; raw format - bgp evpn nh-rmac parser; raw format -
@ -141,17 +151,18 @@ def parse_frr_bgp_evpn_mh_nh_rmac_send(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_mh_local_es_add_zrecv(event): def parse_frr_bgp_evpn_mh_local_es_add_zrecv(event):
""" """
bgp evpn local-es parser; raw format - bgp evpn local-es parser; raw format -
ctf_array(unsigned char, esi, esi, sizeof(esi_t)) ctf_array(unsigned char, esi, esi, sizeof(esi_t))
ctf_integer_network_hex(unsigned int, vtep, vtep.s_addr) ctf_integer_network_hex(unsigned int, vtep, vtep.s_addr)
""" """
field_parsers = {"esi": print_esi, field_parsers = {"esi": print_esi, "vtep": print_net_ipv4_addr}
"vtep": print_net_ipv4_addr}
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_mh_local_es_del_zrecv(event): def parse_frr_bgp_evpn_mh_local_es_del_zrecv(event):
""" """
bgp evpn local-es parser; raw format - bgp evpn local-es parser; raw format -
@ -161,6 +172,7 @@ def parse_frr_bgp_evpn_mh_local_es_del_zrecv(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_mh_local_es_evi_add_zrecv(event): def parse_frr_bgp_evpn_mh_local_es_evi_add_zrecv(event):
""" """
bgp evpn local-es-evi parser; raw format - bgp evpn local-es-evi parser; raw format -
@ -170,6 +182,7 @@ def parse_frr_bgp_evpn_mh_local_es_evi_add_zrecv(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_mh_local_es_evi_del_zrecv(event): def parse_frr_bgp_evpn_mh_local_es_evi_del_zrecv(event):
""" """
bgp evpn local-es-evi parser; raw format - bgp evpn local-es-evi parser; raw format -
@ -179,6 +192,7 @@ def parse_frr_bgp_evpn_mh_local_es_evi_del_zrecv(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_mh_es_evi_vtep_add(event): def parse_frr_bgp_evpn_mh_es_evi_vtep_add(event):
""" """
bgp evpn remote ead evi remote vtep add; raw format - bgp evpn remote ead evi remote vtep add; raw format -
@ -189,6 +203,7 @@ def parse_frr_bgp_evpn_mh_es_evi_vtep_add(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_mh_es_evi_vtep_del(event): def parse_frr_bgp_evpn_mh_es_evi_vtep_del(event):
""" """
bgp evpn remote ead evi remote vtep del; raw format - bgp evpn remote ead evi remote vtep del; raw format -
@ -199,6 +214,7 @@ def parse_frr_bgp_evpn_mh_es_evi_vtep_del(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_mh_local_ead_es_evi_route_upd(event): def parse_frr_bgp_evpn_mh_local_ead_es_evi_route_upd(event):
""" """
bgp evpn local ead evi vtep; raw format - bgp evpn local ead evi vtep; raw format -
@ -209,6 +225,7 @@ def parse_frr_bgp_evpn_mh_local_ead_es_evi_route_upd(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_mh_local_ead_es_evi_route_del(event): def parse_frr_bgp_evpn_mh_local_ead_es_evi_route_del(event):
""" """
bgp evpn local ead evi vtep del; raw format - bgp evpn local ead evi vtep del; raw format -
@ -219,6 +236,7 @@ def parse_frr_bgp_evpn_mh_local_ead_es_evi_route_del(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_local_vni_add_zrecv(event): def parse_frr_bgp_evpn_local_vni_add_zrecv(event):
""" """
bgp evpn local-vni parser; raw format - bgp evpn local-vni parser; raw format -
@ -230,6 +248,7 @@ def parse_frr_bgp_evpn_local_vni_add_zrecv(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_local_l3vni_add_zrecv(event): def parse_frr_bgp_evpn_local_l3vni_add_zrecv(event):
""" """
bgp evpn local-l3vni parser; raw format - bgp evpn local-l3vni parser; raw format -
@ -237,12 +256,15 @@ def parse_frr_bgp_evpn_local_l3vni_add_zrecv(event):
ctf_array(unsigned char, svi_rmac, svi_rmac, sizeof(struct ethaddr)) ctf_array(unsigned char, svi_rmac, svi_rmac, sizeof(struct ethaddr))
ctf_array(unsigned char, vrr_rmac, vrr_rmac, sizeof(struct ethaddr)) ctf_array(unsigned char, vrr_rmac, vrr_rmac, sizeof(struct ethaddr))
""" """
field_parsers = {"vtep": print_net_ipv4_addr, field_parsers = {
"svi_rmac": print_mac, "vtep": print_net_ipv4_addr,
"vrr_rmac": print_mac} "svi_rmac": print_mac,
"vrr_rmac": print_mac,
}
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_local_macip_add_zrecv(event): def parse_frr_bgp_evpn_local_macip_add_zrecv(event):
""" """
bgp evpn local-mac-ip parser; raw format - bgp evpn local-mac-ip parser; raw format -
@ -256,6 +278,7 @@ def parse_frr_bgp_evpn_local_macip_add_zrecv(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_local_macip_del_zrecv(event): def parse_frr_bgp_evpn_local_macip_del_zrecv(event):
""" """
bgp evpn local-mac-ip del parser; raw format - bgp evpn local-mac-ip del parser; raw format -
@ -267,16 +290,20 @@ def parse_frr_bgp_evpn_local_macip_del_zrecv(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_advertise_type5(event): def parse_frr_bgp_evpn_advertise_type5(event):
""" """
local originated type-5 route local originated type-5 route
""" """
field_parsers = {"ip": print_ip_addr, field_parsers = {
"rmac": print_mac, "ip": print_ip_addr,
"vtep": print_net_ipv4_addr} "rmac": print_mac,
"vtep": print_net_ipv4_addr,
}
parse_event(event, field_parsers) parse_event(event, field_parsers)
def parse_frr_bgp_evpn_withdraw_type5(event): def parse_frr_bgp_evpn_withdraw_type5(event):
""" """
local originated type-5 route withdraw local originated type-5 route withdraw
@ -285,8 +312,10 @@ def parse_frr_bgp_evpn_withdraw_type5(event):
parse_event(event, field_parsers) parse_event(event, field_parsers)
############################ evpn parsers - end *############################# ############################ evpn parsers - end *#############################
def main(): def main():
""" """
FRR lttng trace output parser; babel trace plugin FRR lttng trace output parser; babel trace plugin
@ -341,5 +370,6 @@ def main():
else: else:
parse_event(event, {}) parse_event(event, {})
if __name__ == "__main__": if __name__ == "__main__":
main() main()