import math from itertools import chain from django.conf import settings from django.db.models import Count, Q from rest_framework import status from rest_framework.exceptions import ValidationError from api.models import TtModule, TtActivityTemplate, TtActivity from api.translation import __ from api.utils import log_critical_error, get_exception_detail from api.validator import BaseValidator from api.views.admin.base import AdminApiBase class ActivityList(AdminApiBase): def validate_request(self,request): rules = { "academic_term_id": "required|exists:api.TtAcademicTerm,id", "filter": "nullable", "page": "nullable", "per_page": "nullable", "sort_by": "nullable", "order_by": "nullable", } # if need rename field, can put here attribute = { # "email": __("attr.email"), } # validate id first, cause need get pos by id, if wrong id direct return error validator = BaseValidator(request.data,rules,attribute) error = validator.validate() if error: raise ValidationError(error) def post(self, request): try: self.api_log_skip_outgoing_data=True self.validate_request(request) data = self.get_data(request) response = data 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) def get_data(self,request): filters = request.data.get("filter") academic_term_id = request.data.get("academic_term_id") activities = ( TtActivity.objects .filter(academic_term_id=request.data.get('academic_term_id')) .exclude(is_jta=1, jta_parent_id__isnull=False) .select_related("activity_template","department", "academic_term", "module", "week_pattern","activity_type", "zone", "availability", "start_preference", "usage_preference") .prefetch_related( "week", "week_pattern__week", "tag__tag", "location", "location_preset", "location_suitability", "staff", "staff_preset", "staff_suitability", "student_set", "jta_children__module", # tags means the relation add in models, __tag means the relation table tag_id get the tag object #2 ) ) if filters: exclude_fields = ["module_code", "module_name", "activity_template_name"] # extra filter using "OR" to direct get activity id, and must display the id if any(key in filters for key in exclude_fields): activities = self.custom_filter(academic_term_id,activities,filters) activities = self.listing_filter(request,activities,exclude_fields) paginated_details = self.listing_pagination(request,activities) total = paginated_details["total"] paginated_data = paginated_details["paginated_data"] page = paginated_details["page"] return_data=[] for val in paginated_data: real_size = 0 ignored = [] if val.student_set.all(): for sset in val.student_set.all(): if(sset.id not in ignored): real_size += (sset.planned_size or 0) ignored.append(sset.id) data={} data["id"] = val.id data["code"] = val.code data["name"] = val.name data['desc'] = val.desc data['activity_template_name'] = val.activity_template.name if val.activity_template else None data['activity_template_id'] = val.activity_template_id data['department'] = val.department.name if val.department else None data['department_id'] = val.department_id data['sequence_number'] = val.sequence_number data['academic_term'] = val.academic_term.name if val.academic_term else None data['academic_term_id'] = val.academic_term_id data['activity_type'] = val.activity_type.name if val.activity_type else None data['activity_type_color'] = val.activity_type.color if val.activity_type else None data['activity_type_id'] = val.activity_type_id data['zone'] = val.zone.name if val.zone else None data['zone_id'] = val.zone_id data['module_id'] = val.module_id data['duration'] = val.duration data['slot_required'] = val.slot_required data['is_jta'] = val.is_jta data['jta_parent_id'] = val.jta_parent_id data['is_variant'] = val.is_variant data['variant_parent_id'] = val.variant_parent_id data['scheduled'] = True if val.scheduled else False data['staff_requirement_type'] = __("attr.staff_requirement_type_name."+str(val.staff_requirement_type)) data['staff_requirement'] = val.staff_requirement data['location_requirement_type'] = __("attr.location_requirement_type_name."+str(val.location_requirement_type)) data['location_requirement'] = val.location_requirement data['scheduled_start_time'] = val.scheduled_start_time data['scheduled_start_slot'] = val.scheduled_start_slot data['scheduled_day'] = __("attr.days_name."+str(val.scheduled_day)) if val.scheduled_day is not None else None data['suggested_day'] = __("attr.days_name."+str(val.suggested_day)) if val.suggested_day is not None else None data['suggested_time'] = val.suggested_time if val.suggested_time is not None else None data['suggested_time_slot'] = val.suggested_time_slot if val.suggested_time_slot is not None else None data['planned_size'] = val.planned_size data['real_size'] = real_size data['module_size'] = val.module.planned_size if val.module else 0 data["week_pattern_id"] = val.week_pattern.id if val.week_pattern else None data["week_pattern_name"] = val.week_pattern.name if val.week_pattern else "-" if val.week_pattern: data["week_pattern"] = [ {"id": week.id, "week": week.week, "start_date": week.start_date} for week in val.week_pattern.week.all() ] else: data["week_pattern"] = [ {"id": week.id, "week": week.week, "start_date": week.start_date} for week in val.week.all() ] data["availability_id"] = val.availability.id if val.availability else None data["availability_name"] = val.availability.name if val.availability else "-" data["availability_pattern"] = val.availability.pattern if val.availability else val.availability_pattern data["start_preference_id"] = val.start_preference.id if val.start_preference else None data["start_preference_name"] = val.start_preference.name if val.start_preference else "-" data["start_preference_pattern"] = val.start_preference.pattern if val.start_preference else val.start_preference_pattern data["usage_preference_id"] = val.usage_preference.id if val.usage_preference else None data["usage_preference_name"] = val.usage_preference.name if val.usage_preference else "-" data["usage_preference_pattern"] = val.usage_preference.pattern if val.usage_preference else val.usage_preference_pattern if val.jta_children.all(): data["module_name"] = ",".join(child.module.name for child in val.jta_children.all() if child.module) data["module_code"] = ",".join(child.module.code for child in val.jta_children.all() if child.module) else: data["module_name"] = val.module.name if val.module else None data["module_code"] = val.module.code if val.module else None data["staff"] = [ {"id": staff.id, "name": staff.name,"code": staff.code} for staff in val.staff.all() ] data["staff_preset"] = [ {"id": staff_preset.id, "name": staff_preset.name,"code": staff_preset.code} for staff_preset in val.staff_preset.all() ] data["staff_suitability"] = [ {"id": staff_suitability.id, "name": staff_suitability.name,"code": staff_suitability.code} for staff_suitability in val.staff_suitability.all() ] data["location"] = [ {"id": location.id, "name": location.name, "code": location.code} for location in val.location.all() ] data["location_preset"] = [ {"id": location_preset.id, "name": location_preset.name, "code": location_preset.code} for location_preset in val.location_preset.all() ] data["location_suitability"] = [ {"id": location_suitability.id, "name": location_suitability.name, "code": location_suitability.code} for location_suitability in val.location_suitability.all() ] data["student_set"] = [ {"id": student_set.id, "name": student_set.name, "code": student_set.code} for student_set in val.student_set.all() ] data["tag"] = [ {"id": tag.tag.id, "name": tag.tag.name} for tag in val.tag.all() ] for i in range(1,11): extra_data_field = f"extra_data_{i}" data[extra_data_field] = getattr(val,extra_data_field,None) return_data.append(data) return { "total": total, "page": page, "data": return_data, } def custom_filter(self,academic_term_id,activities,filters): activity_ids = set() extra_id = set() if filters.get("module_code"): activity_by_module_code = TtActivity.objects.filter( academic_term_id=academic_term_id, # is_jta=1, # jta_parent_id__isnull=False, module__code__icontains=filters.get("module_code") ).values_list("id", "jta_parent_id") # activity_by_module_code will be somethings [(1,10),(2,None),(3,15)], will filter out None activity_ids.update(filter(None, chain.from_iterable(activity_by_module_code))) if filters.get("module_name"): activity_by_module_name = TtActivity.objects.filter( academic_term_id=academic_term_id, # is_jta=1, # jta_parent_id__isnull=False, module__name__icontains=filters.get("module_name") ).values_list("id", "jta_parent_id") # activity_by_module_name will be somethings [(1,10),(2,None),(3,15)], will filter out None activity_ids.update(filter(None, chain.from_iterable(activity_by_module_name))) if filters.get("activity_template_name"): activity_by_template_name = TtActivity.objects.filter( academic_term_id=academic_term_id, # is_jta=1, # jta_parent_id__isnull=False, activity_template__name__icontains=filters.get("activity_template_name") ).values_list("id", "jta_parent_id") # activity_by_template_name will be somethings [(1,10),(2,None),(3,15)], will filter out None activity_ids.update(filter(None, chain.from_iterable(activity_by_template_name))) if activity_ids: variant_child_ids = ( TtActivity.objects.filter( academic_term_id=academic_term_id, is_variant=1, variant_parent_id__in=activity_ids, ).values_list("id", flat=True) ) extra_id.update(activity_ids) extra_id.update(variant_child_ids) if extra_id: activities = activities.filter(id__in=extra_id) # lastest_activity = ( # TtActivity.objects # .filter(Q(id__in=extra_id) | Q(pk__in=activities.values("pk"))) # .select_related("activity_template","department", "academic_term", "module", "week_pattern","activity_type", "zone", "availability", "start_preference", "usage_preference") # .prefetch_related( # "week", # "week_pattern__week", # "tag__tag", # "location", # "location_preset", # "location_suitability", # "staff", # "staff_preset", # "staff_suitability", # "student_set", # ) # ) # return lastest_activity return activities