chore: add celery types

SELab-2 · Apr 3, 2024 · 9402aca · 9402aca
1 parent 54efb67
commit 9402aca
Show file tree

Hide file tree

Showing 3 changed files with 82 additions and 3 deletions.
diff --git a/backend/poetry.lock b/backend/poetry.lock
diff --git a/backend/pyproject.toml b/backend/pyproject.toml
@@ -6,7 +6,7 @@ authors = ["Topvennie <[email protected]>"]
 readme = "README.md"
 
 [tool.poetry.dependencies]
-python = "^3.12"
+python = "^3.11.4"
 django = "^5.0.3"
 django-sslserver = "^0.22"
 djangorestframework = "^3.15.1"
@@ -20,6 +20,8 @@ celery = {extras = ["redis"], version = "^5.3.6"}
 django-redis = "^5.4.0"
 gunicorn = "^21.2.0"
 whitenoise = "^6.6.0"
+flake8 = "^7.0.0"
+celery-types = "^0.22.0"
 
 
 [build-system]

diff --git a/backend/ypovoli/celery.py b/backend/ypovoli/celery.py
@@ -1,6 +1,9 @@
 import os
 
 from celery import Celery
+from celery.app.task import Task
+
+Task.__class_getitem__ = classmethod(lambda cls, *args, **kwargs: cls)  # type: ignore[attr-defined]
 
 # Set the default Django settings module for the 'celery' program.
 os.environ.setdefault("DJANGO_SETTINGS_MODULE", "ypovoli.settings")