from sanic.response import html, redirect, text from sanic import Blueprint, exceptions from random import choice from ext import * from config import headers import os from image_util import generate_room_preview, get_room from utils import confirm_room_by_order bp = Blueprint("room", url_prefix="/manage/room") @bp.post("/create") async def room_create_post(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!") error = None name = request.form.get('name') if len(name) > 64 or len(name) < 4: error = "Your room name is invalid. Please try another one." if order.room_id: error = "You are already in another room. You need to delete (if it's yours) or leave it before creating another." if order.daily: raise exceptions.BadRequest("You cannot create a room if you have a daily ticket!") if not error: await order.edit_answer('room_name', name) await order.edit_answer('room_id', order.code) await order.edit_answer('room_members', order.code) await order.edit_answer('room_secret', ''.join(choice('0123456789') for _ in range(6))) await order.send_answers() return redirect('/manage/welcome') tpl = request.app.ctx.tpl.get_template('create_room.html') return html(tpl.render(order=order, error=error)) @bp.route("/create") async def room_create(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 order.daily: raise exceptions.BadRequest("You cannot create a room if you have a daily ticket!") tpl = request.app.ctx.tpl.get_template('create_room.html') return html(tpl.render(order=order)) @bp.route("/delete") async def delete_room(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.room_owner: raise exceptions.BadRequest("You are not allowed to delete room of others.") if order.ans('room_confirmed'): raise exceptions.BadRequest("You are not allowed to change your room after it has been confirmed.") if len(order.room_members) > 1: raise exceptions.BadRequest("You can only delete a room once there is nobody else inside.") await order.edit_answer('room_name', None) await order.edit_answer('room_id', None) await order.edit_answer('room_members', None) await order.edit_answer('room_secret', None) await order.send_answers() remove_room_preview (order.code) return redirect('/manage/welcome') @bp.post("/join") async def join_room(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 order.pending_room: raise exceptions.BadRequest("There is already a pending join request. Wait for the room owner to accept or refuse it.") if order.room_id: raise exceptions.BadRequest("You are in another room already. Why would you join another?") if order.daily: raise exceptions.BadRequest("You cannot join a room if you have a daily ticket!") code = request.form.get('code').strip() room_secret = request.form.get('room_secret').strip() if (not code) or (not room_secret): raise exceptions.BadRequest("The code or pin you provided are not valid.") room_owner = await request.app.ctx.om.get_order(code=code) if not room_owner: raise exceptions.BadRequest("The code you provided is not valid.") if room_owner.room_secret != room_secret: raise exceptions.BadRequest("The code or pin you provided is not valid.") if room_owner.room_confirmed: raise exceptions.BadRequest("The room you're trying to join has been confirmed already") if room_owner.bed_in_room != order.bed_in_room: raise exceptions.BadRequest("This room's ticket is of a different type than yours!") #if room_owner.pending_roommates and (order.code in room_owner.pending_roommates): #raise exceptions.BadRequest("What? You should never reach this check, but whatever...") await order.edit_answer('pending_room', code) await order.send_answers() pending_roommates = room_owner.pending_roommates if not order.code in pending_roommates: pending_roommates.append(order.code) await room_owner.edit_answer('pending_roommates', ','.join(pending_roommates)) await room_owner.send_answers() remove_room_preview (code) return redirect('/manage/welcome') @bp.route("/kick/") async def kick_member(request, code, 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 order.room_confirmed: raise exceptions.BadRequest("You cannot kick people out of confirmed rooms") if not order.room_owner: raise exceptions.BadRequest("You cannot kick people if you're not the room owner") to_kick = await request.app.ctx.om.get_order(code=code) if to_kick.room_id != order.code: raise exceptions.BadRequest("You cannot kick people of other rooms") await to_kick.edit_answer('room_id', None) await order.edit_answer('room_members', ','.join([x for x in order.room_members if x != to_kick.code]) or None) await order.send_answers() await to_kick.send_answers() remove_room_preview (order.code) return redirect('/manage/welcome') @bp.route("/renew_secret") async def renew_secret(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.room_id: raise exceptions.BadRequest("What room were you even trying to renew?") if not order.room_owner: raise exceptions.BadRequest("You are not allowed to renew rooms of others.") await order.edit_answer('room_secret', ''.join(choice('0123456789') for _ in range(6))) await order.send_answers() return redirect('/manage/welcome') @bp.route("/cancel_request") async def cancel_request(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.pending_room: raise exceptions.BadRequest("There is no pending room.") room_owner = await request.app.ctx.om.get_order(code=order.pending_room) pending_roommates = room_owner.pending_roommates if order.code in pending_roommates: pending_roommates.remove(order.code) await room_owner.edit_answer('pending_roommates', ','.join(pending_roommates) if pending_roommates else None) await room_owner.send_answers() await order.edit_answer('pending_room', None) await order.send_answers() return redirect('/manage/welcome') @bp.route("/approve/") async def approve_roomreq(request, code, 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.room_owner: raise exceptions.BadRequest("You are not the owner of the room!") if not code in order.pending_roommates: raise exceptions.BadRequest("You cannot accept people that didn't request to join your room") if order.room_confirmed: raise exceptions.BadRequest("You cannot accept people to a confirmed room.") pending_member = await request.app.ctx.om.get_order(code=code) if pending_member.room_id: raise exceptions.BadRequest("You cannot accept people who are in a room.") if pending_member.pending_room != order.code: raise exceptions.BadRequest("You cannot accept people who are in another room or waiting to accept another request.") await pending_member.edit_answer('room_id', order.code) await pending_member.edit_answer('pending_room', None) await order.edit_answer('room_members', ','.join([*order.room_members, pending_member.code])) await order.edit_answer('pending_roommates', (','.join([x for x in order.pending_roommates if x != pending_member.code]) or None)) await pending_member.send_answers() await order.send_answers() remove_room_preview(order.code) return redirect('/manage/welcome') @bp.route("/leave") async def leave_room(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.room_id: raise exceptions.BadRequest("You cannot leave a room without being in it.") if order.room_confirmed: raise exceptions.BadRequest("You cannot leave a confirmed room.") if order.room_id == order.code: raise exceptions.BadRequest("You cannot leave your own room.") room_owner = await request.app.ctx.om.get_order(code=order.room_id) await room_owner.edit_answer('room_members', (','.join([x for x in room_owner.room_members if x != order.code]) or None)) await order.edit_answer('room_id', None) await room_owner.send_answers() await order.send_answers() remove_room_preview (order.room_id) return redirect('/manage/welcome') @bp.route("/reject/") async def reject_roomreq(request, code, 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.room_owner: raise exceptions.BadRequest("You are not the owner of the room!") if not code in order.pending_roommates: raise exceptions.BadRequest("You cannot reject people that didn't request to join your room") if order.room_confirmed: raise exceptions.BadRequest("You cannot reject people to a confirmed room.") pending_member = await request.app.ctx.om.get_order(code=code) if pending_member.room_id: raise exceptions.BadRequest("You cannot reject people who are in a room.") if pending_member.pending_room != order.code: raise exceptions.BadRequest("You cannot reject people who are in another room or waiting to accept another request.") await pending_member.edit_answer('pending_room', None) await order.edit_answer('pending_roommates', (','.join([x for x in order.pending_roommates if x != pending_member.code]) or None)) await pending_member.send_answers() await order.send_answers() return redirect('/manage/welcome') @bp.post("/rename") async def rename_room(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.room_owner: raise exceptions.BadRequest("You are not the owner of the room!") if not order.room_id: raise exceptions.BadRequest("Try joining a room before renaming it.") if order.room_confirmed: raise exceptions.BadRequest("You can't rename a confirmed room!") if order.room_id != order.code: raise exceptions.BadRequest("You are not allowed to rename rooms of others.") name = request.form.get('name') if len(name) > 64 or len(name) < 4: raise exceptions.BadRequest("Your room name is invalid. Please try another one.") await order.edit_answer("room_name", name) await order.send_answers() remove_room_preview(order.code) return redirect('/manage/welcome') @bp.route("/confirm") async def confirm_room(request, order: Order, quotas: Quotas): 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.room_id: raise exceptions.BadRequest("Try joining a room before confirming it.") if order.room_id != order.code: raise exceptions.BadRequest("You are not allowed to confirm rooms of others.") # This is not needed anymore you buy tickets already #if quotas.get_left(len(order.room_members)) == 0: # raise exceptions.BadRequest("There are no more rooms of this size to reserve.") confirm_room_by_order(order, request) return redirect('/manage/welcome') 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): logger.exception(str(ex)) @bp.route("/view/") async def get_view(request, code): room_file_name = f"res/rooms/{code}.jpg" room_data = await get_room(request, code) if not room_data: raise exceptions.NotFound("No room was found with that code.") if not os.path.exists(room_file_name): await generate_room_preview(request, code, room_data) tpl = request.app.ctx.tpl.get_template('view_room.html') return html(tpl.render(preview=room_file_name, room_data=room_data))