from django.db import connection
from rest_framework import serializers, status
from api.models import TtUsagePreference, TtConstraintProfile, TtSetting, TtResourceBreak, TtMaximumWorkspan, \
    TtMaximumHour, \
    TtTravelTable, TtFreeBlock, UserConstraintSetting
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, get_next_available_code, get_start_and_end_slot
from rest_framework.exceptions import ValidationError
from api.translation import __
from api.views.medical.base import MedicalApiBase
from backend.kafka import send_request
from backend.redis_client import redis_client
from django.db.models import Prefetch

class UserConstraintSettingUpdate(MedicalApiBase):
    def validate_request(self,request):
        status_code_string = ",".join(str(v) for v in TtConstraintProfile.STATUS_TO_CODE.values())

        # FE will request by pattern, so can reduce the changes from FE, and BE will convert this pattern become start_slot and end_slot
        rules = {
            "user_constraint_setting": "required|array",
        }

        # 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
        if "user_constraint_setting" in request.data:
            # {"id": "avoid_double_booking_student_set", "mode": 1},
            category_as_key = {}
            required_fields = {"id", "mode"}
            user_constraint_settings = request.data.get("user_constraint_setting")
            available_constraint_setting = UserConstraintSetting.TYPE
            available_mode = UserConstraintSetting.MODE_TO_CODE.values()
            for user_constraint_setting in user_constraint_settings:
                if not required_fields.issubset(user_constraint_setting.keys()):
                    raise ValidationError({
                        "error": __('validation.invalid', field=__("attr.user_constraint_settings")),
                        "errors": {"user_constraint_settings": __('validation.invalid', field=__("attr.user_constraint_settings"))}
                    })

                if user_constraint_setting.get("mode") not in available_mode:
                    # FE passing invalid mode, throw error
                    raise ValidationError({
                        "error": __('validation.invalid', field=__("attr.user_constraint_setting")),
                        "errors": {"user_constraint_setting": __('validation.invalid', field=__("attr.user_constraint_setting"))}
                    })

                if user_constraint_setting.get("id") not in available_constraint_setting:
                    # FE passing invalid id, throw error
                    raise ValidationError({
                        "error": __('validation.invalid', field=__("attr.user_constraint_setting")),
                        "errors": {"user_constraint_setting": __('validation.invalid', field=__("attr.user_constraint_setting"))}
                    })

    def post(self, request):
        # Validate input
        try:
            self.validate_request(request)
            old_new_data = {
                "old_data": [],
                "new_data": [],
            }
            action="user_constraint_setting_update" # this is for audit log only
            user = request.user

            current_setting = UserConstraintSetting.get_setting(user.id)
            # current_setting data is prepare return for FE, so using "id" for type
            current_setting_lookup = {item["id"]: {"type":item["id"], "mode": item["mode"]} for item in current_setting}
            input_constraint_setting = request.data.get("user_constraint_setting")
            create_update_data = []
            for constraint_setting in input_constraint_setting:
                input_type = constraint_setting["id"]
                mode = constraint_setting["mode"]

                skip_audit = False
                if(current_setting_lookup[input_type]["mode"] == mode):
                    skip_audit = True

                if not skip_audit:
                    old_new_data['old_data'].append({
                        "type": current_setting_lookup[input_type]["type"],
                        "mode": current_setting_lookup[input_type]["mode"],
                    })

                create_update_data.append(
                    UserConstraintSetting(
                        user_id=user.id,
                        type=input_type,
                        mode=mode
                    )
                )

                if not skip_audit:
                    old_new_data['new_data'].append({
                        "type": input_type,
                        "mode": mode,
                    })
            UserConstraintSetting.objects.bulk_create(
                create_update_data,
                update_conflicts=True,
                unique_fields=["user", "type"],# if user_id and type match, will update
                update_fields=["mode"] # only update mode column
            )

            # 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":user.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,
                old_data=old_new_data['old_data'],
                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)