from collections import defaultdict

from django.db.models import Prefetch
from rest_framework import serializers, status
from api.models import TtLocation, TtLocationSuitability, TtSetting, TtModule, TtWeekPattern, TtAvailability, \
    TtStartPreference, TtUsagePreference, 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 ModuleCreate(AdminApiBase):
    def validate_request(self,request,settings):
        slot_per_week = settings["slot_per_week"]

        rules = {
            "academic_term_id": "required|exists:api.TtAcademicTerm,id",
            "code": "nullable|not_exists:api.TtModule,code",
            "name": "required",
            "desc": "nullable",
            "department_id": "nullable|exists:api.TtDepartment,id",
            "planned_size": "required|integer",
            "credit_provided": "required|integer",
            "week_pattern_id": "nullable|exists:api.TtWeekPattern,id",
            "week_pattern": "nullable|array|exists:api.TtWeek,id",
            "availability_id": "nullable|exists:api.TtAvailability,id",
            "availability_pattern": "nullable|char_count:" + slot_per_week + "|regex:[01]+",
            "start_preference_id": "nullable|exists:api.TtStartPreference,id",
            "start_preference_pattern": "nullable|char_count:" + slot_per_week + "|regex:[1-9]+",
            "usage_preference_id": "nullable|exists:api.TtUsagePreference,id",
            "usage_preference_pattern": "nullable|char_count:" + slot_per_week + "|regex:[1-9]+",
            "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
        # 2025-10-01 PMO say no need check too much in pattern first, if not take too much time to do it, wait until got feedback only depends how to check
        # if using pattern, the pattern must follow default rules

    def post(self, request):
        # Validate input
        try:
            setting_params = {
                "slot_per_week",
                "minute_per_slot",
            }
            settings = TtSetting.get_multiple_setting(setting_params)
            minute_per_slot = int(settings["minute_per_slot"])
            self.validate_request(request,settings)

            academic_term_id = request.data.get("academic_term_id")
            academic_term = TtAcademicTerm.objects.filter(id=academic_term_id).prefetch_related(Prefetch("week", queryset=TtWeek.objects.order_by("week"))).first()

            default_availability_pattern = academic_term.default_availability
            default_start_preference_pattern = academic_term.default_start
            default_usage_preference_pattern = academic_term.default_usage

            default_week_pattern = [w.id for w in academic_term.week.all()]
            week_pattern_for_redis = convert_to_redis_week(list([w.week for w in academic_term.week.all()]))

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

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

            # this 3 pattern if got give id, force pattern to null
            availability_id = request.data.get("availability_id")
            availability_pattern = request.data.get("availability_pattern")
            if availability_id:
                availability_pattern = None

            start_preference_id = request.data.get("start_preference_id")
            start_preference_pattern = request.data.get("start_preference_pattern")
            if start_preference_id:
                start_preference_pattern = None

            usage_preference_id = request.data.get("usage_preference_id")
            usage_preference_pattern = request.data.get("usage_preference_pattern")
            if usage_preference_id:
                usage_preference_pattern = None

            idata = {}
            # not insert to module table
            ignore_field = ['tag','timestamp','signature']
            valid_fields = {f.column for f in TtModule._meta.fields}

            for key, val in request.data.items():
                if key not in ignore_field:
                    match key:
                        case "availability_pattern" | "start_preference_pattern" | "usage_preference_pattern":
                            # got extra handle the value for these key, so the param same name as the key, so can using this code to handle
                            idata[key] = locals()[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 this 2 value, the loop wont add default pattern, so having this code to handle default pattern
            if not availability_id and not availability_pattern:
                idata['availability_pattern'] = default_availability_pattern
            if not usage_preference_id and not usage_preference_pattern:
                idata['usage_preference_pattern'] = default_usage_preference_pattern
            if not start_preference_id and not start_preference_pattern:
                idata['start_preference_pattern'] = default_start_preference_pattern
            # 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(TtModule, "code", "MV", 10)
            idata['status'] = TtModule.STATUS_TO_CODE['active']
            idata['created_by'] = request.user.id

            module = TtModule.objects.create(**idata)

            old_new_data['new_data']=get_add_new_data(module)

            # custom week pattern store in relation, so need handle like this
            # for create part, no need have a if else to check got pass in or not, if null just update as null only
            week_pattern_id = request.data.get("week_pattern_id")
            week_pattern = request.data.get("week_pattern")
            if week_pattern_id:
                week_pattern = None
                week_pattern_for_redis = []
            else :
                if week_pattern:
                    week_pattern_for_redis = convert_to_redis_week(list(TtWeek.objects.filter(id__in=week_pattern).values_list("week",flat=True)))
                    old_new_data = self.update_m2m_field(module,'week',None,week_pattern,old_new_data)
                else:
                    old_new_data = self.update_m2m_field(module, 'week', None, default_week_pattern, old_new_data)

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

            redis_module_table = "module"
            # from here make it same name as db first, got different, only do extra handle
            redis_module_column = [
                "id","code","name","department_id","academic_term_id","planned_size","real_size","credit_provided",
                "week_pattern_id","week_pattern","availability_id","availability_pattern","start_preference_id",
                "start_preference_pattern","usage_preference_id","usage_preference_pattern"
            ]
            redis_data = {
                "insert": {},
            }
            redis_module_data = {}
            for column in redis_module_column:
                match column:
                    case "week_pattern":
                        redis_module_data[column] = week_pattern_for_redis or []
                    case "usage_preference_id" | "availability_id" | "start_preference_id" | "week_pattern_id":
                        # jose request if no id, dun store null, want to store "0"
                        value = getattr(module, column)
                        redis_module_data[column] = 0 if value in (None,"") else value
                    case _:
                        redis_module_data[column] = getattr(module, column)
            redis_data["insert"].setdefault(redis_module_table, []).append(redis_module_data)

            bulk_sync_to_redis(redis_client, redis_data)

            # kafka push
            method = "module_create"
            kafka_topic = self.kafka_config['MICROSERVICES_TT_TOPIC']
            new_data = old_new_data.get("new_data")
            kafka_module = []
            if new_data and kafka_topic:
                relation_fields = [
                    "week",
                    "tag",
                ]
                kafka_module = generate_kafka_data(new_data, relation_fields)
                kafka_request_data = {
                    "session_id": request.user.name,
                    "module": kafka_module,
                }
                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":module.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)