Celery 处理任务和修改模型字段

Question

我想使用 ffmpeg 和 celery 将视频转换为 mp4 用于异步任务。当用户上传视频时，它将用于original_video并保存。之后，我希望 celery 将其转换为 mp4_720 字段的不同版本。但是，我对如何使用 celery 应用该逻辑感到困惑。

app.models.py：

class Video(models.Model):
    title = models.CharField(max_length=75)
    pubdate = models.DateTimeField(default=timezone.now)
    original_video = models.FileField(upload_to=get_upload_file_name)
    mp4_720 = models.FileField(upload_to=get_upload_file_name, blank=True, null=True)
    converted = models.BooleanField(default=False)

app.views.py：

def upload_video(request):
    if request.POST:
        form = VideoForm(request.POST, request.FILES)
        if form.is_valid():
            video = form.save(commit=False)
            video.save()

            // Celery to convert the video
            convert_video.delay(video)

            return HttpResponseRedirect('/')
    else:
        form = VideoForm()
    return render(request, 'upload_video.html', {
        'form':form
    })

app.tasks.py：

@app.task
def convert_video(video):

    // Convert the original video into required format and save it in the mp4_720 field using the following command:
    //subprocess.call('ffmpeg -i (path of the original_video) (video for mp4_720)')

    // Change the converted boolean field to True

    // Save

基本上我的问题是如何将转换后的视频保存在 mp4_720 中。非常感谢您的帮助和指导。谢谢。

** 更新**

我希望该方法首先转换 video.original_video，然后将转换后的视频保存在 video.mp4_720 字段中。如果一切都已正确完成，请将 video.converted 更改为 True。我如何定义这样做的方法？

Answer 1

首先，您可能不想将 video 对象传递给 celery - 有关详细信息，请参阅 this 问题。

所以你想这样称呼它：

        convert_video.delay(video.id)

然后

logger = logging.getLogger(__name__)  # assuming you have set up logging elsewhere

@app.task
def convert_video(video_id):
    video = Video.objects.get(video_id)

    cmd = ['ffmpeg',  '-i', video.original_video.path, video.mp4_720.path]
    log.info('running %s', ' '.join(cmd))
    proc = subprocess.Popen(cmd)
    proc.subprocess.wait()

    if p.returncode != 0:
        log.error('command failed with ret val %s', p.returncode)
        log.info(p.stderr)
        log.info(p.stdout)
    else:
        video.converted = True
        video.save()
        log.info('video converted ok')