python - img.save(imgstore, "GIF") 或 img.save(imgstore, "JPEG")

Question

if request.method == 'POST':
        form = DocumentForm(request.POST, request.FILES)
        if form.is_valid():            
            u_file = request.FILES['docfile']            
            filename = u_file.name
            img = Image.open(u_file)
            imgstore = StringIO.StringIO("")
            img.save(imgstore, "GIF")

我正在上传一个转换为文本的图像文件。但是我只能对 GIF 文件执行此操作。或者如果我输入“JPEG”而不是 GIF，我只能保存 JPEG 文件。如何使用 img.save(imgstore, "GIF, JPEG") 选项同时保存 JPEG 和 GIF。谢谢

Answer 1

if form.is_valid():            
    u_file = request.FILES['docfile']            

    extension = u_file.split(".").lower()[-1]

    filename = u_file.name
    img = Image.open(u_file)
    imgstore = StringIO.StringIO("")

    if extension == "gif"
        img.save(imgstore, "GIF")
    else:
        img.save(imgstore, "JPEG")

Answer 2

我认为您可以找到使用 libmagic 获得的文件类型。然后从那里决定。 https://github.com/ahupp/python-magic