from unittest import skip from django.db import connection from rest_framework import serializers, status from api.models import TtUsagePreference, TtConstraintProfile, TtSetting, TtResourceBreak, TtMaximumWorkspan, \ TtMaximumHour, \ TtTravelTable, TtFreeBlock, UserConstraintSetting, UserSchedulingPreference 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 backend.kafka import send_request from backend.redis_client import redis_client from django.db.models import Prefetch class UserSchedulingPreferenceUpdate(AdminApiBase): 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_scheduling_preference": "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_scheduling_preference" in request.data: # required column in array required_fields = {"id", "preference", "status"} user_scheduling_preferences = request.data.get("user_scheduling_preference") available_type = UserSchedulingPreference.TYPE available_preference = UserSchedulingPreference.PREFERENCE available_status = UserSchedulingPreference.STATUS_TO_CODE.values() for user_scheduling_preference in user_scheduling_preferences: if not required_fields.issubset(user_scheduling_preference.keys()): raise ValidationError({ "error": __('validation.invalid', field=__("attr.user_scheduling_preference")), "errors": {"user_scheduling_preference": __('validation.invalid', field=__("attr.user_scheduling_preference"))} }) if user_scheduling_preference.get("preference") not in available_preference: raise ValidationError({ "error": __('validation.invalid', field=__("attr.user_scheduling_preference")), "errors": {"user_scheduling_preference": __('validation.invalid', field=__("attr.user_scheduling_preference"))} }) if user_scheduling_preference.get("id") not in available_type: # FE passing invalid id, throw error raise ValidationError({ "error": __('validation.invalid', field=__("attr.user_scheduling_preference")), "errors": {"user_scheduling_preference": __('validation.invalid', field=__("attr.user_scheduling_preference"))} }) if user_scheduling_preference.get("status") not in available_status: # FE passing invalid id, throw error raise ValidationError({ "error": __('validation.invalid', field=__("attr.user_scheduling_preference")), "errors": {"user_scheduling_preference": __('validation.invalid', field=__("attr.user_scheduling_preference"))} }) def post(self, request): # Validate input try: self.validate_request(request) old_new_data = { "old_data": [], "new_data": [], } action="user_scheduling_preference_update" # this is for audit log only user = request.user current_setting = UserSchedulingPreference.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"], "preference": item["preference"], "status": item["status"]} for item in current_setting} input_scheduling_preference = request.data.get("user_scheduling_preference") create_update_data = [] for scheduling_preference in input_scheduling_preference: input_type = scheduling_preference["id"] preference = scheduling_preference["preference"] input_status = scheduling_preference["status"] skip_audit = False if(current_setting_lookup[input_type]["preference"] == preference and current_setting_lookup[input_type]["status"] == input_status): skip_audit = True if not skip_audit: old_new_data['old_data'].append({ "type": current_setting_lookup[input_type]["type"], "preference": current_setting_lookup[input_type]["preference"], "status": current_setting_lookup[input_type]["status"], }) create_update_data.append( UserSchedulingPreference( user_id=user.id, type=input_type, preference=preference, status=input_status ) ) if not skip_audit: old_new_data['new_data'].append({ "type": input_type, "preference": preference, "status": input_status }) UserSchedulingPreference.objects.bulk_create( create_update_data, update_conflicts=True, unique_fields=["user", "type"],# if user_id and type match, will update update_fields=["preference","status"] # only update preference and status 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)