Annotate celery/states.py (#7395)

Kludex · web-flow · commit 314d014e6d87 · 2022-04-03T10:02:06.000+06:00
* Annotate `celery/states.py`

* Add comma on pyproject to avoid conflicts

* Include signals.py as well
diff --git a/celery/states.py b/celery/states.py
@@ -78,7 +78,7 @@
 NONE_PRECEDENCE = PRECEDENCE_LOOKUP[None]
 
 
-def precedence(state):
+def precedence(state: str) -> int:
     """Get the precedence index for state.
 
     Lower index means higher precedence.
@@ -110,16 +110,16 @@ class state(str):
         False
     """
 
-    def __gt__(self, other):
+    def __gt__(self, other: str) -> bool:
         return precedence(self) < precedence(other)
 
-    def __ge__(self, other):
+    def __ge__(self, other: str) -> bool:
         return precedence(self) <= precedence(other)
 
-    def __lt__(self, other):
+    def __lt__(self, other: str) -> bool:
         return precedence(self) > precedence(other)
 
-    def __le__(self, other):
+    def __le__(self, other: str) -> bool:
         return precedence(self) >= precedence(other)
 
 
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,4 +15,6 @@ disallow_untyped_defs = true
 ignore_missing_imports = true
 files = [
     "celery/__main__.py",
+    "celery/states.py",
+    "celery/signals.py",
 ]
