from rest_framework import status
from rest_framework.exceptions import ValidationError

from api.models import User
from api.translation import __
from api.utils import log_critical_error, get_exception_detail
from api.views.admin.base import AdminApiBase

class AdminList(AdminApiBase):
    def post(self, request):
        try:
            self.api_log_skip_outgoing_data = True
            data = self.get_data(request)
            response = {
                "data": 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_permission_data(self, admin):
        data = []

        for val in admin.userpermission_set.all():
            department_ids = list(
                val.userpermissiondepartment_set.values_list("department_id", flat=True)
            )

            data.append({
                "permission": val.permission,
                "grant_to_all_department": val.grant_to_all_department,
                "department_ids": department_ids,
            })

        return data

    def get_data(self, request):
        admins = (
            User.objects
            .prefetch_related(
                "role",
                "userpermission_set__userpermissiondepartment_set"
                # tags means the relation add in models, __tag means the relation table tag_id get the tag object #2
            )
            .filter(user_type=User.USER_TYPE['admin'])
            .order_by("name")
        )
        return_data=[]

        for val in admins:
            data = {}
            data["id"] = val.id
            data["email"] = val.email
            data["name"] = val.name
            data["status"] = val.status
            data["status_text"] = __("attr.user_status." + str(val.status))
            data["is_only_self"] = val.is_only_self
            data["created_at"] = val.created_at

            data["role"] = list(
                val.role.values("id", "name", "color", "type", "is_only_self")
            )

            data["permissions"] = self.get_permission_data(val)

            return_data.append(data)

        return return_data