from sanic import Sanic, response, exceptions from sanic.response import text, html, redirect, raw from jinja2 import Environment, FileSystemLoader from time import time import httpx import re import json import logging from os.path import join from ext import * from config import * from aztec_code_generator import AztecCode from propic import resetDefaultPropic from io import BytesIO from asyncio import Queue import sqlite3 log = logging.getLogger() app = Sanic(__name__) app.static("/res", "res/") app.ext.add_dependency(Order, get_order) app.ext.add_dependency(Quotas, get_quotas) from room import bp as room_bp from propic import bp as propic_bp from karaoke import bp as karaoke_bp from export import bp as export_bp from stats import bp as stats_bp from api import bp as api_bp from carpooling import bp as carpooling_bp from checkin import bp as checkin_bp from admin import bp as admin_bp app.blueprint([room_bp, karaoke_bp, propic_bp, export_bp, stats_bp, api_bp, carpooling_bp, checkin_bp, admin_bp]) @app.exception(exceptions.SanicException) async def clear_session(request, exception): tpl = app.ctx.tpl.get_template('error.html') r = html(tpl.render(exception=exception)) if exception.status_code == 403: r.delete_cookie("foxo_code") r.delete_cookie("foxo_secret") return r @app.before_server_start async def main_start(*_): print(">>>>>> main_start <<<<<<") app.config.REQUEST_MAX_SIZE = 1024 * 1024 * 5 # 5 MB app.ctx.om = OrderManager() if FILL_CACHE: log.info("Filling cache!") await app.ctx.om.updateCache() log.info("Cache fill done!") app.ctx.nfc_counts = sqlite3.connect('data/nfc_counts.db') app.ctx.login_codes = {} app.ctx.tpl = Environment(loader=FileSystemLoader("tpl"), autoescape=True) app.ctx.tpl.globals.update(time=time) app.ctx.tpl.globals.update(PROPIC_DEADLINE=PROPIC_DEADLINE) app.ctx.tpl.globals.update(ITEMS_ID_MAP=ITEMS_ID_MAP) app.ctx.tpl.globals.update(ITEM_VARIATIONS_MAP=ITEM_VARIATIONS_MAP) app.ctx.tpl.globals.update(ROOM_TYPE_NAMES=ROOM_TYPE_NAMES) app.ctx.tpl.globals.update(PROPIC_MIN_SIZE=PROPIC_MIN_SIZE) app.ctx.tpl.globals.update(PROPIC_MAX_SIZE=PROPIC_MAX_SIZE) app.ctx.tpl.globals.update(PROPIC_MAX_FILE_SIZE=sizeof_fmt(PROPIC_MAX_FILE_SIZE)) app.ctx.tpl.globals.update(int=int) app.ctx.tpl.globals.update(len=len) @app.route("/manage/barcode/") async def gen_barcode(request, code): aa = AztecCode(code).image(module_size=8, border=2) img = BytesIO() aa.save(img, format='PNG') return raw(img.getvalue(), content_type="image/png") @app.route(f"/{ORGANIZER}/{EVENT_NAME}/order///open/") async def redirect_explore(request, code, secret, order: Order, secret2=None): r = redirect(app.url_for("welcome")) if order and order.code != code: order = None if not order: async with httpx.AsyncClient() as client: res = await client.get(join(base_url_event, f"orders/{code}/"), headers=headers) print(res.json()) if res.status_code != 200: raise exceptions.NotFound("This order code does not exist. Check that your order wasn't deleted, or the link is correct.") res = res.json() if secret != res['secret']: raise exceptions.Forbidden("The secret part of the url is not correct. Check your E-Mail for the correct link, or contact support!") r.cookies['foxo_code'] = code r.cookies['foxo_secret'] = secret return r @app.route("/manage/privacy") async def privacy(request): tpl = app.ctx.tpl.get_template('privacy.html') return html(tpl.render()) @app.route("/manage/welcome") async def welcome(request, order: Order, quota: Quotas): if not order: raise exceptions.Forbidden("You have been logged out. Please access the link in your E-Mail to login again!") if order.ans("propic_file") is None: await resetDefaultPropic(request, order, False) if order.ans("propic_fursuiter_file") is None: await resetDefaultPropic(request, order, True) pending_roommates = [] if order.pending_roommates: for pr in order.pending_roommates: if not pr: continue pending_roommates.append(await app.ctx.om.get_order(code=pr, cached=True)) room_members = [] if order.room_id: if order.room_id != order.code: room_owner = await app.ctx.om.get_order(code=order.room_id, cached=True) else: room_owner = order room_members.append(room_owner) for member_id in room_owner.ans('room_members').split(','): if member_id == room_owner.code: continue if member_id == order.code: room_members.append(order) else: room_members.append(await app.ctx.om.get_order(code=member_id, cached=True)) tpl = app.ctx.tpl.get_template('welcome.html') return html(tpl.render(order=order, quota=quota, room_members=room_members, pending_roommates=pending_roommates)) @app.route("/manage/download_ticket") async def download_ticket(request, order: Order): if not order: raise exceptions.Forbidden("You have been logged out. Please access the link in your E-Mail to login again!") if not order.status != 'confirmed': raise exceptions.Forbidden("You are not allowed to download this ticket.") async with httpx.AsyncClient() as client: res = await client.get(join(base_url_event, f"orders/{order.code}/download/pdf/"), headers=headers) if res.status_code == 409: raise exceptions.SanicException("Your ticket is still being generated. Please try again later!", status_code=res.status_code) elif res.status_code == 403: raise exceptions.SanicException("You can download your ticket only after the order has been confirmed and paid. Try later!", status_code=400) return raw(res.content, content_type='application/pdf') @app.route("/manage/admin") async def admin(request, order: Order): await request.app.ctx.om.updateCache() if not order: raise exceptions.Forbidden("You have been logged out. Please access the link in your E-Mail to login again!") if not order.isAdmin(): raise exceptions.Forbidden("Birichino :)") tpl = app.ctx.tpl.get_template('admin.html') return html(tpl.render(order=order)) @app.route("/manage/logout") async def logour(request): raise exceptions.Forbidden("You have been logged out.") if __name__ == "__main__": app.run(host="0.0.0.0", port=8188, dev=DEV_MODE, access_log=ACCESS_LOG)