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, TtPathway, TtPathwayPosModuleGroupModule
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 __


class PathwayGenerateRules(AdminApiBase):
    def validate_request(self,request):
        rules = {
            "pos_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("pos_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()
            )

            info = {
                "id": pos.id,
                "code": pos.code,
                "name": pos.name,
            }

            """
            pathway_generate_type
            module_group = [
                {
                    "id": 2
                    "name": "MG1",
                    "modules": [
                        {
                            "pos_module_group_module_id": 13,
                            "module_id": 12,
                            "name": "M1",
                            "credit": 4
                        }
                    ]
                    
                }
            ]
            """
            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")

            module_group_info = []
            for pos_module_group in pos_module_groups:
                # print(pos_module_group)
                append_data = {
                    "pos_module_group_id": pos_module_group.id, # this is is in pos_module_group table, is using this id to calculate pathway, not the exact module_group id to calculate
                    "name": pos_module_group.module_group.name,
                    "module": [],
                }
                module_group_info.append(append_data)

            # make it key by id
            module_group_info_map = {g["pos_module_group_id"]: g for g in module_group_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_group_info_map:
                    module_data = {
                        "pos_module_group_module_id": pos_module_group_module.id,
                        "module_id": pos_module_group_module.module_id,
                        "name": pos_module_group_module.module.name,
                        "code": pos_module_group_module.module.code,
                        "credit": pos_module_group_module.module.credit_provided,
                    }
                    module_group_info_map[pos_module_group_id]["module"].append(module_data)

            data = {}
            data["module_group"] = module_group_info

            data["pathway_generate_type"] = []
            for type in TtPathway.PATHWAY_GENERATE_TYPE:
                data["pathway_generate_type"].append({
                    "code": type,
                    "name": __("attr.pathway_generate_type_name."+type),
                })

            data['generated_pathway'] = list(TtPathway.objects.filter(pos_id=pos_id).values("pathway_string","pathway_string_2"))

            # test = list(TtPathwayPosModuleGroupModule.objects.filter(pos_module_group_module_id__in=[52]).values_list("pathway_id", flat=True))
            # data['test'] = test
            # data['old_pathway_id'] = list(TtPathwayPosModuleGroupModule.objects.filter(pos_module_group_module__pos_module_group_id__in=[20]).values_list("pathway_id", flat=True).distinct())
            # data['new_pathway_id'] = list(TtPathwayPosModuleGroupModule.objects.filter(pos_module_group_module__pos_module_group_id__in=[23]).values_list("pathway_id", flat=True).distinct())
            # data['need_remove'] = list(set(data['old_pathway_id']) - set(data['new_pathway_id']))
            setting_params = {
                "maximum_module_to_auto_calculate_pathway",
            }
            data['based_setting'] = TtSetting.get_multiple_setting(setting_params)
            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)