def test_valid_id():
ticket = Ticket(title='test',
creator="*****@*****.**",
created=datetime.now())
ticket.save()
abort_if_ticket_id_invalid(ticket.id)
ticket.delete()
class TicketService():
# def get_items(self, payment):
# result = [0, 0, 0, 0, 0]
# order_details = db.session.query(OrderDetails).filter_by(order_id=payment['order_id']).all()
# for item in order_details:
# data = item.as_dict()
# result[data['ticket_id'] - 1] += data['count']
# return result
# def count_ticket_left(self, payments):
# reserved = [0, 0, 0, 0, 0]
# paid = [0, 0, 0, 0, 0]
# for payment in payments:
# data = payment.as_dict()
# if data['transaction_status'] is 'deny':
# continue
# count_data = self.get_items(data)
# for i in range(0, 5):
# if data['transaction_status'] == 'capture':
# paid[i] += count_data[i]
# else:
# reserved[i] += count_data[i]
# return {'reserved': reserved, 'paid': paid}
# def include_ticket_left(self, tickets):
# total_amount = [SLOT['commercial'], SLOT['commercial'], SLOT['commercial'], SLOT['commercial'], SLOT['community']]
# payments = db.session.query(Payment).all()
# data = self.count_ticket_left(payments)
# for i in range(0, 5):
# total_amount[i] -= data['reserved'][i]
# total_amount[i] -= data['paid'][i]
# return {
# 'total_amount': total_amount,
# 'reserved': data['reserved'],
# 'paid': data['paid']
# }
def get(self):
tickets = db.session.query(Ticket).all()
_results = []
for ticket in tickets:
data = ticket.as_dict()
if data['proposal_url'] is not None:
data['proposal_url'] = Helper().url_helper(
data['proposal_url'],
current_app.config['GET_PROPOSAL_DOC_DEST'])
_results.append(data)
return {
'error': False,
'data': _results,
'message': 'Ticket retrieved succesfully'
}
# def include_count_data(self, count_data, results):
# for i in range(0, 5):
# results[i]['reserved'] = count_data['reserved'][i]
# results[i]['paid'] = count_data['paid'][i]
# results[i]['available'] = count_data['total_amount'][i]
# return results
def show(self, id):
ticket = db.session.query(Ticket).filter_by(id=id).first()
result = ticket.as_dict()
if result['proposal_url']:
result['proposal_url'] = Helper().url_helper(
result['proposal_url'],
current_app.config['GET_PROPOSAL_DOC_DEST'])
return result
def create(self, payloads):
self.model_ticket = Ticket()
self.model_ticket.ticket_type = payloads['ticket_type']
self.model_ticket.price = payloads['price']
self.model_ticket.information = payloads['information']
self.model_ticket.type = payloads['type']
self.model_ticket.usd_price = payloads['usd_price']
if payloads['type'] == TICKET_TYPES['exhibitor']:
if payloads['proposal_url']:
proposal_url = self.save_file(
payloads['proposal_url']
) if payloads['proposal_url'] is not None else None
self.model_ticket.proposal_url = proposal_url
db.session.add(self.model_ticket)
try:
db.session.commit()
data = self.model_ticket.as_dict()
if payloads['type'] == TICKET_TYPES['exhibitor'] and data[
'proposal_url']:
data['proposal_url'] = Helper().url_helper(
data['proposal_url'],
current_app.config['GET_PROPOSAL_DOC_DEST'])
print(data['proposal_url'])
return {'error': False, 'data': data}
except SQLAlchemyError as e:
data = e.orig.args
return {'error': True, 'data': data}
def update(self, payloads, id):
try:
self.model_ticket = db.session.query(Ticket).filter_by(id=id)
proposal_url = self.save_file(
payloads['proposal_url']
) if payloads['proposal_url'] is not None else None
self.model_ticket.update({
'ticket_type': payloads['ticket_type'],
'price': payloads['price'],
'information': payloads['information'],
'type': payloads['type'],
'usd_price': payloads['usd_price'],
'quota': payloads['quota'],
'proposal_url': proposal_url,
'updated_at': datetime.datetime.now()
})
db.session.commit()
data = self.model_ticket.first().as_dict()
return {'error': False, 'data': data}
except SQLAlchemyError as e:
data = e.orig.args
return {'error': True, 'data': data}
def delete(self, id):
self.model_ticket = db.session.query(Ticket).filter_by(id=id)
if self.model_ticket.first() is not None:
# delete row
self.model_ticket.delete()
db.session.commit()
return {'error': False, 'data': None}
else:
data = 'data not found'
return {'error': True, 'data': data}
def save_file(self, file, id=None):
if file and Helper().allowed_file(
file.filename,
current_app.config['ALLOWED_PROPOSAL_DOC_EXTENSIONS']):
filename = Helper().time_string() + "_" + file.filename.replace(
" ", "_")
file.save(
os.path.join(current_app.config['POST_PROPOSAL_DOC_DEST'],
filename))
return current_app.config['SAVE_PROPOSAL_DOC_DEST'] + filename
else:
return None
