diff --git a/imgin/__init__.py b/imgin/__init__.py index 614cd7c..011b4f6 100644 --- a/imgin/__init__.py +++ b/imgin/__init__.py @@ -59,9 +59,9 @@ def gallery(id=''): @route('/') -def hello(img=''): - if img.endswith("jpeg") and not img.endswith("jpg") and not img.endswith("png"): - img += ".jpg" +def img(img=''): + if not img.endswith("jpeg") and not img.endswith("jpg") and not img.endswith("png"): + img = img + ".jpg" if not path.exists(IMAGE_CACHE + img): get(img, IMAGE_CACHE) return static_file(img, root=IMAGE_CACHE)