from rest_framework import status
from rest_framework.exceptions import ValidationError

from api.models import EsDepartment, EsInvigilator, EsLocation, EsSuitability, EsTag
from api.models.audit_trail import AuditTrail
from api.models.audit_trail_details import AuditTrailDetails
from api.translation import __
from api.utils import get_add_new_data, get_exception_detail, get_ip, get_next_available_code, log_critical_error
from api.validator import BaseValidator
from api.views.admin.base import AdminApiBase


class SuitabilityCreate(AdminApiBase):
    def validate_request(self, request):
        data = request.data.copy()
        if "description" in data and "desc" not in data:
            data["desc"] = data.get("description")
        if "department" in data and "department_id" not in data:
            data["department_id"] = data.get("department")

        status_code_string = ",".join(str(v) for v in EsSuitability.STATUS_TO_CODE.values())

        # if data.get("tag_id") is not None and "tag" not in data:
        #     data["tag"] = data.get("tag_id")

        rules = {
            "code": "nullable|not_exists:api.EsSuitability,code",
            "name": "required",
            "desc": "nullable",
            "department_id": "nullable|exists:api.EsDepartment,id",
            "status": "required|in:" + status_code_string,
            "invigilator_id": "nullable|array|exists:api.EsInvigilator,id",
            "location_id": "nullable|array|exists:api.EsLocation,id",
            "tag_id": "nullable|array|exists:api.EsTag,id",
            "tag": "nullable|array|exists:api.EsTag,id",
        }

        attribute = {
        }

        validator = BaseValidator(data, rules, attribute)
        error = validator.validate()
        if error:
            raise ValidationError(error)

    def post(self, request):
        try:
            self.validate_request(request)
            data = request.data.copy()
            if "description" in data and "desc" not in data:
                data["desc"] = data.get("description")
            if "department" in data and "department_id" not in data:
                data["department_id"] = data.get("department")
            if "tag_id" in data and "tag" not in data:
                data["tag"] = data.get("tag_id")

            old_new_data = {
                "old_data": None,
                "new_data": None,
            }

            action = "suitability_create"

            idata = {}
            ignore_field = ["invigilator_id", "location_id", "tag_id"]
            valid_fields = {f.column for f in EsSuitability._meta.fields}

            for key, val in data.items():
                if key not in ignore_field and key in valid_fields:
                    idata[key] = val

            code = data.get("code")
            if not code:
                idata["code"] = get_next_available_code(EsSuitability, "code", "MV", 10)
            idata["created_by"] = request.user.id

            suitability = EsSuitability.objects.create(**idata)

            old_new_data["new_data"] = get_add_new_data(suitability)

            new_ids = request.data.get("invigilator_id", [])
            old_new_data = self.update_m2m_field(suitability, "invigilator", None, new_ids, old_new_data)

            new_ids = request.data.get("location_id", [])
            old_new_data = self.update_m2m_field(suitability, "location", None, new_ids, old_new_data)

            new_ids = data.get("tag", [])
            old_new_data = self.update_m2m_field(suitability, "tag", None, new_ids, old_new_data, False, True)

            audit_trail = AuditTrail.objects.create(
                user_id=request.user.id,
                type=self.audit_type,
                ip_address=get_ip(request)
            )
            remark_param = {
                "code": suitability.code,
            }

            AuditTrailDetails.custom_insert(
                audit_trail=audit_trail,
                action=action,
                remark_param=remark_param,
                new_data=old_new_data["new_data"],
            )
            response = {}

            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)