import datetime from datetime import time from collections import defaultdict from django.db.models import Prefetch from rest_framework import serializers, status from api.models import TtSetting, TtWeekPattern, TtAvailability, TtStartPreference, \ TtUsagePreference, TtActivityTemplate, TtActivity, TtAcademicTerm, TtWeek 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_update, log_critical_error, get_exception_detail, \ get_add_new_data, get_ip, get_old_new_data, bulk_sync_to_redis, get_multiple_old_new_data, search_old_new_data, \ time_to_slot, convert_to_redis_week from rest_framework.exceptions import ValidationError from api.translation import __ from backend.kafka import send_request from backend.redis_client import redis_client class BookingUpdate(AdminApiBase): def validate_request(self,request): # wont update module id, just want based on module id get the activity template rules = { "id": "required|array|exists:api.TtActivity,id", } # if need rename field, can put here attribute = { # "email": __("attr.email"), } # validate id first, cause need get activity by id, if wrong id direct return error validator = BaseValidator(request.data,rules,attribute) error = validator.validate() if error: raise ValidationError(error) edit_activity = ( TtActivity.objects .select_related("department","activity_type", "zone") .prefetch_related( # "week", # tags means the relation add in models, __tag means the relation table tag_id get the tag object #2 ) .filter(id__in=request.data.get("id")) ) setting_params = { "slot_per_week", } settings = TtSetting.get_multiple_setting(setting_params) slot_per_week = settings["slot_per_week"] # only can update name/code/desc rules = { "name": "nullable", "desc": "nullable", "slot_required": "nullable|integer", "planned_size": "nullable|integer", "activity_type_id": "nullable|exists:api.TtActivityType,id", "suggested_day": "nullable", "suggested_time": "nullable", "department_id": "nullable|exists:api.TtDepartment,id", "zone_id": "nullable|exists:api.TtZone,id", # "staff_suitability": "nullable|array|exists:api.TtSuitability,id", "location_suitability": "nullable|array|exists:api.TtSuitability,id", "from_date": "nullable|date", # will used this date to get latest week id, if week id not found, create the week data and update to resource map "to_date": "nullable|date",# will used this date to get latest week id, if week id not found, create the week data and update to resource map } if request.data.get("code") and edit_activity.count() == 1: activity = edit_activity.first() if request.data.get("code") != activity.code: rules['code'] = "nullable|not_exists:api.TtActivity,code" # FE only pass in 1 of it also need return error, and both cant be null if "from_date" in request.data or "to_date" in request.data: rules['from_date'] = "required|date" rules['to_date'] = "required|date" if "activity_type_id" in request.data: rules["activity_type_id"] = "required|exists:api.TtActivityType,id" # 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) for activity in edit_activity: if not activity.is_booking: raise ValidationError({ "error": __("validation.invalid",field=__("attr.activity")), "errors": {"id": __("validation.invalid",field=__("attr.activity"))} }) return edit_activity def post(self, request): # Validate input try: edit_activity = self.validate_request(request) setting_params = { "minute_per_slot", } settings = TtSetting.get_multiple_setting(setting_params) minute_per_slot = int(settings["minute_per_slot"]) first_activity = edit_activity.first() action="booking_update" # this is for audit log only update_fields = {} ignore_fields = ['id','timestamp','signature'] update_error = False # if more than 1 activity edit, ignore code if edit_activity.count() > 1 or not request.data.get("code"): ignore_fields.append('code') valid_fields = {f.column for f in TtActivity._meta.fields} direct_update = ["name", "code", "desc"] for key, val in request.data.items(): if key not in ignore_fields and key in valid_fields: match key: case "suggested_time": if first_activity.scheduled == 0: if val: suggested_time = datetime.datetime.strptime(request.data["suggested_time"], "%H:%M") suggested_time_slot = time_to_slot(suggested_time, minute_per_slot) update_fields['suggested_time_slot'] = suggested_time_slot update_fields[key] = val else: update_fields['suggested_time_slot'] = None update_fields[key] = None case _: # default the key will same as db column name if key in direct_update: update_fields[key] = val else: update_error = True if first_activity.scheduled == 0: update_fields[key] = val if first_activity.scheduled == 0 and "slot_required" in request.data: if request.data.get("slot_required") != 0: update_fields['duration'] = request.data.get("slot_required") * minute_per_slot else: update_fields['duration'] = 0 # be4 save need get the old new data already, because bulk update is no matter got change or no changes also will update old_new_data = get_multiple_old_new_data(edit_activity,request.data.items(),ignore_fields,valid_fields) if update_fields: # 2026-04-27, ignore updated param, it is useless, this code is replace by "replace in file", so didnt remove param updated update_fields['updated_by'] = request.user.id edit_activity.update(**update_fields) if first_activity.scheduled == 0: if "location_suitability" in request.data: new_ids = request.data.get("location_suitability", []) old_new_data = self.update_m2m_field_bulk(edit_activity,'location_suitability',new_ids,old_new_data) if "from_date" in request.data and "to_date" in request.data: from_date = request.data.get("from_date") to_date = request.data.get("to_date") start_date = datetime.datetime.strptime(from_date, "%Y-%m-%d").date() end_date = datetime.datetime.strptime(to_date, "%Y-%m-%d").date() # only need get first monday, later inside loop, will keep add 7 day to get next monday current_monday = start_date - datetime.timedelta(days=start_date.weekday()) mondays = [] # loop the date to get all monday date, will used to do filter in db while current_monday <= end_date: mondays.append(current_monday) current_monday += datetime.timedelta(days=7) weeks = TtWeek.objects.filter(start_date__in=mondays) if not weeks: raise ValidationError({ "error": __("validation.invalid", field=__("attr.date")), "errors": {"date": __("validation.invalid", field=__("attr.date"))} }) week_ids = [week.id for week in weeks] week_name = [week.week for week in weeks] week_pattern_for_redis = convert_to_redis_week(week_name) old_new_data = self.update_m2m_field_bulk(edit_activity,'week',week_ids,old_new_data) redis_data = { "update": {} } redis_activity_table = "activity" redis_activity_data = {} redis_activity_column = [ "name", "desc", ] other_redis_activity_column = [ "slot_required", "duration", "planned_size", "activity_type_id", "suggested_day", "suggested_time","suggested_time_slot", "department_id", "zone_id" ] # if only 1 activity, can edit code if edit_activity.count() == 1: redis_activity_column.append('code') edit_id = request.data.get("id")[0] success_first_activity = TtActivity.objects.filter(id=first_activity.id).first() # got new data only means got update, then will for loop request.data, cause the old_new_data['new_data'] for foreign key name, like department_id, will return in department, so cant use it if old_new_data['new_data']: # every new activity reset the redis_activity_data and append to redis_data['update'] redis_activity_data = {"id": edit_id} for key, val in request.data.items(): if key in redis_activity_column: redis_activity_data[key] = val if first_activity.scheduled == 0: for column in other_redis_activity_column: match column: case _: value = getattr(success_first_activity, column) if isinstance(value, time): value = value.strftime("%H:%M") redis_activity_data[column] = value if "from_date" in request.data and "to_date" in request.data: redis_activity_data["week_pattern"] = week_pattern_for_redis or [] if "location_suitability" in request.data: suitability = request.data.get("location_suitability") redis_activity_data["location_suitability_id"] = suitability[0] if isinstance(suitability,list) and suitability else None if len(redis_activity_data)>1:# > 1 means not only id redis_data["update"].setdefault(redis_activity_table, []).append(redis_activity_data) bulk_sync_to_redis(redis_client, redis_data) if old_new_data['old_data'] or old_new_data['new_data']: #kafka update/ split by id kafka_activity = [] method = "booking_update" kafka_topic = self.kafka_config['MICROSERVICES_TT_TOPIC'] if kafka_topic: new_data = old_new_data.get("new_data") relation_fields = [ "location_suitability", "week", ] relation_data = generate_kafka_data_update(new_data, relation_fields) update_fields.pop("updated_by", None) for act in edit_activity: data = {"id": act.id, **update_fields, **relation_data} kafka_activity.append(data) kafka_request_data = { "session_id": request.user.name, "activity": kafka_activity, } send_request(kafka_topic, kafka_request_data, None, method) # 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) ) name = ",".join(edit_activity.values_list("name", flat=True)) remark_param = { "name":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, new_data=old_new_data['new_data'], old_data=old_new_data['old_data'], ) response = { } if first_activity.scheduled and update_error: first_message = {"error": __("validation.is_scheduled",activity= first_activity.name)} errors = {"errors": {"id": __("validation.is_scheduled",activity= first_activity.name)}} return self.api_response(error=first_message,errors=errors) 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)