
    |<h                     F    d dl mZ d dlmZ  G d dej
                        Zy)    )rest_framework)UserProfilec                   2     e Zd Z G d d      Z fdZ xZS )UserProfileFilterc                       e Zd ZeZg dZy)UserProfileFilter.Meta)useruser__user_typeuser__is_activeuser__is_staffuser__is_superuseruser__emailN)__name__
__module____qualname__r   modelfields     A/var/www/html/dp2/backend_v2.1/server/apps/userprofile/filters.pyMetar      s    
r   r   c                     | j                   j                  }|j                  rt        |   |      S |j
                  r)t        |   |      j                  |j
                        S |j                         S )z=Filter queryset to only show users in the same active company)user__companies)requestr	   is_superusersuperfilter_querysetactive_companyfilternone)selfquerysetr	   	__class__s      r   r   z!UserProfileFilter.filter_queryset   sq    ||   7*844 7*84;; $ 3 3 <  
 }}r   )r   r   r   r   r   __classcell__)r#   s   @r   r   r      s    
 
 r   r   N)django_filtersr   filtersapps.userprofile.modelsr   	FilterSetr   r   r   r   <module>r)      s    4 /)) r   