indymedia
/
epicyon
mirror of https://gitlab.com/bashrc2/epicyon
7
4
Fork 1
Code Issues 14 Releases Wiki Activity

Check that string looks like a url when downloading image

merge-requests/30/head
Bob Mottram 2022-01-05 15:21:44 +00:00
parent 36507ea846
commit 5341e5a274
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
session.py
View File

@ -441,12 +441,27 @@ def post_image(session, attach_image_filename: str, federation_list: [],
return None return None
def _looks_like_url(url: str) -> bool:
"""Does the given string look like a url
"""
if not url:
return False
if '.' not in url:
return False
if '://' not in url:
return False
return True
def download_image(session, base_dir: str, url: str, def download_image(session, base_dir: str, url: str,
image_filename: str, debug: bool, image_filename: str, debug: bool,
force: bool = False) -> bool: force: bool = False) -> bool:
"""Downloads an image with an expected mime type """Downloads an image with an expected mime type
""" """
if not url: if not _looks_like_url(url):
if debug:
print('WARN: download_image, ' +
url + ' does not look like a url')
return None return None
# try different image types # try different image types
@ -510,7 +525,7 @@ def download_image_any_mime_type(session, url: str,
"""http GET for an image with any mime type """http GET for an image with any mime type
""" """
# check that this looks like a url # check that this looks like a url
if '://' not in url: if not _looks_like_url(url):
if debug: if debug:
print('WARN: download_image_any_mime_type, ' + print('WARN: download_image_any_mime_type, ' +
url + ' does not look like a url') url + ' does not look like a url')