import datetime from collections import defaultdict from django.db.models import Prefetch from rest_framework import serializers, status from api.models import TtSetting, TtActivityTemplate, TtActivity, TtAcademicTerm, TtWeek, TtActivityType 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, time_to_slot, convert_to_redis_week, get_next_available_code, \ get_slot_default_pattern from rest_framework.exceptions import ValidationError from api.translation import __ from backend.kafka import send_request from backend.redis_client import redis_client class BookingCreate(AdminApiBase): def validate_request(self,request): setting_params = { "slot_per_week", } settings = TtSetting.get_multiple_setting(setting_params) slot_per_week = settings["slot_per_week"] rules = { "code": "nullable|not_exists:api.TtActivity,code", "name": "required", "desc": "nullable", "slot_required": "required|integer", "planned_size": "required", "activity_type_id": "required|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": "required", # 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": "required",# will used this date to get latest week id, if week id not found, create the week data and update to resource map } # 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) # add validation for activity type must checked booking = 1 activity_type_id = request.data.get("activity_type_id") activity_type = TtActivityType.objects.filter(id=activity_type_id,booking=True) if not activity_type: raise ValidationError({ "error": __("validation.invalid", field=__("attr.activity_type")), "errors": {"activity_type_id": __("validation.invalid", field=__("attr.activity_type"))} }) def post(self, request): # Validate input try: setting_params = { "minute_per_slot", "slot_per_week" } settings = TtSetting.get_multiple_setting(setting_params) minute_per_slot = int(settings["minute_per_slot"]) slot_per_week = int(settings["slot_per_week"]) self.validate_request(request) old_new_data = { "old_data": None, "new_data": None, } slot_default_pattern = get_slot_default_pattern(slot_per_week) action="booking_create" # this is for audit log only # based on date to get week id input_date = request.data.get("date") 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"))} }) idata = {} # not insert to activity table ignore_field = ['date','timestamp','signature'] valid_fields = {f.column for f in TtActivity._meta.fields} for key, val in request.data.items(): if key not in ignore_field: match key: case _: if key in valid_fields:#for insert, key can check in default part, cause if got special handle, i can check the way to use key also # default the key will same as db column name idata[key] = val idata['availability_pattern'] = slot_default_pattern['availability'] idata['usage_preference_pattern'] = slot_default_pattern['usage_preference'] idata['start_preference_pattern'] = slot_default_pattern['start_preference'] if request.data.get("slot_required") != 0: idata['duration'] = request.data.get("slot_required") * minute_per_slot else: idata['duration'] = 0 # both type force to wildcard, since if no set suitability, is allow to choose all resources idata["staff_requirement"] = 1 # hardcode 1 first, cause seem like UI also cant set it # idata["staff_requirement_type"] = TtActivity.STAFF_REQUIREMENT_TYPE['wildcard'] if request.data.get("staff_suitability") else TtActivity.STAFF_REQUIREMENT_TYPE['none'] idata["staff_requirement_type"] = TtActivity.STAFF_REQUIREMENT_TYPE['wildcard'] idata["location_requirement"] = 1 # hardcode 1 first, cause seem like UI also cant set it idata["location_requirement_type"] = TtActivity.LOCATION_REQUIREMENT_TYPE['wildcard'] # if FE no pass in code or is null or empty, will used default code code = request.data.get("code") if not code: idata["code"] = get_next_available_code(TtActivity, "code", "MV", 10) idata['status'] = TtActivity.STATUS_TO_CODE['active'] idata['created_by'] = request.user.id idata['is_booking'] = 1 # FE possible giving null, need used request.data.get to check, so null will be skip if request.data.get("suggested_time"): suggested_time = datetime.datetime.strptime(request.data["suggested_time"], "%H:%M") idata['suggested_time_slot'] = time_to_slot(suggested_time,minute_per_slot) activity = TtActivity.objects.create(**idata) old_new_data['new_data']=get_add_new_data(activity) # even only have 1 week, but make it in array so can reuse some code 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(activity,'week',None,week_ids,old_new_data) # staff_suitability_ids = request.data.get("staff_suitability") # if staff_suitability_ids: # old_new_data = self.update_m2m_field(activity, 'staff_suitability', None, staff_suitability_ids,old_new_data) location_suitability_ids = request.data.get("location_suitability") if location_suitability_ids: old_new_data = self.update_m2m_field(activity, 'location_suitability',None,location_suitability_ids, old_new_data) redis_activity_table = "activity" # from here make it same name as db first, got different, only do extra handle redis_activity_column = [ "id", "code", "name", "desc", "planned_size", "real_size", "slot_required","activity_type_id", "duration", "suggested_day", "suggested_time_slot","suggested_time","scheduled_start_slot", "department_id", "module_id", "academic_term_id", "activity_template_id", "week_pattern_id", "week_pattern", "availability_id", "availability_pattern","start_preference_id", "start_preference_pattern", "usage_preference_id", "usage_preference_pattern", "staff_requirement_type","staff_required_no", "staff_preset", "staff_suitability_id", "location_requirement_type","location_required_no", "location_preset","location_suitability_id", "is_jta", "jta_parent_id", "is_variant", "variant_parent_id", "status", "scheduled" ] redis_data = { "insert": {} } redis_activity_data = {} for column in redis_activity_column: match column: case "staff_suitability_id": # 2025-12-17 request for redis side store as int(only can have 1 suitability) first # redis_activity_data[column] = list(current_activity_template.staff_suitability.values_list("id",flat=True)) # redis_activity_data[column] = activity.staff_suitability.values_list("id",flat=True).first() or None # suitability_list = list(activity.staff_suitability.all()) # redis_activity_data[column] = suitability_list[0].id if suitability_list else None redis_activity_data[column] = None case "location_suitability_id": # 2025-12-17 request for redis side store as int(only can have 1 suitability) first # redis_activity_data[column] = list(current_activity_template.location_suitability.values_list("id",flat=True)) # redis_activity_data[column] = activity.location_suitability.values_list("id",flat=True).first() or None suitability_list = list(activity.location_suitability.all()) redis_activity_data[column] = suitability_list[0].id if suitability_list else None case "staff_preset": redis_activity_data[column] = [] case "location_preset": redis_activity_data[column] = [] case "week_pattern": redis_activity_data[column] = week_pattern_for_redis case "student_sets": # when create dun have student set, so can force to empty list redis_activity_data[column] = [] case "staff_required_no": redis_activity_data[column] = activity.staff_requirement case "location_required_no": redis_activity_data[column] = activity.location_requirement case "usage_preference_id" | "availability_id" | "start_preference_id" | "week_pattern_id": # jose request if no id, dun store null, want to store "0" value = getattr(activity, column) redis_activity_data[column] = 0 if value in (None,"") else value case _: redis_activity_data[column] = getattr(activity, column) redis_data["insert"].setdefault(redis_activity_table, []).append(redis_activity_data) bulk_sync_to_redis(redis_client, redis_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) ) remark_param = { "name":activity.name, } # kafka push method = "booking_create" kafka_topic = self.kafka_config['MICROSERVICES_TT_TOPIC'] new_data = old_new_data.get("new_data") kafka_activity = [] if new_data and kafka_topic: relation_fields = [ # "staff_suitability", "location_suitability", "week", "tag", ] kafka_activity = generate_kafka_data(new_data, relation_fields) kafka_request_data = { "session_id": request.user.name, "activity": kafka_activity, } send_request(kafka_topic, kafka_request_data, None ,method) # 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'], ) response = { "activity_id": activity.id } 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)