[wip3] Room preview generation + OpenGraph tags

This commit is contained in:
Andrea 2024-01-18 01:03:48 +01:00
parent 123a1dc3c5
commit 6fe87b6ea1
7 changed files with 96 additions and 21 deletions

2
.gitignore vendored
View File

@ -154,6 +154,8 @@ cython_debug/
res/propic/propic_* res/propic/propic_*
res/rooms/*
# PyCharm # PyCharm
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can # JetBrains specific template is maintained in a separate JetBrains.gitignore that can
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore

View File

@ -86,6 +86,12 @@ CATEGORIES_LIST_MAP = {
'rooms': [], 'rooms': [],
'dailys': [] 'dailys': []
} }
SPONSORSHIP_COLOR_MAP = {
'super': (251, 140, 0),
'normal': (142, 36, 170)
}
# Create a bunch of "room" items which will get added to the order once somebody gets a room. # Create a bunch of "room" items which will get added to the order once somebody gets a room.
# Map item_name -> room capacity # Map item_name -> room capacity
ROOM_CAPACITY_MAP = { ROOM_CAPACITY_MAP = {

Binary file not shown.

77
room.py
View File

@ -3,7 +3,8 @@ from sanic import Blueprint, exceptions
from random import choice from random import choice
from ext import * from ext import *
from config import headers from config import headers
from PIL import Image from PIL import Image, ImageDraw, ImageFont
import textwrap
import os import os
jobs = [] jobs = []
@ -67,6 +68,7 @@ async def delete_room(request, order: Order):
await order.edit_answer('room_members', None) await order.edit_answer('room_members', None)
await order.edit_answer('room_secret', None) await order.edit_answer('room_secret', None)
await order.send_answers() await order.send_answers()
remove_room_preview (order.code)
return redirect('/manage/welcome') return redirect('/manage/welcome')
@bp.post("/join") @bp.post("/join")
@ -116,7 +118,7 @@ async def join_room(request, order: Order):
await room_owner.edit_answer('pending_roommates', ','.join(pending_roommates)) await room_owner.edit_answer('pending_roommates', ','.join(pending_roommates))
await room_owner.send_answers() await room_owner.send_answers()
remove_room_preview (code)
return redirect('/manage/welcome') return redirect('/manage/welcome')
@bp.route("/kick/<code>") @bp.route("/kick/<code>")
@ -139,7 +141,7 @@ async def kick_member(request, code, order: Order):
await order.send_answers() await order.send_answers()
await to_kick.send_answers() await to_kick.send_answers()
remove_room_preview (order.code)
return redirect('/manage/welcome') return redirect('/manage/welcome')
@bp.route("/renew_secret") @bp.route("/renew_secret")
@ -286,7 +288,7 @@ async def rename_room(request, order: Order):
await order.edit_answer("room_name", name) await order.edit_answer("room_name", name)
await order.send_answers() await order.send_answers()
remove_room_preview (order.code)
return redirect('/manage/welcome') return redirect('/manage/welcome')
@bp.route("/confirm") @bp.route("/confirm")
@ -358,23 +360,59 @@ async def confirm_room(request, order: Order, quotas: Quotas):
async def get_room (request, code): async def get_room (request, code):
order_data = await request.app.ctx.om.get_order(code=code) order_data = await request.app.ctx.om.get_order(code=code)
if not order_data or not order_data.room_owner: return None if not order_data or not order_data.room_owner: return None
members_map = [] members_map = [{'name': order_data.name, 'propic': order_data.propic, 'sponsorship': order_data.sponsorship}]
for member_code in order_data.room_members: for member_code in order_data.room_members:
if member_code == order_data.code: continue
member_order = await request.app.ctx.om.get_order(code=member_code) member_order = await request.app.ctx.om.get_order(code=member_code)
if not member_order: continue if not member_order: continue
members_map.append ({'name': member_order.name, 'propic': member_order.propic, 'sponsorship': member_order.sponsorship}) members_map.append ({'name': member_order.name, 'propic': member_order.propic, 'sponsorship': member_order.sponsorship})
return {'name': order_data.room_name, 'members': members_map} return {'name': order_data.room_name,
'confirmed': order_data.room_confirmed,
'capacity': order_data.room_person_no,
'members': members_map}
async def generate_image(request, code): async def get_room_with_order (request, code):
order_data = await request.app.ctx.om.get_order(code=code)
if not order_data or not order_data.room_owner: return None
def remove_room_preview(code):
preview_file = f"res/rooms/{code}.jpg"
try:
if os.path.exists(preview_file): os.remove(preview_file)
except Exception as ex:
if (EXTRA_PRINTS): print(ex)
async def generate_room_preview(request, code, room_data):
font_path = f'res/font/pt-serif-caption-latin-400-normal.ttf'
main_fill = (16, 149, 193)
jobs.append(code) jobs.append(code)
try: try:
room_data = await get_room(request, code) room_data = await get_room(request, code) if not room_data else room_data
width = 230 * int(room_data['capacity']) + 130
return room_data font = ImageFont.truetype(font_path, 20)
except Exception: with Image.new('RGB', (width, 270), (17, 25, 31)) as to_save:
# Remove fault job i_draw = ImageDraw.Draw(to_save)
if len(jobs) > 0: jobs.pop() # Draw room's name
raise exceptions.SanicException("Could not get that room's data at the moment. Try again, later.", status_code=500) room_name_len = i_draw.textlength(room_data['name'], font)
i_draw.text((((width / 2) - room_name_len / 2), 10), room_data['name'], font=font, fill=main_fill)
# Draw members
for m in range (room_data['capacity']):
member = room_data['members'][m] if m < len(room_data['members']) else { 'name': 'Empty', 'propic': '../new.png', 'sponsorship': None }
font = ImageFont.truetype(font_path, 20)
with Image.open(f'res/propic/{member['propic'] or 'default.png'}') as to_add:
to_save.paste(to_add.resize ((180, 180)), (90 + (230 * m), 45))
name_len = i_draw.textlength(str(member['name']), font)
calc_size = 0
if name_len > 180:
calc_size = 180 * 20 / name_len if name_len > 180 else 20
font = ImageFont.truetype(font_path, calc_size)
name_len = i_draw.textlength(str(member['name']), font)
name_loc = ((90 + (230 * m)) + (90 - name_len / 2), 235 + (calc_size/2))
name_color = SPONSORSHIP_COLOR_MAP[member['sponsorship']] if member['sponsorship'] in SPONSORSHIP_COLOR_MAP.keys() else main_fill
i_draw.text(name_loc, str(member['name']), font=font, fill=name_color)
to_save.save(f'res/rooms/{code}.jpg', 'JPEG')
except Exception as err:
if EXTRA_PRINTS: print(err)
finally: finally:
# Remove fault job # Remove fault job
if len(jobs) > 0: jobs.pop() if len(jobs) > 0: jobs.pop()
@ -383,12 +421,11 @@ async def generate_image(request, code):
@bp.route("/view/<code>") @bp.route("/view/<code>")
async def get_view(request, code): async def get_view(request, code):
if code in jobs:
raise exceptions.SanicException("The room's preview is being generated... Wait a little longer", status_code=409)
room_file_name = f"res/rooms/{code}.jpg" room_file_name = f"res/rooms/{code}.jpg"
room_data = await get_room(request, code)
if not os.path.exists(room_file_name): if not os.path.exists(room_file_name) and code not in jobs:
await generate_image(request, code) await generate_room_preview(request, code, room_data)
tpl = request.app.ctx.tpl.get_template('viewRoom.html') tpl = request.app.ctx.tpl.get_template('view_room.html')
return html(tpl.render(imgPath=room_file_name)) return html(tpl.render(preview=room_file_name, room_data=room_data))

View File

@ -1,6 +1,7 @@
<!doctype html> <!doctype html>
<html> <html>
<head> <head>
{% block head %}{% endblock %}
<meta charset="utf-8" /> <meta charset="utf-8" />
<title>{% block title %}{% endblock %}</title> <title>{% block title %}{% endblock %}</title>
<meta name="viewport" content="width=400rem" /> <meta name="viewport" content="width=400rem" />

20
tpl/view_room.html Normal file
View File

@ -0,0 +1,20 @@
{% extends "base.html" %}
{% block title %}{{room_data['name']}}{% endblock %}
{% block head %}
<!--Open Graph tags here-->
<meta property="og:title" content="{{room_data['name']}}" />
<meta property="og:image:type" content="image/jpeg" />
<meta property="og:image:alt" content="View of a room" />
{% endblock %}
{% block main %}
<main class="container">
<header>
<picture>
<source srcset="/res/furizon.png" media="(prefers-color-scheme:dark)">
<img src="/res/furizon-light.png" style="height:4rem;text-align:center;"
onload="window.location.href='/manage/nosecount';">
</picture>
</header>
</main>
{% endblock %}

View File

@ -1,5 +1,14 @@
{% extends "base.html" %} {% extends "base.html" %}
{% block title %}{{order.name}}'s Booking{% endblock %} {% block title %}{{order.name}}'s Booking{% endblock %}
{% block head %}
<!--Open Graph tags here-->
<meta property="og:title" content="Furizon booking management" />
<meta property="og:image:type" content="image/png" />
<meta property="og:image:alt" content="Furizon's logo" />
<meta property="og:image" content="https://reg.furizon.net/res/furizon.png" />
<meta property="og:image:secure_url" content="https://reg.furizon.net/res/furizon.png" />
{% endblock %}
{% block main %} {% block main %}
<main class="container"> <main class="container">
<header> <header>