siteweb/drupal2spip_lal
7
1
Fork 0
Code Pull Requests Releases Wiki Activity

feat(convert): averti en cas d extension de fichier suspecte

Browse Source
This commit is contained in:
François Poulain 2020-08-15 18:36:38 +02:00
parent 75eae59ee4
commit ae576d1a38
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
drupal2spip_lal/base/convert.py
View File

@ -86,8 +86,16 @@ def download(src, filename, force_download):
return path return path
def get_extension(path):
ext = path.split('.')[-1]
if len(ext) < 2 or len(ext) > 4:
logger.warn('Extension curieuse: {}'.format(path))
ext = None
return ext or 'unknown'
def fetch_document(src, filename, force_download): def fetch_document(src, filename, force_download):
extension = filename.split('.')[-1] or 'unknown' extension = get_extension(filename)
directory = os.path.join(settings.SPIP_LOGO_DIR, extension) directory = os.path.join(settings.SPIP_LOGO_DIR, extension)
if not os.access(directory, os.F_OK): if not os.access(directory, os.F_OK):
try: try:
@ -110,8 +118,8 @@ def fetch_and_remove_logo(article, force_download):
SPIP gère les logos à la façon d'un hack : un fichier dans IMG nommé SPIP gère les logos à la façon d'un hack : un fichier dans IMG nommé
'arton{}.{}'.format(article.pk, ext) 'arton{}.{}'.format(article.pk, ext)
""" """
ext = src.split('.')[-1] extension = get_extension(src)
filename = 'arton{}.{}'.format(article.pk, ext) filename = 'arton{}.{}'.format(article.pk, extension)
download(src, filename, force_download) download(src, filename, force_download)
def remove_img(img): def remove_img(img):