无法将模型导入 Django 中的 Celery 任务

Question

我正在尝试配置 Django 和 Celery，当我将任务导入 models.py 文件并同时将模型导入到tasks.py 文件时遇到问题。否则芹菜正在工作。请参阅下面的代码...

core/celery.py

from __future__ import absolute_import, unicode_literals
import os
from celery import Celery
from django.conf import settings


# set the default Django settings module for the 'celery' program.
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'core.settings')

app = Celery('core')
app.config_from_object('django.conf:settings')
app.autodiscover_tasks(lambda: settings.INSTALLED_APPS)


@app.task(bind=True)
def debug_task(self):
    print('Request: {0!r}'.format(self.request))


users/models.py

from django.db import models
from core.tasks import celery_test #this triggers the error

class CustomUser(AbstractUser):
    username = None
    email = models.EmailField(_('email address'), unique=True)


    def __str__(self):
        return self.email


core/tasks.py

from celery.decorators import task
from users.models import CustomUser

@task(name="celery_test_task")
def celery_test_task():
    print(CustomUser)

core/settings/base.py

BROKER_URL = 'redis://localhost:6379'
CELERY_RESULT_BACKEND = 'redis://localhost:6379'
CELERY_ACCEPT_CONTENT = ['application/json']
CELERY_TASK_SERIALIZER = 'json'
CELERY_RESULT_SERIALIZER = 'json'
CELERY_TIMEZONE = 'Africa/Nairobi'

这是错误消息：

File "...\src\users\models.py", line 6, in <module>
    from core.tasks import celery_test
File "...\src\core\tasks.py", line 2, in <module>
    from users.models import CustomUser
ImportError: cannot import name 'CustomUser' from 'users.models' (...\src\users\models.py)

Answer 1

确实，这看起来像是一种循环依赖，但有时它是不可避免的，所以你无法避免它。

通常在这些情况下，将其导入到函数中可能会有所帮助（请参阅 azundo 的评论），但通常这不起作用。

Django 有一个内置的应用程序注册表，您可以访问它来检索本例中加载的模型，请参阅：

# core/tasks.py

from celery.decorators import task
from django.apps import apps

@task(name="celery_test_task") 
def celery_test_task():
    # call apps via Django
    model = apps.get_model(app_label='users', model_name='CustomUser')
    # Now models is accessible, such as model.objects.get_or_create()

请参阅Django refs了解更多信息，您还可以添加更多参数以使其专门适合您的情况。