from rest_framework import status

from api.models import EsZone, EsSuitability
from api.models.es_location import EsLocation
from api.models.es_location_unavailable_seat import EsLocationUnavailableSeat
from api.validator import BaseValidator
from api.views.admin.base import AdminApiBase
from api.utils import log_critical_error, get_exception_detail
from rest_framework.exceptions import ValidationError
from api.translation import __

class LocationRules(AdminApiBase):
    def validate_request(self,request):
        rules = {
            "id": "nullable|array|exists:api.EsLocation,id",
        }
        # if need rename field, can put here
        attribute = {
            # "email": __("attr.email"),
        }
        # validate id first, cause need get by id, if wrong id direct return error
        validator = BaseValidator(request.data,rules,attribute)
        error = validator.validate()
        if error:
            raise ValidationError(error)
        # custom error after basic validation

        if error:
            raise ValidationError(error)

    def post(self, request):
        # Validate input
        try:
            self.api_log_skip_outgoing_data=True
            self.validate_request(request)

            ids = request.data.get("id")
            if not ids:
                ids = []
            obj = None
            if isinstance(ids, (list, tuple)) and len(ids) == 1:
                obj = (
                    EsLocation.objects
                    .select_related("zone")
                    .prefetch_related("eslocationunavailableseat_set", "suitability")
                    .filter(id__in=ids)
                    .first()
                )

            info = None
            if obj:
                info={}
                info["id"] = obj.id
                info["code"] = obj.code
                info["name"] = obj.name
                info["desc"] = obj.desc
                info["seat_order"] = obj.seat_order
                info["seat_order_text"] = __("attr.location_seat_order." + str(obj.seat_order))
                info["row"] = obj.row
                info["column"] = obj.column
                info["is_partition"] = obj.is_partition
                info['zone'] = obj.zone.name if obj.zone else None
                info['zone_id'] = obj.zone_id
                info["status"] = obj.status
                info["status_text"] = __("attr.location_status." + str(obj.status))
                info["suitability"] = obj.suitability.all().values_list("id", flat=True)

                info['unavailable_seats'] = [
                    {
                        "type": unavailable_seat.type,
                        "type_text": __(f"attr.location_unavailable_seat_type.{unavailable_seat.type}"),
                        "x": unavailable_seat.cord_x,
                        "y": unavailable_seat.cord_y
                    }
                    for unavailable_seat in obj.eslocationunavailableseat_set.order_by("cord_x", "cord_y").all()
                ]

                for i in range(1,11):
                    extra_data_field = f"extra_data_{i}"
                    info[extra_data_field] = getattr(obj,extra_data_field,None)

            data = {}
            data['delete_confirmation_api'] = False
            data["allow_remove"] = [
                "desc",
                "seat_order",
                "row",
                "column",
                "is_partition",
                "zone_id",
                "unavailable_seats",
                #"tag",
                *[f"extra_data_{i}" for i in range(1, 11)]
            ]
            data['status'] = [
                {
                    "id": val,
                    "name": __(f"attr.location_status.{val}")
                }
                for key, val in EsLocation.STATUS_TO_CODE.items()
            ]

            data['seat_order'] = [
                {
                    "id": val,
                    "name": __(f"attr.location_seat_order.{val}")
                }
                for key, val in EsLocation.SEAT_ORDER_TO_CODE.items()
            ]

            data['unavailable_seat_type'] = [
                {
                    "id": val,
                    "name": __(f"attr.location_unavailable_seat_type.{val}")
                }
                for key, val in EsLocationUnavailableSeat.TYPE_TO_CODE.items()
            ]

            data['suitability'] = list(
                EsSuitability.objects.filter(status=EsSuitability.STATUS_TO_CODE['active']).values("id", "code", "name")
            )

            models = {
                "zone": (EsZone, ["id", "code", "name"]),
            }
            for key, (model, fields) in models.items():
                data[key] = list(
                    model.objects.filter(status=model.STATUS_TO_CODE['active']).values(*fields)
                )

            response = {
                "info": info,
                "rules": data,
            }

            return self.api_response(data=response)
        except ValidationError as e:
            first_message = e.detail['error']
            errors = e.detail['errors']
            return self.api_response(error=first_message,errors=errors,code=status.HTTP_400_BAD_REQUEST)
        except Exception as e:
            e_details = get_exception_detail(e)
            log_critical_error(user_id=None,descr=e_details['descr'],url=e_details['url'],trace=e_details['trace'])
            return self.api_response(error=__("message.internal_server_error"),code=status.HTTP_500_INTERNAL_SERVER_ERROR)