feat: Single endpoint for creating creating an order

app/api/custom/orders.py

@@ -1,22 +1,28 @@
+import json
+
 from flask import Blueprint, jsonify, request
 from flask_jwt_extended import current_user, jwt_required
 from flask_limiter.util import get_remote_address
 from sqlalchemy.orm.exc import NoResultFound
 
-
 from app import limiter
 from app.models import db
 from app.api.auth import return_file
-from app.api.helpers.db import safe_query
+from app.api.helpers.db import safe_query, get_count
 from app.api.helpers.mail import send_email_to_attendees
 from app.api.helpers.errors import ForbiddenError, UnprocessableEntityError, NotFoundError
 from app.api.helpers.order import calculate_order_amount, create_pdf_tickets_for_holder
 from app.api.helpers.storage import UPLOAD_PATHS
 from app.api.helpers.storage import generate_hash
 from app.api.helpers.ticketing import TicketingManager
+from app.api.schema.attendees import AttendeeSchema
+from app.api.schema.orders import OrderSchema
 from app.api.helpers.permission_manager import has_access
 from app.models.discount_code import DiscountCode
 from app.models.order import Order
+from app.models.order import OrderTicket
+from app.models.ticket import Ticket
+from app.models.ticket_holder import TicketHolder
 
 order_blueprint = Blueprint('order_blueprint', __name__, url_prefix='/v1/orders')
 ticket_blueprint = Blueprint('ticket_blueprint', __name__, url_prefix='/v1/tickets')
@@ -82,8 +88,11 @@ def resend_emails():
 
 @order_blueprint.route('/calculate-amount', methods=['POST'])
 @jwt_required
-def calculate_amount():
-    data = request.get_json()
+def calculate_amount(*args):
+    if not args:
+        data = request.get_json()
+    else:
+        data = json.loads(args[0])
     tickets = data['tickets']
     discount_code = None
     if 'discount-code' in data:
@@ -93,3 +102,72 @@ def calculate_amount():
             return UnprocessableEntityError({'source': 'discount-code'}, 'Discount Usage Exceeded').respond()
 
     return jsonify(calculate_order_amount(tickets, discount_code))
+
+
+@order_blueprint.route('/create-order', methods=['POST'])
+@jwt_required
+def create_order():
+    data = request.get_json()
+    tickets = data['tickets']
+    attendee = data['attendee']
+    for attribute in attendee:
+        attendee[attribute.replace('-', '_')] = attendee.pop(attribute)
+    schema = AttendeeSchema()
+    json_api_attendee = {"data": {"attributes": data['attendee'], "type": "attendee"}}
+    result = schema.load(json_api_attendee)
+    if result.errors:
+        return jsonify(result.errors)
+    ticket_ids = [int(ticket['id']) for ticket in tickets]
+    ticket_info_list = db.session.query(Ticket).filter(Ticket.id.in_(ticket_ids)).filter_by(event_id=data['event_id'],
+                                                                                            deleted_at=None).all()
+    ticket_ids_found = [ticket_information.id for ticket_information in ticket_info_list]
+    tickets_not_found = list(set(ticket_ids) - set(ticket_ids_found))
+    if tickets_not_found:
+        return jsonify(status='Order Unsuccessful', error='Ticket with id {} was not found.'.format(tickets_not_found))
+    for ticket in tickets:
+        ticket_info = None
+        for ticket_information in ticket_info_list:
+            if ticket_information.id == int(ticket['id']):
+                ticket_info = ticket_information
+        if (ticket_info.quantity - get_count(db.session.query(TicketHolder.id).filter_by(
+              ticket_id=int(ticket['id']), deleted_at=None))) < ticket['quantity']:
+            return jsonify(status='Order Unsuccessful', error='Ticket already sold out.')
+    attendee_list = []
+    for ticket in tickets:
+        for ticket_amount in range(ticket['quantity']):
+            attendee = TicketHolder(**result[0], event_id=int(data['event_id']), ticket_id=int(ticket['id']))
+            db.session.add(attendee)
+            attendee_list.append(attendee)
+    ticket_pricing = calculate_amount(json.dumps(data))
+    if not has_access('is_coorganizer', event_id=data['event_id']):
+        data['status'] = 'initializing'
+    # create on site attendees
+    # check if order already exists for this attendee.
+    # check for free tickets and verified user
+    ticket_pricing = json.loads(ticket_pricing.data)
+    order = Order(amount=ticket_pricing['total_amount'], user_id=current_user.id, event_id=int(data['event_id']),
+                  status=data['status'])
+    db.session.add(order)
+    db.session.commit()
+    db.session.refresh(order)
+    order_tickets = {}
+    for holder in attendee_list:
+        holder.order_id = order.id
+        db.session.add(holder)
+        if not order_tickets.get(holder.ticket_id):
+            order_tickets[holder.ticket_id] = 1
+        else:
+            order_tickets[holder.ticket_id] += 1
+
+    create_pdf_tickets_for_holder(order)
+
+    for ticket in order_tickets:
+        od = OrderTicket(order_id=order.id, ticket_id=ticket, quantity=order_tickets[ticket])
+        db.session.add(od)
+
+    order.quantity = order.tickets_count
+    db.session.add(order)
+    db.session.commit()
+    db.session.refresh(order)
+    order_schema = OrderSchema()
+    return order_schema.dump(order)