
    h5                         d dl mZ d dlmZ ddlmZ d dl mZ d dlmZ d dl	m Z
 ddlmZ dd	lmZ d d
lmZ  G d dej                         Zy)    )viewsets)IsAuthenticated   )UserProfile)filters)LimitOffsetPagination)rest_framework)UserProfileFilter)UserProfileSerializer)CompanyPermissionc                       e Zd ZeZej                  ej                  gZ	e
Zg dZej                  j                         ZeZdZdZegZd Zd Zy)UserProfileViewSet)
first_name	last_nameuser__usernamer   usernamec                 P    |j                  | j                  j                         y)z>Automatically set the user to the currently authenticated user)userN)saverequestr   )self
serializers     ?/var/www/html/dp2/backend_v2.1/server/apps/userprofile/views.pyperform_createz!UserProfileViewSet.perform_create   s    T\\../    c                    | j                   dk(  s| j                   dk(  r| j                  j                  j                  r| j                  j                  d      S | j                  j                  j                  r;| j                  j                  d| j                  j                  j                        S | j                  j                         S | j                  j                  d| j                  j                        S )NlistretrieveT)user__is_active)r   user__companies)r   r   )actionr   r   is_superuserquerysetfilteractive_companynone)r   s    r   get_querysetzUserProfileViewSet.get_queryset   s    ;;& DKK:$=||  --}}++D+AA""11}}++$($(LL$5$5$D$D ,  
 }}))++ }}##Dt||?P?P#QQr   N)__name__
__module____qualname__r   pagination_classbackend_filtersDjangoFilterBackendr   SearchFilterfilter_backendsr
   filterset_classsearch_fieldsr   objectsallr#   r   serializer_classlookup_fieldlookup_url_kwargr   permission_classesr   r'    r   r   r   r      sf    ,++O (OAM""&&(H,#L!)*0Rr   r   N)r	   r   rest_framework.permissionsr   modelsr   r   utils.paginations.paginationr   django_filtersr,   r
   serializersr   apps.permissions.permissionsr   ModelViewSetr   r8   r   r   <module>r@      s4    # 6  " > < & . : R..  Rr   