from rest_framework import serializers, status
from api.models import TtUsagePreference, TtWeekPattern, TtAcademicTerm, TtWeek
from api.models.audit_trail import AuditTrail
from api.models.audit_trail_details import AuditTrailDetails
from api.views.admin.base import AdminApiBase
from api.validator import BaseValidator
from api.utils import decrypt_aes_128_cbc, encrypt_aes_128_cbc, generate_kafka_data, log_critical_error, get_exception_detail, \
    get_add_new_data, get_ip, bulk_sync_to_redis, convert_to_redis_week, get_next_available_code
from rest_framework.exceptions import ValidationError
from api.translation import __
from backend.kafka import send_request
from backend.redis_client import redis_client


class WeekPatternCreate(AdminApiBase):
    def validate_request(self,request):
        rules = {
            "code": "nullable|not_exists:api.TtWeekPattern,code",
            "name": "required",
            "academic_term_id": "required|exists:api.TtAcademicTerm,id" ,
            "department_id": "nullable|exists:api.TtDepartment,id",
            "week_id": "required|array|exists:api.TtWeek,id",
            "tag": "nullable|array|exists:api.TtTag,id",
        }

        # if need rename field, can put here
        attribute = {
            # "email": __("attr.email"),
        }

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

        # custom error after basic validation
        # check the week_id is in academic term or not
        academic_term = TtAcademicTerm.objects.filter(id=request.data.get("academic_term_id")).prefetch_related("week").first()
        requested_week_ids = request.data.get("week_id") or []  # array from request
        valid_week_ids = set(academic_term.week.values_list("id", flat=True))
        invalid_ids = [wid for wid in requested_week_ids if wid not in valid_week_ids]
        if invalid_ids:
            error = {
                "error": __("validation.exists",field=__("attr.week_id"), values=invalid_ids),
                "errors": {"week_id":__("validation.exists",field=__("attr.week_id"), values=invalid_ids)},
            }
            raise ValidationError(error)

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

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

            action="week_pattern_create" # this is for audit log only

            idata = {}
            ignore_field = ["week_id"]
            valid_fields = {f.column for f in TtWeekPattern._meta.fields}

            for key, val in request.data.items():
                if key not in ignore_field:
                    match key:
                        case _:
                            if key in valid_fields:#for insert, key can check in default part, cause if got special handle, i can check the way to use key also
                                # default the key will same as db column name
                                idata[key] = val
            # if FE no pass in code or is null or empty, will used default code
            code = request.data.get("code")
            if not code:
                idata["code"] = get_next_available_code(TtWeekPattern, "code", "MV", 10)
            idata['status'] = TtWeekPattern.STATUS_TO_CODE['active']
            idata['created_by'] = request.user.id

            week_pattern = TtWeekPattern.objects.create(**idata)

            old_new_data['new_data']=get_add_new_data(week_pattern)

            new_ids = request.data.get("week_id", [])
            old_new_data = self.update_m2m_field(week_pattern,'week',None,new_ids,old_new_data)
            week_for_redis = []
            if new_ids:
                weeks_data = TtWeek.objects.filter(id__in=new_ids).order_by("week")
                week_name = list(weeks_data.values_list("week",flat=True))
                week_for_redis = convert_to_redis_week(week_name)

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

            redis_week_pattern_table = "week_pattern"
            # from here make it same name as db first, got different, only do extra handle
            redis_week_pattern_column = [
                "id","code","name","week","department_id"
            ]
            redis_data = {
                "insert": {},
            }
            redis_week_pattern_data = {}
            for column in redis_week_pattern_column:
                match column:
                    case "week":
                        redis_week_pattern_data[column] = week_for_redis
                    case _:
                        redis_week_pattern_data[column] = getattr(week_pattern, column)
            redis_data["insert"].setdefault(redis_week_pattern_table, []).append(redis_week_pattern_data)

            bulk_sync_to_redis(redis_client, redis_data)

            # kafka_push
            method = "week_pattern_create"
            kafka_topic = self.kafka_config['MICROSERVICES_TT_TOPIC']
            new_data = old_new_data.get("new_data")
            kafka_week_pattern = []
            if new_data and kafka_topic:
                relation_fields = [
                    "week",
                    "tag",
                ]
                kafka_week_pattern = generate_kafka_data(new_data, relation_fields)
                kafka_request_data = {
                    "session_id": request.user.name,
                    "week_pattern": kafka_week_pattern,
                }
                send_request(kafka_topic, kafka_request_data, None ,method)


            # insert audit trail and details
            # can insert the parent first, then only insert details
            audit_trail = AuditTrail.objects.create(
                user_id=request.user.id,
                type=self.audit_type,
                ip_address=get_ip(request)
            )
            remark_param = {
                "name":week_pattern.name,
            }

            # call to a function for insert details, so in future if want change to used worker, can modify in function only
            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)