diff --git a/meme_otron/meme_otron.py b/meme_otron/meme_otron.py index 9b7da89..ac2111f 100644 --- a/meme_otron/meme_otron.py +++ b/meme_otron/meme_otron.py @@ -72,14 +72,17 @@ def compute_part(*args: str, input_data: Optional[bytes] = None, if input_data is None or len(input_data) == 0: if len(args) <= 1: return None, 'Image: received no input data nor URL' + elif not utils.validate_url(args[1]): + return None, 'Image: invalid URL format' else: - return None, 'Image: URL mode not yet implemented' + input_data = utils.read_web(args[1]) + if input_data is None: + return None, 'Image: could not reach URL' + img = img_factory.build_image_only(input_data) + if img is None: + return None, 'Image: invalid image format' else: - img = img_factory.build_image_only(input_data) - if img is None: - return None, 'Image: invalid image format' - else: - return img, None + return img, None else: meme = meme_db.get_meme(meme_id) if meme is None: