siTO 6 d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z m
Z
dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlm Z ddl m Z ddl m Z m!Z! ddl"m#Z# d Z$ G d de Z% G d de Z& G d de Z' G d de Z(y) zD
models.py
This module is used to register models for project app
N)date)apps)ValidationError)models)static)reverse reverse_lazy)timezone) format_html) gettext_lazy)HorillaCompanyManager)Company)Employee)horilla_middlewares)_thread_locals) HorillaModel upload_pathrender_templatec t | dkD rt t d | j d \ }}t | dk st | dk rt t d t | }t | dkD s
|t d vrt t d y# t $ r}t t d |d}~ww xY w) zM
this method is used to validate the format of duration like fields.
z)Invalid format, it should be HH:MM format: < z Invalid timezInvalid formatN)lenr _splitintrange
ValueError)valuehourminuteerrors 2/var/www/python-projects/worksol/project/models.pyvalidate_time_formatr& s 5zA~ a K LMM
> {{3' f t9q=CK!O!!$O"PQ QV t9q=F%) 3!!N"34 4 4 > a 0 12=>s A>B# # C,CCc e Zd Zd ed fd ed fd ed fd ed fd ed
fd ed fgZ ej d
d ed Z ej e dd ed Z
ej e dd ed Z ej edd ed Z ej ed Z ej dd ed Z ej edd ed Z ej& ed Z ej* edd ej. ! Z ed" Zd5d#Zd$ Z d% Z d& Z d' Z d( Z d) Z!d* Z"d+ Z#d, Z$d- Z%d. Z&d/ Z'd6d0Z( fd1Z)d2 Z* G d3 d4 Z+ xZ,S )7ProjectnewNew in_progress In Progress completed Completedon_holdzOn Hold cancelled CancelledexpiredExpired TName)
max_lengthunique verbose_nameproject_managerszProject Managersblank related_namer8 project_memberszProject Members Statuschoicesr6 defaultr8
Start Dater8 End Datenullr; r8 z Project File upload_tor; rG r8 DescriptionF)rG editable on_delete
company_idc n t | j |k r | j S | j d| dz S )z
Returns a truncated version of the description attribute.
Parameters:
length (int): The maximum length of the returned description.
Nz...)r description)selflengths r% get_descriptionzProject.get_descriptionV sC 4##$.
!!'6*U2
c | j j }|r*dj |D cg c]
}t | c} }|S yc c}w )z!
managers column
N)managersalljoinstrrP employeesemployeeemployee_names_strings r% get_managerszProject.get_managersc sJ MM%%' $*KK/898X 9% ! ) ( 9 Ac | j j }|r*dj |D cg c]
}t | c} }|S yc c}w )z
members column
rU N)membersrW rX rY rZ s r% get_memberszProject.get_membersn sJ LL$$& $*KK/898X 9% ! ) ( 9r_ c & d| j d}|S zV
Method will retun the api to the avatar or path to the profile image
z!https://ui-avatars.com/api/?name=z&background=randomtitlerP urls r%
get_avatarzProject.get_avatary 2$** =OP
rS c x | j r.| j j }t d }t d|| S y )Nz images/ui/project/document.pngz View)documentrh r r )rP document_url image_urls r% get_document_htmlzProject.get_document_html sB == ==,,L ?@I
rS c t t dd }|j j t dd| j
i }| j j v s| j j v sst fd| j j D sGt fd| j j D s|j j d rd | d
S y ) zH
This method generates an onclick URL for task viewing.
requestNz task-view
project_idkwargsc 3 V K | ] }|j j v " y wN)
task_managersrW .0taskr\ s r% z#Project.redirect.. s% XD8t 1 1 5 5 77X &)c 3 V K | ] }|j j v " y wrv ) task_membersrW rx s r% r{ z#Project.redirect.. s% W48t 0 0 4 4 66Wr| zproject.view_projectz onclick="window.location.href='z ?view=list'" ) getattrr user employee_getr pkrV rW ra anytask_sethas_perm)rP rq rh r\ s @r% redirectzProject.redirect s .)T:<<,,; dgg/FG
))+ +4<<++--XDMMDUDUDWXXW4==CTCTCVWW||$$%;<5cU-H HrS c 8 t dd| j i }|S )z0
This method to get detail url
zproject-detailed-viewr rs r r rg s r% get_detail_urlzProject.get_detail_url s 2D$''?K
rS c 8 t dd| j i }|S )/
This method to get update url
zupdate-projectr rs r rg s r% get_update_urlzProject.get_update_url s +T477OD
rS c 8 t dd| j i }|S )z0
This method to get archive url
zproject-archiverr rs r rg s r% get_archive_urlzProject.get_archive_url s ,lDGG5LM
rS c h | j j }| j }t d|| S )Na )r countrf r )rP
task_countrf s r% get_task_badge_htmlzProject.get_task_badge_html s7 ]]((*
rS c j t dd| j i }t d }d| ddz d| dz S )/
This method to get delete url
zdelete-projectrr rs z-Are you sure you want to delete this project?',r r r rP rh messages r% get_delete_urlzProject.get_delete_url sE +\4774KLCD3%qzC AgYa.00rS c t dd| i S )?
This method for get custom column for action.
zcbv/projects/actions.htmlinstancepathcontextr rP s r% actionszProject.actions s
, &
rS c | j ryyz
archive status
Archivez
Un-Archive is_activer s r% archive_statuszProject.archive_status >> rS c | j P| j | j k r
t ddi | j t j k rd| _ y y y )Nrl z End date is less than start dater2 )end_date
start_dater r todaystatusr s r% cleanz
Project.clean sP == $}}t.%z3U&VWW}}tzz|+ ' , %rS c > | j d u t t j dd }}|r<|j j d x}r |dk7 rt
j | | _ t | ( |i | |r$t j j d| dd y y )Nrq selected_companyrW Todo F)rf projectsequence is_end_stage)r r r r sessiongetr findrM supersave ProjectStageobjectscreate)rP argsrt is_newrq cid __class__s r% r z Project.save s ''T/7 . . 4,
goo 1 12D EEsE3RW< %ll3 /DO
d%f% ' ' dQU
(
rS c | j S rv re r s r% __str__zProject.__str__ s zzrS c 0 e Zd ZdZ ed Z ed Zy) Project.Meta7
Meta class to add the additional info
r( ProjectsN)__name__
__module__ __qualname____doc__r r8 verbose_name_plural rS r% Metar s |
mrS r )2 returnN)-r r r r PROJECT_STATUSr CharFieldrf ManyToManyFieldr rV ra r DateFieldr r FileFieldr rl TextFieldrO
ForeignKeyr PROTECTrM r
r rR r^ rb ri ro r r r r r r r r r r r r
__classcell__r s @r% r( r( 2 s % -() a n% AiL! a n% AiL!
N
F Dqy QE%v%%')* H %f$$&() G V
3AhKF "!!q ?J v TAjMRH v T1^CTH #&""-0@AK"""dUfnnJ $L1G
) ) $
1
(
, ,rS r( c
F e Zd ZdZ ej
d ed Z ej e ej ddd ed Z ej ddd
Z
ej d ed Z ed
ZdefdZddZ fdZ fdZ G d d Z xZS )r z
ProjectStage model
r4 Titler6 r8 Tproject_stagesr( )rL rG r; r< r8 F)rG r; rK z Is end stagerB r8 project__company_idr c | j S rv re r s r% r zProjectStage.__str__ ** rS c > | j r| j }|j j d j | j
}|rL|j j d j
}t t d|j d y y )NT)r )idz Already exist an end stage - .)
r r r filterexcluder firstr r rf )rP r existing_end_stage end_stages r% r zProjectStage.clean s llG!(!7!7!>!> ! "? "gg!
" # 2 2 9 9t 9 L R R T %5ioo5FaHI "
rS c | j ]t j j | j j d j
}|r|j dz | _ t | |i | y )Nr r r ) r r r r r order_bylastr r )rP r rt
last_stager s r% r zProjectStage.save! sf == $$++DLL+A*%
* 3 3a 7
d%f%rS c t j j | j | j }|D ]' }|xj dz c_ |j ) t
| |i | y )N)r sequence__gtr )r r r r r r r delete)rP r rt project_stages_afterstager s r% r zProjectStage.delete, sg + 3 3 : :LLt}} ;
* E NNa N JJL ''rS c 8 e Zd ZdZddgZ ed Z ed Zy)ProjectStage.Metar r rf
Project StagezProject StagesNr r r r unique_togetherr r8 r r rS r% r r 8 s) %g. ) 0 1rS r r )r r r r r r r rf r r( CASCADEr IntegerFieldr BooleanFieldr r
r rY r r r r r r r s @r% r r s
F !G* EE f ..
%y\
G #v""D5IH&6&&u1^CTUL#$9:G &
(2 2rS r c L e Zd ZdZd ed fd ed fd ed fd ed fgZ ej d
ed Z ej e
ej d
ed Z ej e
ej d
d ed Z ej ed
ed Z ej ed
d ed Z ej edd ed Z ej( d
d
ed Z ej( d
d
ed Z ej. ed
d
ed Z ej4 ed Z ej8 d!" Z e d# Z d6d%Z G d& d' Z!d( Z"d) Z#d* Z$d+ Z%d, Z&d- Z'd. Z(d/ Z)d0 Z*d1 Z+d2 Z,d3 Z-d4 Z.d5 Z/y$)7Taskz
Task model
to_dozTo Dor+ r, r- r. r2 r3 r4 r r Tr( )rL rG r8 tasksr rL rG r< r8 z
Task Managers)r; r8 z Task Membersr: r> r? r@ rC rF rE z Task FilerH rJ rD r )rB r Nc R | j r| j j \| j j | j k s#| j j | j kD rt dt d i | j t j k rd| _ y y )Nr zDThe task end date must be between the project's start and end dates.r2 )r r r r r r r r r s r% r z
Task.cleanl s == $ )>)>)J %%
5<<**T]]:%"A b% ==4::< '#DK (rS c 8 e Zd ZdZddgZ ed Z ed Zy) Task.Metar r rf r TasksNr r rS r% r r | s& %g.y jrS r c | j S rv re r s r% r z Task.__str__ r rS c 6 | j r | j S dS )z>
Return project if have,otherwise return none
Noner r s r%
if_projectzTask.if_project s
$||t||77rS c 8 t dd| j i }|S )z%
detail view of task
ztask-detail-viewr rs r r rg s r% task_detail_viewzTask.task_detail_view s
($ A
rS c ^ t | j j | j S )z
to get status
)dict TASK_STATUSr r r s r%
status_columnzTask.status_column s$ D$$%))$++66rS c | j j }|r*dj |D cg c]
}t | c} }|S yc c}w )z&
return task managers
rU r )rw rW rX rY )rP rV managermanagers_name_strings r% r^ zTask.get_managers sH %%))+ #);;H/UG /U#V ' ' 0Vr_ c | j j }|r*dj |D cg c]
}t | c} }|S yc c}w )z%
return task members
rU r )r~ rW rX rY )rP ra membermembers_name_strings r% rb zTask.get_members sH ##'') "(++.Qvs6{.Q"R & & /Rr_ c t dd| i S )r zcbv/tasks/task_actions.htmlr r r r s r% r z Task.actions s & . &
rS c & d| j d}|S rd re rg s r% ri zTask.get_avatar rj rS c t dd| i S )z?
This method for get custom document coloumn .
z cbv/tasks/task_document.htmlr r r r s r% document_colzTask.document_col s
/ &
rS c t dd| i S )z:
This method for get detail view actions.
z"cbv/tasks/task_detail_actions.htmlr r r r s r% detail_view_actionszTask.detail_view_actions
5 &
rS c 8 t dd| j i }|S )z'
to get the update url
zupdate-task-allr rs r rg s r% r zTask.get_update_url s 'tww@
rS c | j ryyr r r s r% r zTask.archive_status r rS c 8 t dd| j i }|S )z$
to get archive url
ztask-all-archivetask_idrs r rg s r% r zTask.get_archive_url s
()TWW1EF
rS c t t dd| j i }| d}t d }d| ddz d| dz S )z#
to get delete url
z delete-taskr$ rs z?task_all=truez*Are you sure you want to delete this task?r r )r r r )rP rh url_with_paramsr s r% r zTask.get_delete_url sR
mY,@A E 0@A?#1%+'!n<>1Y<G
F ..
'
E +F**'M
*6))7>ARL V
W1X;F "!!t4a oVJ v TAjMRH v T1[>H #&""-0@AK"v""1-H#$9:G$ ) ) 87
0
=rS r c ~ e Zd ZdZd ed fd ed fgZ ej eej dd ed Z
ej e ej dd
ed Z ej e
ej ed
Z ej ej" ed Z ej& dddeg ed Z ej& edd ed Z ej. dd ed Z ed Z G d d Zd Z d Z d Z d Z d Z d! Z!d" Z"d# Z#d$ Z$ G d% d Zy&)' TimeSheetz
TimeSheet model
in_Progressr, r- r. Tproject_timesheetr( r task_timesheetr r )rL r8 Dater z00:00
z Hours Spent)rG rB r6
validatorsr8 r> r? r@ rJ )r; rG r8 project_id__company_idc e Zd ZdZy)TimeSheet.Meta)z-idN)r r r orderingr rS r% r r1 6 s rS r c | j
t ddi | j | j dk( r
t ddi | j rx| j }| j r| j }||j
j
vr||j j
vr||j j j
vr||j j j
vrt t d | j r`|| j j j
vr:|| j j j
vrt t d | j t j j j kD rt dt d i y y )
Nrr zProject name is Required.r rO z/Please provide a description to your Time sheetz"Employee not included in this taskz%Employee not included in this projectr z You cannot choose a future date.)rr r rO employee_idr$ rw rW r~ r rV ra r r datetimer )rP r\ rz s r% r zTimeSheet.clean9 sy ?? "!<1L"MN N #t'7'72'=! QR
''H|||| D$6$6$:$:$<< $(9(9(=(=(? ? $ (=(=(A(A(C C $ (<(<(@(@(B B)!,P*QRR DOO$<$<$@$@$BB $(?(?(C(C(E E)!,S*TUUyy8,,22499;;%vq1S/T&UVV <# rS c | j d| j d| j d| j d| j S )N )r4 rr r$ r
time_spentr s r% r zTimeSheet.__str__T sB ""#1T__$5Qt||nAdii[PQRVRaRaQbccrS c ^ t | j j | j S rv )r TIME_SHEET_STATUSr r r s r% r zTimeSheet.status_columnW s" D**+// <rV s 2 - - ! ) 6 > $ ' 6 4 5
>&J,l J,Z@2< @2FD=< D=NG/ G/rS