from dataclasses import dataclass from sanic import Request, exceptions import httpx import re from utils import * from config import * from os.path import join import json from time import time @dataclass class Order: def __init__(self, data): self.time = time() self.data = data self.status = {'n': 'pending', 'p': 'paid', 'e': 'expired', 'c': 'canceled'}[self.data['status']] if not len(self.data['positions']): self.status = 'canceled' self.code = data['code'] self.pending_update = False self.email = data['email'] self.has_card = False self.sponsorship = None self.has_early = False self.has_late = False self.first_name = None self.last_name = None self.country = 'xx' self.address = None self.checked_in = False self.room_type = None self.daily = False self.dailyDays = [] self.room_person_no = 0 self.answers = [] idata = data['invoice_address'] if idata: self.address = f"{idata['street']} - {idata['zipcode']} {idata['city']} - {idata['country']}" self.country = idata['country'] for p in self.data['positions']: if p['item'] in CATEGORIES_LIST_MAP['tickets']: self.position_id = p['id'] self.position_positionid = p['positionid'] self.position_positiontypeid = p['item'] self.answers = p['answers'] for i, ans in enumerate(self.answers): if(TYPE_OF_QUESTIONS[self.answers[i]['question']] == QUESTION_TYPES['file_upload']): self.answers[i]['answer'] = "file:keep" self.barcode = p['secret'] self.checked_in = bool(p['checkins']) if p['item'] in CATEGORIES_LIST_MAP['dailys']: self.daily = True self.dailyDays.append(CATEGORIES_LIST_MAP['dailys'].index(p['item'])) if p['item'] in CATEGORIES_LIST_MAP['memberships']: self.has_card = True if p['item'] == ITEMS_ID_MAP['sponsorship_item']: sponsorshipType = keyFromValue(ITEM_VARIATIONS_MAP['sponsorship_item'], p['variation']) self.sponsorship = sponsorshipType[0].replace ('sponsorship_item_', '') if len(sponsorshipType) > 0 else None if p['attendee_name']: self.first_name = p['attendee_name_parts']['given_name'] self.last_name = p['attendee_name_parts']['family_name'] if p['item'] == ITEMS_ID_MAP['early_arrival_admission']: self.has_early = True if p['item'] == ITEMS_ID_MAP['late_departure_admission']: self.has_late = True if p['item'] == ITEMS_ID_MAP['bed_in_room']: roomTypeLst = keyFromValue(ITEM_VARIATIONS_MAP['bed_in_room'], p['variation']) roomTypeId = roomTypeLst[0] if len(roomTypeLst) > 0 else None self.bed_in_room = p['variation'] self.room_person_no = ROOM_CAPACITY_MAP[roomTypeId] if roomTypeId in ROOM_CAPACITY_MAP else None self.total = float(data['total']) self.fees = 0 self.refunds = 0 for fee in data['fees']: self.fees += float(fee['value']) for ref in data['refunds']: self.refunds += float(ref['amount']) answers = ['payment_provider', 'shirt_size', 'birth_date', 'fursona_name', 'room_confirmed', 'room_id'] self.payment_provider = data['payment_provider'] self.comment = data['comment'] self.phone = data['phone'] self.loadAns() def loadAns(self): self.shirt_size = self.ans('shirt_size') self.is_artist = True if self.ans('is_artist') != 'No' else False self.is_fursuiter = True if self.ans('is_fursuiter') != 'No' else False self.is_allergic = True if self.ans('is_allergic') != 'No' else False self.notes = self.ans('notes') self.badge_id = int(self.ans('badge_id')) if self.ans('badge_id') else None self.propic_locked = self.ans('propic_locked') self.propic_fursuiter = self.ans('propic_fursuiter') self.propic = self.ans('propic') self.carpooling_message = json.loads(self.ans('carpooling_message')) if self.ans('carpooling_message') else {} self.karaoke_songs = json.loads(self.ans('karaoke_songs')) if self.ans('karaoke_songs') else {} self.birth_date = self.ans('birth_date') self.birth_location = self.ans('birth_location') self.name = self.ans('fursona_name') self.room_id = self.ans('room_id') self.room_confirmed = self.ans('room_confirmed') self.room_name = self.ans('room_name') self.pending_room = self.ans('pending_room') self.pending_roommates = self.ans('pending_roommates').split(',') if self.ans('pending_roommates') else [] self.room_members = self.ans('room_members').split(',') if self.ans('room_members') else [] self.room_owner = (self.code == self.room_id) self.room_secret = self.ans('room_secret') self.app_token = self.ans('app_token') self.nfc_id = self.ans('nfc_id') self.can_scan_nfc = True if self.ans('can_scan_nfc') != 'No' else False self.actual_room = self.ans('actual_room') self.staff_role = self.ans('staff_role') self.telegram_username = self.ans('telegram_username').strip('@') if self.ans('telegram_username') else None def __getitem__(self, var): return self.data[var] def ans(self, name): for p in self.data['positions']: for a in p['answers']: if a.get('question_identifier', None) == name: if a['answer'] in ['True', 'False']: return bool(a['answer'] == 'True') return a['answer'] return None def isBadgeValid (self): return self.ans('propic') and (not self.is_fursuiter or self.ans('propic_fursuiter')) def isAdmin (self): return self.code in ADMINS or self.staff_role in ADMINS_PRETIX_ROLE_NAMES async def edit_answer_fileUpload(self, name, fileName, mimeType, data : bytes): if(mimeType != None and data != None): async with httpx.AsyncClient() as client: localHeaders = dict(headers) localHeaders['Content-Type'] = mimeType localHeaders['Content-Disposition'] = f'attachment; filename="{fileName}"' res = await client.post(join(base_url, 'upload'), headers=localHeaders, content=data) res = res.json() await self.edit_answer(name, res['id']) else: await self.edit_answer(name, None) self.loadAns() async def edit_answer(self, name, new_answer): found = False self.pending_update = True for key in range(len(self.answers)): if self.answers[key].get('question_identifier', None) == name: if new_answer != None: print('EXISTING ANSWER UPDATE', name, '=>', new_answer) self.answers[key]['answer'] = new_answer found = True else: print('DEL ANSWER', name, '=>', new_answer) del self.answers[key] break if (not found) and (new_answer is not None): async with httpx.AsyncClient() as client: res = await client.get(join(base_url_event, 'questions/'), headers=headers) res = res.json() for r in res['results']: if r['identifier'] != name: continue print('ANSWER UPDATE', name, '=>', new_answer) self.answers.append({ 'question': r['id'], 'answer': new_answer, 'options': r['options'] }) self.loadAns() async def send_answers(self): async with httpx.AsyncClient() as client: print("POSITION ID IS", self.position_id) for i, ans in enumerate(self.answers): # Fix for karaoke fields if ans['question'] == 40: del self.answers[i]['options'] del self.answers[i]['option_identifiers'] res = await client.patch(join(base_url_event, f'orderpositions/{self.position_id}/'), headers=headers, json={'answers': self.answers}) if res.status_code != 200: for ans, err in zip(self.answers, res.json()['answers']): if err: print('ERROR ON', ans, err) raise exceptions.ServerError('There has been an error while updating this answers.') for i, ans in enumerate(self.answers): if(TYPE_OF_QUESTIONS[self.answers[i]['question']] == QUESTION_TYPES['file_upload']): self.answers[i]['answer'] = "file:keep" self.pending_update = False self.time = -1 self.loadAns() @dataclass class Quotas: def __init__(self, data): self.data = data def get_left(self, capacity): for quota in self.data['results']: if quota['id'] == ROOM_QUOTA_ID[capacity]: return quota['available_number'] return 0 async def get_quotas(request: Request=None): async with httpx.AsyncClient() as client: res = await client.get(join(base_url_event, 'quotas/?order=id&with_availability=true'), headers=headers) res = res.json() return Quotas(res) async def get_order(request: Request=None): await request.receive_body() return await request.app.ctx.om.get_order(request=request) class OrderManager: def __init__(self): self.lastCacheUpdate = 0 self.empty() def empty(self): self.cache = {} self.order_list = [] async def updateCache(self): t = time() if(t - self.lastCacheUpdate > CACHE_EXPIRE_TIME): print("Re-filling cache!") await self.fill_cache() self.lastCacheUpdate = t def add_cache(self, order): self.cache[order.code] = order if not order.code in self.order_list: self.order_list.append(order.code) def remove_cache(self, code): if code in self.cache: del self.cache[code] self.order_list.remove(code) async def fill_cache(self): await loadItems() await loadQuestions() self.empty() p = 0 async with httpx.AsyncClient() as client: while 1: p += 1 res = await client.get(join(base_url_event, f"orders/?page={p}"), headers=headers) if res.status_code == 404: break data = res.json() for o in data['results']: o = Order(o) if o.status in ['canceled', 'expired']: self.remove_cache(o.code) else: self.add_cache(Order(o)) async def get_order(self, request=None, code=None, secret=None, nfc_id=None, cached=False): # if it's a nfc id, just retorn it if nfc_id: for order in self.cache.values(): if order.nfc_id == nfc_id: return order await self.updateCache() # If a cached order is needed, just get it if available if code and cached and code in self.cache and time()-self.cache[code].time < 3600: return self.cache[code] # If it's a request, ignore all the other parameters and just get the order of the requestor if request: code = request.cookies.get("foxo_code") secret = request.cookies.get("foxo_secret") if re.match('^[A-Z0-9]{5}$', code or '') and (secret is None or re.match('^[a-z0-9]{16,}$', secret)): print('Fetching', code, 'with secret', secret) async with httpx.AsyncClient() as client: res = await client.get(join(base_url_event, f"orders/{code}/"), headers=headers) if res.status_code != 200: if request: raise exceptions.Forbidden("Your session has expired due to order deletion or change! Please check your E-Mail for more info.") else: self.remove_cache(code) return None res = res.json() order = Order(res) if order.status in ['canceled', 'expired']: self.remove_cache(order.code) if request: raise exceptions.Forbidden(f"Your order has been deleted. Contact support with your order identifier ({res['code']}) for further info.") else: self.add_cache(order) if request and secret != res['secret']: raise exceptions.Forbidden("Your session has expired due to a token change. Please check your E-Mail for an updated link!") return order