from flask import Blueprint, jsonify, abort, request
from api import db
from api.helpers import check_api_key
from api.models import Order, OrderField, OrderOrderField

mod = Blueprint("orders", __name__, url_prefix="/orders")

mod.before_request(check_api_key)


@mod.route("/<int:order_id>/pwinty_id", methods=["POST"])
def save_pwinty_id(order_id):
    data = request.json
    order = Order.query.get_or_404(order_id)
    field = OrderField.query.by_name(name="pwinty_id")
    order_field = order.fields.filter_by(fieldid=field.id).first()
    if not order_field:
        order_field = OrderOrderField(orderid=order.id, fieldid=field.id)

    setattr(order_field, "value", data["value"])
    db.session.add(order_field)
    db.session.commit()

    return jsonify(order_field.to_json())