vassili/seitime-frappe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

Merge pull request #989 from anandpdoshi/anand-feb-3

Browse source 
[fix] set random filename if not found, when extracting image from html
This commit is contained in:
Pratik Vyas 2015-02-05 13:33:30 +05:30
commit 346d67736d
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
frappe/utils/file_manager.py
View file

@ -6,7 +6,7 @@ import frappe
import os, base64, re
import hashlib
import mimetypes
from frappe.utils import get_site_path, get_hook_method, get_files_path
from frappe.utils import get_site_path, get_hook_method, get_files_path, random_string
from frappe import _
from frappe import conf
from copy import copy
@ -89,8 +89,16 @@ def extract_images_from_html(doc, fieldname):
def _save_file(match):
data = match.group(1)
data = data.split("data:")[1]
headers, content = data.split(",")
filename = headers.split("filename=")[-1]
if "filename=" in headers:
filename = headers.split("filename=")[-1]
else:
mtype = headers.split(";")[0]
extn = mimetypes.guess_extension(mtype)
filename = random_string(7) + extn
# TODO fix this
file_url = save_file(filename, content, doc.doctype, doc.name, decode=True).get("file_url")
if not frappe.flags.has_dataurl:
@ -99,7 +107,7 @@ def extract_images_from_html(doc, fieldname):
return '<img src="{file_url}"'.format(file_url=file_url)
if content:
content = re.sub('<img\s*src=\s*["\'](data:[^"\']*)["\']', _save_file, content)
content = re.sub('<img\s*src=\s*["\'](.*?)["\']', _save_file, content)
if frappe.flags.has_dataurl:
doc.set(fieldname, content)