import datetime from rest_framework import serializers, status from api.models import TtUsagePreference, TtSetting from api.models.audit_trail import AuditTrail from api.models.audit_trail_details import AuditTrailDetails from api.models.tt_academic_term import TtAcademicTerm from api.models.tt_location import TtLocation from api.models.tt_staff import TtStaff from api.models.tt_week import TtWeek from api.views.admin.base import AdminApiBase from api.validator import BaseValidator from api.utils import create_resource_map_all, decrypt_aes_128_cbc, encrypt_aes_128_cbc, generate_kafka_data, \ get_resource_map_redis_data, get_slot_default_pattern, log_critical_error, get_exception_detail, \ get_add_new_data, get_ip, bulk_sync_to_redis, convert_to_redis_week, get_next_available_code, \ import_get_resource_map_redis_data from rest_framework.exceptions import ValidationError from api.translation import __ from backend.kafka import send_request from backend.redis_client import redis_client class AcademicTermCreate(AdminApiBase): def validate_request(self,request): status_code_string = ",".join(str(v) for v in TtAcademicTerm.STATUS_TO_CODE.values()) rules = { "code": "nullable|not_exists:api.TtAcademicTerm,code", "name": "required", "start_date": "required|date", "end_date": "required|date", "start_time": "required|time", "end_time": "required|time", "status": "required|in:"+status_code_string+"?academic_term_status", } # 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 def post(self, request): # Validate input try: self.validate_request(request) old_new_data = { "old_data": None, "new_data": None, } action="academic_term_create" # this is for audit log only idata = {} ignore_field = [] valid_fields = {f.column for f in TtAcademicTerm._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 # get setting setting_params = { "initial_date", "initial_day", "minute_per_slot", "slot_per_day", "slot_per_week", } settings = TtSetting.get_multiple_setting(setting_params) default_availability = "" default_usage = "" default_start = "" start_time = datetime.datetime.strptime(request.data.get("start_time"), "%H:%M").time() end_time = datetime.datetime.strptime(request.data.get("end_time"), "%H:%M").time() time_count = [0,0] for i in range(int(settings.get("slot_per_week"))): time_format = datetime.time(time_count[0], time_count[1]) if start_time <= time_format < end_time: default_availability += "0" else: default_availability += "1" default_usage += "5" default_start += "5" time_count[1] += int(settings.get("minute_per_slot")) # add minutes if time_count[1] >= 60: # move to next hour time_count[1] -= 60 time_count[0] += 1 if time_count[0] >= 24: # reset after 23:59 time_count[0] = 0 # calculate default usage/start/availability idata['default_availability'] = default_availability idata['default_usage'] = default_usage idata['default_start'] = default_start idata['created_by'] = request.user.id # 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(TtAcademicTerm, "code", "MV", 10) academic_term = TtAcademicTerm.objects.create(**idata) # get latest week from week table, if empty, will based on setting start date as week 1 latest_week = TtWeek.objects.order_by("-week").first() week_idata = [] week_for_redis = [] if latest_week: current_week_id = latest_week.id current_week = int(latest_week.week) current_start_date = latest_week.start_date else: current_week_id = 1 current_week = 1 current_start_date = datetime.datetime.strptime(settings['initial_date'],"%Y-%m-%d").date() # insert the first 1 first, so can using the same code to handle got first week or no fisrt week week_idata.append( TtWeek( id=current_week_id, week=current_week, start_date=current_start_date, ) ) # must convert to datetime format only can compare academic_term_start_date = datetime.datetime.strptime(academic_term.start_date,"%Y-%m-%d").date() academic_term_end_date = datetime.datetime.strptime(academic_term.end_date,"%Y-%m-%d").date() # calculate how many week need to insert while current_start_date <= academic_term_end_date: current_week_id = current_week_id + 1 current_week = current_week + 1 current_start_date = current_start_date + datetime.timedelta(days=7) week_idata.append( TtWeek( id=current_week_id, week=current_week, start_date=current_start_date, ) ) if week_idata: TtWeek.objects.bulk_create(week_idata) # update staff and location resource map w_ids = [week.id for week in week_idata] resource_map = get_slot_default_pattern(settings.get("slot_per_week")) update_resource_map_tables = {"staff", "location"} for i in update_resource_map_tables: create_resource_map_all(i, w_ids, resource_map["resource_map"]) old_new_data['new_data']=get_add_new_data(academic_term) # week 53 is 2026-01-05, week 54 is 2026-01-12, if start time is 2026-01-08, week 1 for this academic term will be week 53(2026-01-05) start_weekday = academic_term_start_date.weekday() this_week_monday = academic_term_start_date - datetime.timedelta(days=start_weekday) weeks_data = TtWeek.objects.filter(start_date__gte=this_week_monday,start_date__lte=academic_term.end_date).order_by("week") week_ids = weeks_data.values_list("id",flat=True) week_name = list(weeks_data.values_list("week",flat=True)) week_for_redis = convert_to_redis_week(week_name) old_new_data = self.update_m2m_field(academic_term,'week',None,week_ids,old_new_data) # new_ids = request.data.get("tag", []) # old_new_data = self.update_m2m_field(academic_term,'tag',None,new_ids,old_new_data,False,True) redis_academic_term_table = "academic_term" # from here make it same name as db first, got different, only do extra handle redis_academic_term_column = [ "id","code","name","start_date","end_date","start_time","end_time","week","default_availability","default_usage","default_start" ] redis_data = { "insert": {}, "update": {}, } redis_academic_term_data = {} for column in redis_academic_term_column: match column: case "week": redis_academic_term_data[column] = week_for_redis case _: redis_academic_term_data[column] = getattr(academic_term, column) redis_data["insert"].setdefault(redis_academic_term_table, []).append(redis_academic_term_data) if week_idata: # update staff/location resource_map to redis staff_ids = list(TtStaff.objects.values_list("id", flat=True)) location_ids = list(TtLocation.objects.values_list("id", flat=True)) # this part no need used the value for allocated_activity_ids, so if staff_ids: redis_data["update"]["staff"] = import_get_resource_map_redis_data("staff", staff_ids) if location_ids: redis_data["update"]["location"] = import_get_resource_map_redis_data("location", location_ids) bulk_sync_to_redis(redis_client, redis_data) #kafka push method = "academic_term_create" kafka_topic = self.kafka_config['MICROSERVICES_TT_TOPIC'] new_data = old_new_data.get("new_data") kafka_request_data = { "session_id": request.user.name, } if new_data and kafka_topic: relation_fields = [ "week", ] kafka_request_data["academic_term"] = generate_kafka_data(new_data, relation_fields) # if new week created, also need push kafka if week_idata: week_data = [ { "id": week.id, "week": week.week, "start_date": week.start_date.strftime("%Y-%m-%d") } for week in week_idata ] kafka_request_data["week"] = week_data 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) ) remark_param = { "name":academic_term.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'], ) 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)