black-ified

Author: wdesmedt

nornir_srl/connections/srlinux.py

@@ -596,7 +596,7 @@ def get_nwi_itf(self, nw_instance: str = "*") -> Dict[str, Any]:
 
         res = jmespath.search(path_spec["jmespath"], resp[0])
         return {"nwi_itfs": res}
-    
+
     def get_lag(self, lag_id: str = "*") -> Dict[str, Any]:
         path_spec = {
             "path": f"/interface[name=lag{lag_id}]",
@@ -618,16 +618,33 @@ def get_es(self) -> Dict[str, Any]:
                 oper:"oper-state",itfs:interface[]."ethernet-interface"|join(\' \',@), vrfs:association."network-instance"[].{ni:name, "peers":"_peers"}}',
             "datatype": "state",
         }
+
         def set_es_peers(resp):
-            for bgp_inst in resp[0].get("system/network-instance/protocols/evpn/ethernet-segments", {}).get("bgp-instance", []):
+            for bgp_inst in (
+                resp[0]
+                .get("system/network-instance/protocols/evpn/ethernet-segments", {})
+                .get("bgp-instance", [])
+            ):
                 for es in bgp_inst.get("ethernet-segment", []):
                     for vrf in es.get("association", {}).get("network-instance", []):
-                        es_peers = vrf["bgp-instance"][0].get("computed-designated-forwarder-candidates", {}).get("designated-forwarder-candidate", [])
+                        es_peers = (
+                            vrf["bgp-instance"][0]
+                            .get("computed-designated-forwarder-candidates", {})
+                            .get("designated-forwarder-candidate", [])
+                        )
                         vrf["_peers"] = ", ".join(
-                                f"{peer['address']} (DF)" if peer["designated-forwarder"] else peer["address"]
-                                for peer in es_peers
+                            f"{peer['address']} (DF)"
+                            if peer["designated-forwarder"]
+                            else peer["address"]
+                            for peer in es_peers
                         )
-        if not "evpn" in self.get(paths=["/system/features"], datatype="state")[0]["system/features"]:
+
+        if (
+            not "evpn"
+            in self.get(paths=["/system/features"], datatype="state")[0][
+                "system/features"
+            ]
+        ):
             return {"es": []}
         resp = self.get(
             paths=[path_spec.get("path", "")], datatype=path_spec["datatype"]
@@ -636,7 +653,6 @@ def set_es_peers(resp):
         res = jmespath.search(path_spec["jmespath"], resp[0])
         return {"es": res}
 
-
     def get(
         self,
         paths: List[str],

nornir_srl/fsc.py

@@ -482,6 +482,7 @@ def _subinterfaces(task: Task) -> Result:
         i_filter=ctx.obj["i_filter"],
     )
 
+
 @cli.command()
 @click.pass_context
 @click.option(
@@ -500,9 +501,7 @@ def _lag(task: Task) -> Result:
     f_filter = (
         {k: v for k, v in [f.split("=") for f in field_filter]} if field_filter else {}
     )
-    result = ctx.obj["target"].run(
-        task=_lag, name="lag", raise_on_error=False
-    )
+    result = ctx.obj["target"].run(task=_lag, name="lag", raise_on_error=False)
     print_report(
         result=result,
         name="LAGs",
@@ -512,6 +511,7 @@ def _lag(task: Task) -> Result:
         i_filter=ctx.obj["i_filter"],
     )
 
+
 @cli.command()
 @click.pass_context
 @click.option(
@@ -696,6 +696,7 @@ def _lldp_neighbors(task: Task) -> Result:
         i_filter=ctx.obj["i_filter"],
     )
 
+
 @cli.command()
 @click.pass_context
 @click.option(
@@ -722,8 +723,9 @@ def _es(task: Task) -> Result:
         failed_hosts=result.failed_hosts,
         box_type=ctx.obj["box_type"],
         f_filter=f_filter,
-        i_filter=ctx.obj["i_filter"]
+        i_filter=ctx.obj["i_filter"],
     )
-    
+
+
 if __name__ == "__main__":
     cli(obj={})