3
ÒÞ`:
  ã               @   sL   d dl mZ d dlmZ d dlmZ dd„ Zdd„ Zd dlmZ dd	„ Zd
S )é    )ÚHttpResponseRedirect)Úredirect)Úreversec                s   ‡ fdd„}ˆ j |_ |S )Nc                s¼   t dƒ d| jkr¤| jd }t d|ƒ |dks6|dkrBttdƒƒS |dkrVttdƒƒS |d	krjttd
ƒƒS |dkr~ttdƒƒS ttdƒƒS |dkr¬t dƒ ttƒS nt dƒ ˆ | f|ž|ŽS )Nz%checkifcustomeralreadylogin decoratorÚuserrolezuserrole isÚcustomerÚcompanyzcustomer-dashboardÚdriverzdriver-dashboardÚadminzadmin-dashboardÚ
superadminzsuperadmin-dashboardzcustomer-loginznot customerzin else)ÚprintÚsessionr   r   r   Ú	loginlink)ÚrequestÚargsÚkwargsr   )Úfunction© ú=/var/www/html/Stanley/GoodsDeliveryOLD/Customer/decorators.pyÚwrap   s$    



z)checkifcustomeralreadylogin.<locals>.wrap)Ú__doc__)r   r   r   )r   r   Úcheckifcustomeralreadylogin   s    r   c                s   ‡ fdd„}ˆ j |_ |S )Nc                s‚   t dƒ d}| jr"|d| j 7 }nt dƒ d| jkrj| jd }t d|ƒ |dkrr|dkrrt d	ƒ t|ƒS nt|ƒS ˆ | f|ž|ŽS )
Nzchecklogincustomer decoratorz/login/z?next=zno pathr   zuserrole isr   r   znot customer or company)r   Úpathr   r   )r   r   r   r   r   )r   r   r   r   &   s    



z checklogincustomer.<locals>.wrap)r   )r   r   r   )r   r   Úchecklogincustomer%   s    r   c                s   ‡ fdd„}ˆ j |_ |S )Nc                sr   t dƒ d}| jr"|d| j 7 }nt dƒ d| jkrZ| jd }t d|ƒ |dkrbt|ƒS nt|ƒS ˆ | f|ž|ŽS )Nzchecklogincompany decoratorz/login/z?next=zno pathr   zuserrole isr   )r   r   r   r   )r   r   r   r   r   )r   r   r   r   >   s    



zchecklogincompany.<locals>.wrap)r   )r   r   r   )r   r   Úchecklogincompany=   s    r   N)	Údjango.httpr   Údjango.shortcutsr   Údjango.urlsr   r   r   r   r   r   r   r   Ú<module>   s    