B
    GDb                 @   s   d dl 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	 d	ejZG d
d deZG dd deeje	jZdS )    N)	FilterSet)mixins)serializers)TerminalWorkCode)viewsets)APPCustomizeResponseModelMixinc               @   s   e Zd ZG dd dZdS )WorkCodeSerializerc               @   s   e Zd ZeZdZdS )zWorkCodeSerializer.Meta)idcodealiasN)__name__
__module____qualname__r   ZmodelZfields r   r   DG:\easytimepro\master/mysite/mobile\api_rest\views\view_work_code.pyMeta   s   r   N)r   r   r   r   r   r   r   r   r      s   r   c               @   s   e Zd ZdS )WorkCodeFilterN)r   r   r   r   r   r   r   r      s   r   c                   s:   e Zd Zej ZeZdeiZ	dd Z
 fddZ  ZS )WorkCodeViewSetlistc             C   s   | j | jd S )N)serializer_class_dictgetaction)selfr   r   r   get_serializer_class    s    z$WorkCodeViewSet.get_serializer_classc                s(   t t|  }| jjj}|j|d}|S )N)
company_id)superr   get_querysetZrequestuserr   filter)r   querysetr   )	__class__r   r   r   #   s    
zWorkCodeViewSet.get_queryset)r   r   r   r   Zobjectsallr   r   Zserializer_classr   r   r   __classcell__r   r   )r    r   r      s
   
r   )Zdjango_filtersZdjango_filters.rest_frameworkr   Zrest_frameworkr   r   Zmysite.iclock.modelsr   Zmysite.mobile.api_restr   Z(mysite.mobile.api_rest.app_customize_mixr   ZModelSerializerr   r   ZListModelMixinZGenericViewSetr   r   r   r   r   <module>   s   