from datetime import datetime from django.utils import timezone from rest_framework import status from rest_framework.exceptions import ValidationError from api.models import EsExamPeriod, EsExamPeriodUnavailability from api.models.audit_trail import AuditTrail from api.models.audit_trail_details import AuditTrailDetails from api.translation import __ from api.utils import log_critical_error, get_exception_detail, get_ip from api.validator import BaseValidator from api.views.admin.base import AdminApiBase def _parse_clock(value): for fmt in ("%H:%M:%S", "%H:%M"): try: return datetime.strptime(value, fmt).time() except (TypeError, ValueError): continue return None class ExamPeriodUpdate(AdminApiBase): def validate_request(self,request): rules = { "id": "required|array|exists:api.EsExamPeriod,id", } # if need rename field, can put here attribute = { # "email": __("attr.email"), } # validate id first validator = BaseValidator(request.data,rules,attribute) error = validator.validate() if error: raise ValidationError(error) edit_obj = ( EsExamPeriod.objects .prefetch_related("esexamperiodunavailability_set") .filter(id__in=request.data.get("id")) ) obj = edit_obj.first() status_code_string = ",".join(str(v) for v in EsExamPeriod.STATUS_TO_CODE.values()) rules = { "name": "nullable", "start_date": "nullable|date", "end_date": "nullable|date", "start_time": "nullable|time", "end_time": "nullable|time", "unavailable_dates": "nullable|array", "status": "nullable|in:"+status_code_string, } if request.data.get("code") and edit_obj.count() == 1 and request.data.get("code") != obj.code: rules['code'] = "nullable|not_exists:api.EsExamPeriod,code" validator = BaseValidator(request.data,rules,attribute) error = validator.validate() if error: raise ValidationError(error) unavailable_dates = request.data.get("unavailable_dates", []) if request.data.get("start_date") or request.data.get("end_date") or unavailable_dates: #must have start and end date when update rules = { "start_date": "required|date", "end_date": "required|date", } validator = BaseValidator(request.data, rules, attribute) error = validator.validate() if error: raise ValidationError(error) start_date = datetime.strptime(request.data['start_date'], "%Y-%m-%d") end_date = datetime.strptime(request.data['end_date'], "%Y-%m-%d") if start_date > end_date: message = __("validation.gt.date", field=__("attr.end_date"), value=request.data['start_date']) raise ValidationError({ 'error': message, 'errors': { 'end_date': message }, }) if unavailable_dates: rules = { "unavailable_dates": "required|date", } for unavailable_date in unavailable_dates: validator = BaseValidator({ "unavailable_dates": unavailable_date }, rules, attribute) error = validator.validate() if error: raise ValidationError(error) check_date = datetime.strptime(unavailable_date, "%Y-%m-%d") if check_date < start_date or check_date > end_date: message = __("validation.between.date", field=__("attr.unavailable_date"), start=request.data['start_date'], end=request.data['end_date']) raise ValidationError({ 'error': message, 'errors': { 'unavailable_dates': message }, }) start_raw = request.data.get("start_time") end_raw = request.data.get("end_time") if start_raw and end_raw: start_time = _parse_clock(start_raw) end_time = _parse_clock(end_raw) if start_time > end_time: message = __("validation.gt.date", field=__("attr.end_time"), value=start_raw) raise ValidationError({ 'error': message, 'errors': { 'end_time': message }, }) elif start_raw or end_raw: for obj in edit_obj: start_time = _parse_clock(start_raw) if start_raw else obj.start_time end_time = _parse_clock(end_raw) if end_raw else obj.end_time if start_time and end_time and start_time > end_time: message = __("validation.gt.date", field=__("attr.end_time"), value=str(start_time)) raise ValidationError({ 'error': message, 'errors': { 'end_time': message }, }) return edit_obj def post(self, request): # Validate input try: edit_obj = self.validate_request(request) action="exam_period_update" # this is for audit log only old_new_data = { "new_data": {}, "old_data": {}, } updated = False update_fields = {} # if more than 1 role edit, ignore code if edit_obj.count() == 1: current_edit_obj = edit_obj.first() if "code" in request.data: old_new_data.setdefault("old_data", {}).setdefault(current_edit_obj.id, {})["code"] = current_edit_obj.code old_new_data.setdefault("new_data", {}).setdefault(current_edit_obj.id, {})["code"] = request.data.get("code") update_fields['code'] = request.data['code'] updated = True ignore_fields = ['id','timestamp','signature','unavailable_dates'] valid_fields = {f.column for f in EsExamPeriod._meta.fields} for key, val in request.data.items(): if key not in ignore_fields and key in valid_fields: updated = True match key: case _: # default the key will same as db column name update_fields[key] = val if updated and update_fields: for obj in edit_obj: for key, val in update_fields.items(): old_val = str(getattr(obj, key)) if old_val != val: old_new_data.setdefault("old_data", []).setdefault(obj.id, {})[key] = str(getattr(obj, key)) old_new_data.setdefault("new_data", []).setdefault(obj.id, {})[key] = val update_fields['updated_by'] = request.user.id update_fields['updated_at'] = timezone.now() edit_obj.update(**update_fields) if "unavailable_dates" in request.data: unavailable_dates = request.data['unavailable_dates'] for obj in edit_obj: current_unavailable_dates = [str(rp.unavailable_date) for rp in obj.esexamperiodunavailability_set.all()] if set(unavailable_dates) != set(current_unavailable_dates): updated=True obj.esexamperiodunavailability_set.all().delete() EsExamPeriodUnavailability.bulk_insert(obj.id, unavailable_dates) old_new_data.setdefault("old_data", []).setdefault(obj.id, {})["unavailable_date"] = current_unavailable_dates old_new_data.setdefault("new_data", []).setdefault(obj.id, {})["unavailable_date"] = unavailable_dates if old_new_data['old_data'] or old_new_data['new_data']: # 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) ) code = ",".join(edit_obj.values_list("code", flat=True)) remark_param = { "code":code } # 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'], old_data=old_new_data['old_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)