图像保存（imgstore，“GIF”）或图像保存（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文件。如何使用图像保存（imgstore，“GIF，JPEG”）选项。谢谢

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