from collections import defaultdict
from rest_framework import serializers, status

from api.models import TtStaffSuitability, TtDepartment, TtZone, TtAvailability, TtStartPreference, TtUsagePreference, \
    TtSuitability, TtTag, TtSetting, TtModule, TtWeekPattern, TtWeek, TtAcademicTerm, TtPos, TtPosModuleGroup, \
    TtPosModuleGroupModule
from api.views.admin.base import AdminApiBase
from api.validator import BaseValidator
from api.utils import decrypt_aes_128_cbc, encrypt_aes_128_cbc, log_critical_error, get_exception_detail, \
    calculate_preset_pattern_time
from rest_framework.exceptions import ValidationError
from api.translation import __


def _serialize_pos_module_group_rule(pos_module_group):
    return {
        "id": pos_module_group.id,
        "code": pos_module_group.code,
        "module_group_id": pos_module_group.module_group_id,
        "module_group_code": pos_module_group.module_group.code,
        "name": pos_module_group.module_group.name,
    }


class PosUpdateModuleRules(AdminApiBase):
    def validate_request(self,request):
        rules = {
            "id": "required|exists:api.TtPos,id",
        }
        # if need rename field, can put here
        attribute = {
            # "email": __("attr.email"),
        }
        # validate id first, cause need get pos 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)

            pos_id = request.data.get("id")
            pos = (
                TtPos.objects
                .select_related("department", "academic_term")
                # .prefetch_related(
                #     # "module_group__module",
                #     # "tag__tag"
                #     # tags means the relation add in models, __tag means the relation table tag_id get the tag object #2
                # )
                .filter(id=pos_id)
                .first()
            )
            pos_module_groups = TtPosModuleGroup.objects.filter(pos_id=pos_id).select_related("module_group")
            pos_module_group_ids = list(pos_module_groups.values_list("id", flat=True))
            pos_module_group_modules = TtPosModuleGroupModule.objects.filter(pos_module_group_id__in=pos_module_group_ids).select_related("module")
            info = {
                "code": pos.code,
                "name": pos.name,
                "desc": pos.desc,
                "department_id": pos.department_id,
                "academic_term_id": pos.academic_term_id,
                "planned_size": pos.planned_size,
                "max_credit": pos.max_credit,
            }

            module_info = []
            for pos_module_group in pos_module_groups:
                # print(pos_module_group)
                append_data = {
                    "pos_module_group_id": pos_module_group.id,
                    # "name": pos_module_group.module_group.name,
                    "module": [],
                }
                module_info.append(append_data)

            # make it key by id
            module_info_map = {g["pos_module_group_id"]: g for g in module_info}
            for pos_module_group_module in pos_module_group_modules:
                pos_module_group_id = pos_module_group_module.pos_module_group_id
                if pos_module_group_id in module_info_map:
                    module_info_map[pos_module_group_id]["module"].append(pos_module_group_module.module_id)

            info['modules'] = list(module_info_map.values())
            data = {}

            data['module'] = TtModule.objects.filter(academic_term=info['academic_term_id'],status=TtModule.STATUS_TO_CODE['active']).order_by("name").values("id","code","name","planned_size","real_size","credit_provided")
            data['pos_module_group'] = []
            for pos_module_group in pos_module_groups:
                data['pos_module_group'].append(
                    _serialize_pos_module_group_rule(pos_module_group)
                )

            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)