U
    YHgQ                     @   sd   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Z
G dd	 d	e	ejZd
S )    )absolute_import)	BaseQuery)db)replace_chars)TimestampMixinc                   @   s6   e Zd Zdd Zdd Zdd Zddd	Zd
d ZdS )TimecategoryQueryc                 C   s   |  tj|k S N)filterTimecategoryidfirst)selfZtime_category_id r   5/var/www/dostadmin/dostadmin/db_model/timecategory.py	get_by_id	   s    zTimecategoryQuery.get_by_idc                 C   s   |  tj S r   )order_byr
   timeallr   r   r   r   get_timecategory_log   s    z&TimecategoryQuery.get_timecategory_logc                 C   s   |  tj|k S r   )r	   r
   r   r   )r   Zcurrent_hourr   r   r   !get_timecategory_for_current_hour   s    z3TimecategoryQuery.get_timecategory_for_current_hour6pc                 C   s*   |  tjt| k }|r&|jS dS )N)r	   r
   namer   lowerr   r   )r   ip_nameZ	batch_recr   r   r   find_timecategory_id_with_name   s    z0TimecategoryQuery.find_timecategory_id_with_namec                 C   s   |  tj|tj S r   )r	   r
   r   in_r    active_user_count_till_yesterdayr   )r   Z	timeslotsr   r   r   get_least_users_timeslot   s    z*TimecategoryQuery.get_least_users_timeslotN)r   )__name__
__module____qualname__r   r   r   r   r   r   r   r   r   r      s
   
r   c                   @   s   e Zd ZdZeZG dd dZG dd dZej	ej
ddZej	eddd	d
Zej	edd	dZe	ej
ZejddddZdd ZdS )r
   timecategoryc                   @   s   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdedfedfedfedfedfedfe	dfe
dfedfedfedfedfedffZdS )z%Timecategory.AnnouncementTimeCategoryz08:00:00z09:00:00z10:00:00z11:00:00z12:00:00z13:00:00z14:00:00z15:00:00z16:00:00z17:00:00z18:00:00z19:00:00z20:00:00) zUser's timeslotz8 AMz9 AMz10 AMz11 AMz12 noonz1 PMz2 PMz3 PMz4 PMz5 PMz6 PMz7 PMz8 PMN)r    r!   r"   EIGHT_AMNINE_AMTEN_AM	ELEVEN_AM	TWELVE_PMONE_PMTWO_PMTHREE_PMFOUR_PMFIVE_PMSIX_PMSEVEN_PMEIGHT_PMCHOICESr   r   r   r   AnnouncementTimeCategory&   s8   r3   c                   @   sD   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdS )zTimecategory.CategoryZ8aZ9aZ10aZ11aZ12pZ1pZ2pZ3pZ4pZ5pr   Z7pZ8pZ9pN)r    r!   r"   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   ZNINE_PMr   r   r   r   CategoryF   s   r4   T)primary_key   F)uniquenullable)r8   
Experiencez-Timecategory.id == Experience.timecategory_id)backrefprimaryjoinc                 C   s   d| j  d t| j S )NzTimeCategory  )r   strr   r   r   r   r   __repr__a   s    zTimecategory.__repr__N)r    r!   r"   __tablename__r   query_classr3   r4   r   ColumnIntegerr   Stringr   r   r   relationshipexperiencesr>   r   r   r   r   r
   "   s    r
   N)
__future__r   flask_sqlalchemyr   	dostadminr   utils.helpers.helpersr   dostadmin.mixinsr   r   Modelr
   r   r   r   r   <module>   s   