furizon_webint/ext.py

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']]
		self.secret = data['secret']

		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 = key_from_value(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 = key_from_value(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.room_issues = []
		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 is not None and self.room_id is not None and self.code.strip() == self.room_id.strip())
		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 set_room_errors (self, to_set):
			for s in to_set: self.room_issues.append (s)

	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):
				if TYPE_OF_QUESTIONS[ans['question']] == QUESTION_TYPES["multiple_choice_from_list"]: # if multiple choice
					identifier = ans['question_identifier']
					if self.ans(identifier) == "": #if empty answer 
						await self.edit_answer(identifier, None)
				# 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("[TIME] 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 load_items()
		await load_questions()
		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))
			self.lastCacheUpdate = time()
			for o in self.cache.values():
				if o.code == o.room_id:
					print(o.room_name)
					await validate_room(None, o, self)
	
	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