67 lines
1.4 KiB
Python
67 lines
1.4 KiB
Python
|
import os
|
||
|
|
import random
|
||
|
|
from io import BytesIO
|
||
|
|
|
||
|
|
from flask import Flask, render_template, request, send_file
|
||
|
|
from flask_caching import Cache
|
||
|
|
from PIL import Image, ImageOps
|
||
|
|
|
||
|
|
app = Flask(__name__)
|
||
|
|
|
||
|
|
GREY = "G"
|
||
|
|
COLOR = "RGB"
|
||
|
|
|
||
|
|
cache = Cache(config={"CACHE_TYPE": "SimpleCache"})
|
||
|
|
|
||
|
|
app = Flask(__name__)
|
||
|
|
cache.init_app(app)
|
||
|
|
|
||
|
|
|
||
|
|
def get_cropped_image(x, y, grey=False):
|
||
|
|
"""crops a random image from collection"""
|
||
|
|
im_src = random.choice(os.listdir("./images"))
|
||
|
|
im = Image.open(f"images/{im_src}")
|
||
|
|
out = BytesIO()
|
||
|
|
max_x, max_y = im.size
|
||
|
|
if x < max_x or y < max_y:
|
||
|
|
im = ImageOps.fit(im, (x, y))
|
||
|
|
if grey:
|
||
|
|
im = ImageOps.grayscale(im)
|
||
|
|
im.save(out, "WEBP", quality=50)
|
||
|
|
out.seek(0)
|
||
|
|
return out
|
||
|
|
|
||
|
|
|
||
|
|
def make_response(x, y, color_mode=COLOR):
|
||
|
|
im = get_cropped_image(x, y, color_mode == GREY)
|
||
|
|
return send_file(im, mimetype="image/webp")
|
||
|
|
|
||
|
|
|
||
|
|
@app.route("/")
|
||
|
|
def hello():
|
||
|
|
u = request.host
|
||
|
|
return render_template("index.html", url=u)
|
||
|
|
|
||
|
|
|
||
|
|
@app.route("/bookmarklet")
|
||
|
|
def bookmarklet():
|
||
|
|
u = request.host
|
||
|
|
return render_template("bookmarklet.html", url=u)
|
||
|
|
|
||
|
|
|
||
|
|
@app.route("/<int:x>/<int:y>")
|
||
|
|
@cache.cached(10)
|
||
|
|
def generate(x, y):
|
||
|
|
return make_response(x, y, COLOR)
|
||
|
|
|
||
|
|
|
||
|
|
@app.route("/g/<int:x>/<int:y>")
|
||
|
|
@cache.cached(10)
|
||
|
|
def generate_grey(x, y):
|
||
|
|
return make_response(x, y, GREY)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
port = int(os.environ.get("PORT", 5000))
|
||
|
|
app.run(host="0.0.0.0", port=port)
|