initial

server.py

@@ -0,0 +1,93 @@
+#!/usr/bin/python3
+import os
+import flask
+import argparse
+import sys
+import PIL.Image
+
+app = flask.Flask("Picture factory app", static_folder=None)
+PICTURE_DIR = "pictures/"
+
+def generatePicture(pathToOrig, scaleX, scaleY, encoding):
+    '''Generate an pictures with the requested scales and encoding if it doesn't already exist'''
+
+    CACHE_DIR = os.path.join("cache/")
+
+    if os.path.isfile(CACHE_DIR):
+        raise OSError("Picture cache dir is occupied by a file!")
+    if not os.path.isdir(CACHE_DIR):
+        os.mkdir(CACHE_DIR)
+
+    filename, extension = os.path.splitext(os.path.basename(pathToOrig))
+    if not encoding:
+        encoding = extension.strip(".")
+
+    # just python things... #
+    if encoding.lower() == "jpg":
+        encoding = "jpeg"
+
+    # open image #
+    image = PIL.Image.open(os.path.join(PICTURE_DIR, pathToOrig))
+
+    # ensure sizes are valid #
+    x, y = image.size
+    if not scaleY:
+        scaleY = y
+    if not scaleX:
+        scaleX = x
+    scaleX = min(x, scaleX)
+    scaleY = min(y, scaleY)
+
+    # generate new paths #
+    newFile = "x-{x}-y-{y}-{fname}.{ext}".format(x=scaleX, y=scaleY, fname=filename, ext=encoding)
+    newPath = os.path.join(CACHE_DIR, newFile)
+
+    # save image with new size and encoding #
+    if image.mode in ("RGBA", "P") and encoding in ("jpeg", "webp"):
+        image = image.convert("RGB")
+    image.thumbnail((scaleX, scaleY), PIL.Image.ANTIALIAS)
+    image.save(newPath, encoding)
+
+    # strip the STATIC_DIR because we will use send_from_directory for safety #
+    REPLACE_ONCE = 1
+    return newPath.replace(PICTURE_DIR, "", REPLACE_ONCE)
+
+@app.route("/media/<path:path>")
+def sendPicture(path):
+    cache_timeout = 2592000
+
+    y1 = flask.request.args.get("scaley")
+    x1 = flask.request.args.get("scalex")
+    y2 = flask.request.args.get("y")
+    x2 = flask.request.args.get("x")
+
+    # check variables #
+    scaleY, scaleX = (None, None)
+    if y1:
+        scaleY = round(float(y1))
+    elif y2:
+        scaleY = round(float(y2))
+
+    if x1:
+        scaleX = round(float(x1))
+    elif y1:
+        scaleX = round(float(x2))
+    
+    path = generatePicture(path, scaleX, scaleY, flask.request.args.get("encoding"))
+    raw = flask.send_from_directory(".", path, cache_timeout=cache_timeout)
+    response = flask.make_response(raw)
+    response.headers['X-ATHQ-INTERNAL-FID'] = path
+    return response
+
+if __name__ == "__main__":
+
+    parser = argparse.ArgumentParser(description='Picture Factory',
+                        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+
+    # general parameters #
+    parser.add_argument("-i", "--interface", default="127.0.0.1", help="Interface to listen on")
+    parser.add_argument("-p", "--port",      default="5000",      help="Port to listen on")
+
+    # startup #
+    args = parser.parse_args()
+    app.run(host=args.interface, port=args.port)