o
    Mg                     @   sP   d dl mZ d dlmZmZ d dlmZ d dlmZ dd Z	dd Z
d	d
 ZdS )    )messages)redirectrender)timezone)Categoryc              
   C   s   | j dkrn| jd }z>tjj|| jjdd	 r(t
j| ddd tdW S t|| jd	}d
| jv r8d|_nd|_|  t
j| ddd tdW S  tym } zt| t
j| ddd tdW  Y d }~S d }~ww t| dS )NPOSTcategory_nameF)namecreated_by__company
is_deletedzCategory name should be unique.danger
extra_tags	/category)r	   
created_by	is_activeTzCategory created successfully!successzError creating category: zcreate_category.html)methodr   getupperr   objectsfilterusercompanyexistsr   errorr   r   saver   	Exceptionprintr   )requestr   categorye r"   4/var/www/rescue_company/qr_manager/views/category.pycreate_category   s.   




r$   c              
   C   s   zUt jj|drI| jjt jj|djjkr<t jj|d}d|_d|_| j|_t	
 |_|  tj| ddd tdW S tj| dd	d tdW S tj| d
d	d tdW S  tyw } zt| tj| dd	d tdW  Y d }~S d }~ww )NpkTFzCategory deleted successfully!r   r   r   /You are not authorized to delete this category.r   Category not found.Error deleting category: )r   r   r   r   r   r   r   r   
deleted_byr   nowdeleted_dater   r   r   r   r   r   r   )r   category_to_deleter    r!   r"   r"   r#   delete_category#   s*   



r.   c              
   C   s  | j dkrztjj|dr}| jjtjj|djjkrptjj|d}t| j d| jv rI| jd}t|d z|dkr=dnd}||_	W n   Y nd|_	d| jv r_| jd}t|d ||_
|  tj| dd	d
 tdW S tj| ddd
 tdW S tj| ddd
 tdW S  ty } zt| tj| ddd
 tdW  Y d }~S d }~ww tjj|dstj| ddd
 tdS tjj|d}t| dd|idS )Nr   r%   r   onTFr   zCategory updated successfully!r   r   r   r'   r   r(   r)   zedit_category.htmlr    )context)r   r   r   r   r   r   r   r   r   r   r	   r   r   r   r   r   r   r   )r   category_idr    r   r   r!   category_detailsr"   r"   r#   edit_category?   sJ   









r3   N)django.contribr   django.shortcutsr   r   django.utilsr   qr_manager.modelsr   r$   r.   r3   r"   r"   r"   r#   <module>   s    