Additional typing info

Author: Ktmi

kytos/core/controller.py

@@ -52,6 +52,8 @@
 from kytos.core.queue_monitor import QueueMonitorWindow
 from kytos.core.switch import Switch
 
+import kytos.core.interface
+
 __all__ = ('Controller',)
 
 
@@ -129,7 +131,7 @@ def __init__(self, options=None, loop: AbstractEventLoop = None):
         #: dict: Current existing switches.
         #:
         #: The key is the switch dpid, while the value is a Switch object.
-        self.switches = {}  # dpid: Switch()
+        self.switches = dict[str, Switch]() # dpid: Switch()
         self._switches_lock = threading.Lock()
 
         #: datetime.datetime: Time when the controller finished starting.
@@ -676,7 +678,10 @@ async def msg_out_event_handler(self):
                 self.log.exception("Unhandled exception on msg_out",
                                    exc_info=exc)
 
-    def get_interface_by_id(self, interface_id):
+    def get_interface_by_id(
+        self,
+        interface_id: str
+    ) -> "kytos.core.interface.Interface":
         """Find a Interface  with interface_id.
 
         Args:

kytos/core/napps/base.py

@@ -14,6 +14,8 @@
 from kytos.core.events import KytosEvent
 from kytos.core.logs import NAppLog
 
+import kytos.core.controller
+
 __all__ = ('KytosNApp',)
 
 LOG = NAppLog()
@@ -165,7 +167,7 @@ def _update_repo_file(self, destination=None):
 class KytosNApp(Thread, metaclass=ABCMeta):
     """Base class for any KytosNApp to be developed."""
 
-    def __init__(self, controller, **kwargs):
+    def __init__(self, controller: "kytos.core.controller.Controller", **kwargs):
         """Contructor of KytosNapps.
 
         Go through all of the instance methods and selects those that have

kytos/core/switch.py

@@ -80,7 +80,7 @@ def __init__(self, dpid, connection=None, features=None):
         #:     (eth_type, mac_src, mac_dst) and the value is the timestamp of
         #:     the last flood.
         self.flood_table = {}
-        self.interfaces = {}
+        self.interfaces = dict[int, Interface]()
         self.flows = []
         self.description = {}
         self._id = dpid

kytos/core/tag_ranges.py

@@ -150,8 +150,8 @@ def range_intersection(
     return intersections
 
 def range_difference(
-    ranges_a: list[Optional[list[int]]],
-    ranges_b: list[Optional[list[int]]]
+    ranges_a: list[list[int]],
+    ranges_b: list[list[int]]
 ) -> list[list[int]]:
     """The operation is two validated list of ranges
      (ranges_a - ranges_b).
@@ -240,8 +240,8 @@ def range_difference(
 
 
 def range_addition(
-    ranges_a: list[Optional[list[int]]],
-    ranges_b: list[Optional[list[int]]]
+    ranges_a: list[list[int]],
+    ranges_b: list[list[int]]
 ) -> tuple[list[list[int]], list[list[int]]]:
     """Addition between two validated list of ranges.
      Simulates the addition between two sets.
@@ -280,28 +280,28 @@ def range_addition(
     merged_ranges = list[list[int]]()
     intersections = list[list[int]]()
 
-    top = ordered_ranges[0]
-
-    for tag_range in ordered_ranges[1:]:
-        match top, tag_range:
-            case [a_start, a_end], [b_start, b_end] if (
-                a_end + 1 == b_start
-            ):
-                top = [a_start, b_end]
-            case [a_start, a_end], [b_start, b_end] if (
-                b_start <= a_end and a_end <= b_end
-            ):
-                top = [a_start, b_end]
-                intersections.append([b_start, a_end])
-            case [a_start, a_end], [b_start, b_end] if (
-                b_end < a_end
-            ):
-                intersections.append([b_start, b_end])
-            case _, _:
-                merged_ranges.append(top)
-                top = tag_range
-
-    merged_ranges.append(top)
+    if ordered_ranges:
+        top = ordered_ranges[0]
+
+        for tag_range in ordered_ranges[1:]:
+            match top, tag_range:
+                case [a_start, a_end], [b_start, b_end] if (
+                    a_end + 1 == b_start
+                ):
+                    top = [a_start, b_end]
+                case [a_start, a_end], [b_start, b_end] if (
+                    b_start <= a_end and a_end <= b_end
+                ):
+                    top = [a_start, b_end]
+                    intersections.append([b_start, a_end])
+                case [a_start, a_end], [b_start, b_end] if (
+                    b_end < a_end
+                ):
+                    intersections.append([b_start, b_end])
+                case _, _:
+                    merged_ranges.append(top)
+                    top = tag_range
+        merged_ranges.append(top)
 
     return (
         [