| | import io
|
| | import re
|
| |
|
| | import flask
|
| | import src.functions.config
|
| | from PIL import Image, ImageDraw, ImageFont
|
| |
|
| | imageBP = flask.Blueprint("image", __name__)
|
| |
|
| |
|
| | @imageBP.route("/api/v1/image/<image_type>")
|
| | async def imageFunction(image_type):
|
| | text = flask.request.args.get("text")
|
| | extention = flask.request.args.get("extention")
|
| | if image_type == "poster":
|
| | img = Image.new("RGB", (342, 513), color=(255, 255, 255))
|
| | draw = ImageDraw.Draw(img)
|
| |
|
| | font_size = 1
|
| | font = ImageFont.truetype(
|
| | font="./build/fonts/Roboto-Regular.ttf", size=font_size
|
| | )
|
| | img_fraction = 0.9
|
| | breakpoint = img_fraction * img.size[0]
|
| | jumpsize = 75
|
| | while True:
|
| | if font.getsize(text)[0] < breakpoint:
|
| | font_size += jumpsize
|
| | else:
|
| | jumpsize = jumpsize // 2
|
| | font_size -= jumpsize
|
| | font = ImageFont.truetype(
|
| | font="./build/fonts/Roboto-Regular.ttf", size=font_size
|
| | )
|
| | if jumpsize <= 1:
|
| | break
|
| |
|
| | width, height = draw.textsize(text, font=font)
|
| | draw.text(
|
| | ((342 - width) / 2, (513 - height) / 2), text, fill="black", font=font
|
| | )
|
| | output = io.BytesIO()
|
| | img.save(output, format=extention)
|
| | output.seek(0, 0)
|
| | return flask.send_file(
|
| | output, mimetype="image/%s" % (extention), as_attachment=False
|
| | )
|
| | elif image_type == "backdrop":
|
| | img = Image.new("RGB", (1280, 720), color=(255, 255, 255))
|
| | draw = ImageDraw.Draw(img)
|
| |
|
| | font_size = 1
|
| | font = ImageFont.truetype(
|
| | font="./build/fonts/Roboto-Regular.ttf", size=font_size
|
| | )
|
| | img_fraction = 0.9
|
| | breakpoint = img_fraction * img.size[0]
|
| | jumpsize = 75
|
| | while True:
|
| | if font.getsize(text)[0] < breakpoint:
|
| | font_size += jumpsize
|
| | else:
|
| | jumpsize = jumpsize // 2
|
| | font_size -= jumpsize
|
| | font = ImageFont.truetype(
|
| | font="./build/fonts/Roboto-Regular.ttf", size=font_size
|
| | )
|
| | if jumpsize <= 1:
|
| | break
|
| |
|
| | width, height = draw.textsize(text, font=font)
|
| | draw.text(
|
| | ((1280 - width) / 2, (720 - height) / 2), text, fill="black", font=font
|
| | )
|
| | output = io.BytesIO()
|
| | img.save(output, format=extention)
|
| | output.seek(0, 0)
|
| | return flask.send_file(
|
| | output, mimetype="image/%s" % (extention), as_attachment=False
|
| | )
|
| | elif image_type == "thumbnail":
|
| | id = flask.request.args.get("id")
|
| | config, drive = src.functions.credentials.refreshCredentials(
|
| | src.functions.config.readConfig()
|
| | )
|
| | params = {
|
| | "fileId": id,
|
| | "fields": "thumbnailLink",
|
| | "supportsAllDrives": True,
|
| | }
|
| | res = drive.files().get(**params).execute()
|
| | if res.get("thumbnailLink"):
|
| | thumbnail = re.sub(r"(s[^s]*)$", "s3840", res["thumbnailLink"])
|
| | return flask.redirect(thumbnail, code=302)
|
| | else:
|
| | return (
|
| | flask.jsonify(
|
| | {
|
| | "code": 500,
|
| | "content": None,
|
| | "message": "The thumbnail does not exist on Google's servers.",
|
| | "success": False,
|
| | }
|
| | ),
|
| | 500,
|
| | )
|
| |
|
