MBjdZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZGdd eZGd d eZGd d eZGddeZGddeZGddej2Zy)zQ budget/forms.py =============== Forms for Budget Management (Module 1 of EFMS). )forms) gettext_lazy) ModelForm)reload_queryset) PERIOD_ANNUALPERIOD_MONTHLYPERIOD_QUARTERLY PERIOD_WEEKLYBudgetAllocation BudgetHead BudgetItemBudgetSettings BudgetSubHeadc eZdZGddZy)BudgetSettingsFormcHeZdZeZdgZdejddddiZy)BudgetSettingsForm.Metaoverrun_limit_percentoh-input w-1000.010classstepminattrsN) __name__ __module__ __qualname__rmodelfieldsr NumberInputwidgets0/var/www/python-projects/worksol/budget/forms.pyMetars3)* #%6U%6%6 0&M& r'r)N)rr r!r)r&r'r(rrs   r'rc2eZdZGddZfdZxZS)BudgetHeadFormcpeZdZeZgdZejddiejddddZ y) BudgetHeadForm.Meta)name description is_activerrrrrows)r.r/N) rr r!r r"r#r TextInputTextarear%r&r'r(r)r-&s<5#EOO74D*EF)5>>:JTU0VW r'r)cNt||i|t|jy)N)super__init__rr#selfargskwargs __class__s r(r8zBudgetHeadForm.__init__.s! $)&) $r'rr r!r)r8 __classcell__r=s@r(r+r+%s  %%r'r+c2eZdZGddZfdZxZS)BudgetSubHeadFormceZdZeZgdZejddiejddiejddddZ y ) BudgetSubHeadForm.Meta)headr.r/r0roh-select oh-select-2rrr1r2)rEr.r/N) rr r!rr"r#rSelectr4r5r%r&r'r(r)rD4sP= ELL1H'IJ#EOO74D*EF)5>>:JTU0VW r'r)ct||i|t|jtj j d|jd_y)NTr0rE)r7r8rr#r objectsfilterquerysetr9s r(r8zBudgetSubHeadForm.__init__=sF $)&) $'1'9'9'@'@4'@'P F$r'r>r@s@r(rBrB3s  QQr'rBc2eZdZGddZfdZxZS)BudgetItemFormceZdZeZgdZejddiejddiejddddZ y ) BudgetItemForm.Meta)sub_headr.r/r0rrFrrr1r2)rQr.r/N) rr r!rr"r#rrGr4r5r%r&r'r(r)rPDsPA$ G5L+MN#EOO74D*EF)5>>:JTU0VW r'r)ct||i|t|jtj j d|jd_y)NTrIrQ)r7r8rr#rrJrKrLr9s r(r8zBudgetItemForm.__init__MsG $)&) $+8+@+@+G+GRV+G+W J(r'r>r@s@r(rNrNCs  XXr'rNc<eZdZGddZfdZfdZxZS)BudgetAllocationFormc eZdZeZgdZejddiejddddejdd d ejdd d dejdd dddZ y)BudgetAllocationForm.Meta)item fiscal_year period_type period_valueallocated_amountrrFrr20002100)rrmaxztogglePeriodValue(this))ronchange153rrrN) rr r!r r"r#rrGr$r%r&r'r(r)rVTs  !ELL1H'IJ,5,, 0O(5<< 7E^_.E-- 0TJ!2 1 1 0&M! r'r)c2t||i|t|jtj j djd|jd_d|jd_ td|jd_ y)NTrIsub_head__headrWFrZz,Period Value (Q1-4 / Month 1-12 / Week 1-53)) r7r8rr#rrJrKselect_relatedrLrequired_labelr9s r(r8zBudgetAllocationForm.__init__ms~ $)&) $'1'9'9'@'@4'@'P'_'_ (  F$05 N#,,-.\,] N#)r'ct|}|jd}|jd}|tk(rd|d<|S|tk(r,|r d|cxkrdksn|j dt d|S|tk(r,|r d|cxkrdksn|j dt d|S|tk(r*|r d|cxkrdksn|j dt d |S) NrYrZrz Quarter must be between 1 and 4. zMonth must be between 1 and 12.5zWeek must be between 1 and 53.) r7cleangetrr add_errorrfr r )r: cleaned_datarYrZr=s r(rlzBudgetAllocationForm.cleanvsw} "&&}5 #''7 - '+/L (, ,\(>Q(>~q1S/TUN *\(?R(?~q1R/ST  M )\(?R(?~q1Q/RSr')rr r!r)r8rlr?r@s@r(rTrTSs  2^r'rTcreZdZdZej edejdddZdZ y ) BudgetBulkUploadFormz Upload an Excel/CSV file to bulk create or update budget items. Expected columns: Head Name, Sub-Head Name, Item Name, Description Optional: Fiscal Year, Period Type, Period Value, Allocated Amount zUpload File (.xlsx or .csv)rz .xlsx,.csv)racceptr)rgwidgetc2|jd}|jj}|jds/|jdst j t d|jdkDrt j t d|S)Nfilez.xlsxz.csvz'Only .xlsx and .csv files are accepted.iPzFile must be smaller than 5 MB.)ror.lowerendswithrValidationErrorrfsize)r:fr.s r( clean_filezBudgetBulkUploadForm.clean_filesw   f %vv||~ g&$--*?''*S(TU U 66O #''*K(LM Mr'N) rr r!__doc__r FileFieldrf FileInputrur{r&r'r(rqrqs; 5??-.u/?<%XY D r'rqN)r|djangordjango.utils.translationrrf base.formsr base.methodsrmodelsrr r r r r rrrrr+rBrNrTFormrqr&r'r(rsx 6 (      %Y % Q Q XY X 494v5::r'