154 lines
5.0 KiB
Python
154 lines
5.0 KiB
Python
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
|
|
from os.path import join
|
|
from ext import *
|
|
from config import *
|
|
from aztec_code_generator import AztecCode
|
|
from io import BytesIO
|
|
|
|
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
|
|
|
|
app.blueprint([room_bp,propic_bp])
|
|
|
|
@app.exception(exceptions.SanicException)
|
|
async def clear_session(request, exception):
|
|
tpl = app.ctx.tpl.get_template('error.html')
|
|
response = html(tpl.render(exception=exception))
|
|
|
|
if exception.status_code == 403:
|
|
del response.cookies["foxo_code"]
|
|
del response.cookies["foxo_secret"]
|
|
return response
|
|
|
|
@app.before_server_start
|
|
async def main_start(*_):
|
|
print(">>>>>> main_start <<<<<<")
|
|
app.ctx.tpl = Environment(loader=FileSystemLoader("tpl"), autoescape=True)
|
|
app.ctx.tpl.globals.update(time=time)
|
|
app.ctx.tpl.globals.update(int=int)
|
|
app.ctx.tpl.globals.update(len=len)
|
|
|
|
app.ctx.order_cache = {}
|
|
|
|
@app.route("/manage/barcode/<code>")
|
|
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("/manage/nosecount")
|
|
async def nose_count(request, order: Order):
|
|
p = 0
|
|
orders = {}
|
|
async with httpx.AsyncClient() as client:
|
|
while 1:
|
|
p += 1
|
|
res = await client.get(join(base_url, f"orders/?include_canceled_positions=false&page={p}"), headers=headers)
|
|
|
|
if res.status_code == 404: break
|
|
|
|
data = res.json()
|
|
for o in data['results']:
|
|
orders[o['code']] = Order(o)
|
|
|
|
orders = {key:value for key,value in sorted(orders.items(), key=lambda x: len(x[1].room_members), reverse=True)}
|
|
|
|
tpl = app.ctx.tpl.get_template('nosecount.html')
|
|
return html(tpl.render(orders=orders, order=order))
|
|
|
|
@app.route("/furizon/beyond/order/<code>/<secret>/open/<secret2>")
|
|
async def redirect_explore(request, code, secret, order: Order, secret2=None):
|
|
|
|
response = 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, f"orders/{code}/"), headers=headers)
|
|
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!")
|
|
response.cookies['foxo_code'] = code
|
|
response.cookies['foxo_secret'] = secret
|
|
return response
|
|
|
|
@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!")
|
|
|
|
pending_roommates = []
|
|
if order.pending_roommates:
|
|
for pr in order.pending_roommates:
|
|
if not pr: continue
|
|
pending_roommates.append(await get_order(code=pr, insecure=True))
|
|
|
|
room_members = []
|
|
if order.room_id:
|
|
if order.room_id != order.code:
|
|
room_owner = await get_order(code=order.room_id, insecure=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 get_order(code=member_id, insecure=True))
|
|
|
|
room_sizes = {1: 0, 2: 0, 3: 0, 4: 0, 5: 0}
|
|
for o in order:
|
|
room_sizes[len(o.room_members)] += 1
|
|
|
|
|
|
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, 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/logout")
|
|
async def logour(request):
|
|
raise exceptions.Forbidden("You have been logged out.", status_code=403)
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0", port=8188, dev=DEV_MODE)
|