Rewrite ifs to structural pattern matching for Python 3.10+

Signed-off-by: Sergey Vasilyev <nolar@nolar.info>

Author: nolar

looptime/math.py

@@ -34,80 +34,92 @@ def __float__(self) -> float:
     #
 
     def __eq__(self, other: object) -> bool:
-        if isinstance(other, (int, float)):
-            return round(self._value * self.__rr) == round(other * self.__rr)
-        else:
-            return NotImplemented
+        match other:
+            case int() | float():
+                return round(self._value * self.__rr) == round(other * self.__rr)
+            case _:
+                return NotImplemented
 
     def __ne__(self, other: object) -> bool:
-        if isinstance(other, (int, float)):
-            return round(self._value * self.__rr) != round(other * self.__rr)
-        else:
-            return NotImplemented
+        match other:
+            case int() | float():
+                return round(self._value * self.__rr) != round(other * self.__rr)
+            case _:
+                return NotImplemented
 
     def __ge__(self, other: object) -> bool:
-        if isinstance(other, (int, float)):
-            return round(self._value * self.__rr) >= round(other * self.__rr)
-        else:
-            return NotImplemented
+        match other:
+            case int() | float():
+                return round(self._value * self.__rr) >= round(other * self.__rr)
+            case _:
+                return NotImplemented
 
     def __gt__(self, other: object) -> bool:
-        if isinstance(other, (int, float)):
-            return round(self._value * self.__rr) > round(other * self.__rr)
-        else:
-            return NotImplemented
+        match other:
+            case int() | float():
+                return round(self._value * self.__rr) > round(other * self.__rr)
+            case _:
+                return NotImplemented
 
     def __le__(self, other: object) -> bool:
-        if isinstance(other, (int, float)):
-            return round(self._value * self.__rr) <= round(other * self.__rr)
-        else:
-            return NotImplemented
+        match other:
+            case int() | float():
+                return round(self._value * self.__rr) <= round(other * self.__rr)
+            case _:
+                return NotImplemented
 
     def __lt__(self, other: object) -> bool:
-        if isinstance(other, (int, float)):
-            return round(self._value * self.__rr) < round(other * self.__rr)
-        else:
-            return NotImplemented
+        match other:
+            case int() | float():
+                return round(self._value * self.__rr) < round(other * self.__rr)
+            case _:
+                return NotImplemented
 
     #
     # Arithmetics:
     #
 
     def __add__(self, other: object) -> float:
-        if isinstance(other, (int, float)):
-            return (round(self._value * self.__rr) + round(other * self.__rr)) / self.__rr
-        else:
-            return NotImplemented
+        match other:
+            case int() | float():
+                return (round(self._value * self.__rr) + round(other * self.__rr)) / self.__rr
+            case _:
+                return NotImplemented
 
     def __sub__(self, other: object) -> float:
-        if isinstance(other, (int, float)):
-            return (round(self._value * self.__rr) - round(other * self.__rr)) / self.__rr
-        else:
-            return NotImplemented
+        match other:
+            case int() | float():
+                return (round(self._value * self.__rr) - round(other * self.__rr)) / self.__rr
+            case _:
+                return NotImplemented
 
     def __mul__(self, other: object) -> float:
-        if isinstance(other, (int, float)):
-            return (round(self._value * self.__rr) * round(other * self.__rr)) / (self.__rr ** 2)
-        else:
-            return NotImplemented
+        match other:
+            case int() | float():
+                return (round(self._value * self.__rr) * round(other * self.__rr)) / (self.__rr ** 2)
+            case _:
+                return NotImplemented
 
     def __floordiv__(self, other: object) -> float:
-        if isinstance(other, (int, float)):
-            return (round(self._value * self.__rr) // round(other * self.__rr))
-        else:
-            return NotImplemented
+        match other:
+            case int() | float():
+                return (round(self._value * self.__rr) // round(other * self.__rr))
+            case _:
+                return NotImplemented
 
     def __truediv__(self, other: object) -> float:
-        if isinstance(other, (int, float)):
-            return (round(self._value * self.__rr) / round(other * self.__rr))
-        else:
-            return NotImplemented
+        match other:
+            case int() | float():
+                return (round(self._value * self.__rr) / round(other * self.__rr))
+            case _:
+                return NotImplemented
 
     def __mod__(self, other: object) -> float:
-        if isinstance(other, (int, float)):
-            return (round(self._value * self.__rr) % round(other * self.__rr)) / (self.__rr)
-        else:
-            return NotImplemented
+        match other:
+            case int() | float():
+                return (round(self._value * self.__rr) % round(other * self.__rr)) / (self.__rr)
+            case _:
+                return NotImplemented
 
     # See the StdLib's comments on pow() on why it is Any, not float.
     def __pow__(self, power: float, modulo: None = None) -> Any:

looptime/patchers.py

@@ -30,14 +30,15 @@ def patch_event_loop(
         **kwargs: Any,
 ) -> loops.LoopTimeEventLoop:
     result: loops.LoopTimeEventLoop
-    if isinstance(loop, loops.LoopTimeEventLoop):
-        return loop
-    else:
-        new_class = make_event_loop_class(loop.__class__)
-        loop.__class__ = new_class
-        loop = cast(loops.LoopTimeEventLoop, loop)
-        loop.setup_looptime(**kwargs)
-        return loop
+    match loop:
+        case loops.LoopTimeEventLoop():
+            return loop
+        case _:
+            new_class = make_event_loop_class(loop.__class__)
+            loop.__class__ = new_class
+            loop = cast(loops.LoopTimeEventLoop, loop)
+            loop.setup_looptime(**kwargs)
+            return loop
 
 
 def new_event_loop(**kwargs: Any) -> loops.LoopTimeEventLoop:

looptime/timeproxies.py

@@ -23,10 +23,11 @@ def __repr__(self) -> str:
         return f"<Loop time: {self._value}>"
 
     def __matmul__(self, other: object) -> LoopTimeProxy:
-        if isinstance(other, asyncio.AbstractEventLoop):
-            return self.__class__(other)
-        else:
-            return NotImplemented
+        match other:
+            case asyncio.AbstractEventLoop():
+                return self.__class__(other)
+            case _:
+                return NotImplemented
 
     @property
     def _value(self) -> float: