B
    b                 @   s   d dl Z d dlZd dlmZ d dlZd dlmZmZ d dl	mZ
 d dlmZ d dlmZ G dd dejjZG d	d
 d
ejZG dd dejejeZdS )    N)gettext_lazy)mixinsserializers)r   )UtilGenericViewSet)AttReportSettingc               @   s   e Zd ZG dd dZdS )AttReportSettingFilterc               @   s   e Zd ZeZdgZdS )zAttReportSettingFilter.MetaidN)__name__
__module____qualname__r   modelfields r   r   AG:\easytimepro\master/mysite/att\api\views\view_report_setting.pyMeta   s   r   N)r	   r
   r   r   r   r   r   r   r      s   r   c               @   s*   e Zd Ze Zdd ZG dd dZdS )AttReportSettingSerializerc             C   sn   |j s
dS t|j }g }xH|D ]@}t|}|d|d |d< tt|d |d< || q W t	|S )N Zoragin_namename)
Zatt_itemjsonloadscopyZdeepcopygetstr_appenddumps)selfobjitemsZ_itemsitemZ_itemr   r   r   	get_items   s    

z$AttReportSettingSerializer.get_itemsc               @   s   e Zd ZeZdZdS )zAttReportSettingSerializer.Meta__all__N)r	   r
   r   r   r   r   r   r   r   r   r   (   s   r   N)r	   r
   r   r   ZSerializerMethodFieldr   r    r   r   r   r   r   r      s   r   c                   sD   e Zd ZdZeZej Ze	Z
eedZdd Z fddZ  ZS )AttReportSettingViewSetz"
    ReportParamViewSet Class
    )listZretrievec             C   s   | j | jtjS )N)serializer_dictr   actionatt_serializersZNoneSerializer)r   r   r   r   get_serializer_class?   s    z,AttReportSettingViewSet.get_serializer_classc                s    t t|  j| jjjd}|S )N)Z
company_id)superr"   get_querysetfilterZrequestuserZcurrent_company)r   Z	query_set)	__class__r   r   r)   C   s    z$AttReportSettingViewSet.get_queryset)r	   r
   r   __doc__r   r   ZobjectsallZquerysetr   Zfilter_classr   r$   r'   r)   __classcell__r   r   )r,   r   r"   -   s   
r"   )r   r   Zdjango.utils.translationr   r   Zdjango_filtersZrest_frameworkr   r   Zmysite.att.apir&   Zmysite.att.api.utils_classr   Zmysite.att.modelsr   Z	FilterSetr   ZModelSerializerr   ZListModelMixinZRetrieveModelMixinr"   r   r   r   r   <module>   s   