Pythonでディレクトリ内のファイルをモデルに保存する方法

July 08, 2022

本題の前に、前回記事にてサーバーにアップロードした webm ファイルを ffmpeg を使って mp4 に変換する際に、subprocess.call('', shell=True)のコードを紹介しました。

ウェブカメラで撮影した動画(webmファイル)のdurationがInfinityになる問題

前回記事のコードはこちらになります。

import os
import subprocess as sp

input_file = ''
file_name = os.path.splitext(os.path.basename(input_file))[0]

cmd_list = ['ffmpeg', '-i', input_file, file_name + '.mp4']
cmd = ' '.join(cmd_list)
sp.call(cmd, shell=True)

subprocess におけるshell=Trueとshell=Falseの挙動が違うようですが、shell=Trueよりもshell=Falseのほうが安全なようなので、次のように変更しました。

input_file = ''
file_name = os.path.splitext(os.path.basename(input_file))[0]

cmd_list = ['ffmpeg', '-i', input_file, file_name + '.mp4']
sp.call(cmd, shell=False)

以下の記事によると、subprocess に渡す args パラメータは、str もしくは list で、一般には list が推奨とのこと。

また、shell=Trueのときは str、shell=Falseのときは list を使用するようです。

subprocess で shell=True でリストを与えたときの挙動

Python: subprocess call with shell=False not working

それで本題ですが、ファイルの変換にてmp4ファイルが作られますが、これを MoviePyで処理をした後、モデルに保存したい。

MoviePy で処理してできたファイルは、一時的なディレクトリに保存されているものとします。

from django.core.files import File

(略)

# webmファイルと一緒に送られてきたデータをコピー
copied = request.POST.copy()
copied_file = request.FILES.copy()

copied["user"] = request.user

f = open('mp4ファイルのパス',　mode='rb')

copied_file["video"] = F(f)

form = SampleForm(copied, copied_file)
if form.is_valid():
    form.save()
else:
    print("ERROR")
f.close()
    
# 一時的に保存されたファイルは`os.remove()`で削除する。

【django】画像アップロードして保存前にリサイズやサニタイズ処理を実行する方法

from django.core.files import File の File オブジェクトについて。

Internally, Django uses a django.core.files.File instance any time it needs to represent a file.

Most of the time you’ll use a File that Django’s given you (i.e. a file attached to a model as above, or perhaps an uploaded file).

If you need to construct a File yourself, the easiest way is to create one using a Python built-in file object:
from django.core.files import File

#Create a Python file object using open()
f = open('/path/to/hello.world', 'w')
myfile = File(f)
https://django.readthedocs.io/en/stable/topics/files.html

これで、ウェブカメラで撮影した動画をサーバーに保存できるようになりました。