소스 검색

[fix] detect image extension from content headers if extension isn't of image type

version-14
Anand Doshi 9 년 전
부모
커밋
138c7ad012
1개의 변경된 파일7개의 추가작업 그리고 0개의 파일을 삭제
  1. +7
    -0
      frappe/core/doctype/file/file.py

+ 7
- 0
frappe/core/doctype/file/file.py 파일 보기

@@ -21,6 +21,7 @@ import os
import requests
import requests.exceptions
import StringIO
import mimetypes, imghdr

class FolderNotEmpty(frappe.ValidationError): pass

@@ -146,6 +147,12 @@ class File(NestedSet):

image = Image.open(StringIO.StringIO(r.content))
filename, extn = self.file_url.rsplit("/", 1)[1].rsplit(".", 1)

mimetype = mimetypes.guess_type(filename + "." + extn)[0]
if mimetype is None or not mimetype.startswith("image/"):
# detect file extension by reading image header properties
extn = imghdr.what(filename + "." + extn, h=r.content)

filename = "/files/" + strip(urllib.unquote(filename))

thumbnail = ImageOps.fit(


불러오는 중...
취소
저장