U ŷig@@sddlmZmZddlmZmZmZmZddlm Z ddl m Z m Z ddl mZddlmZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZdd lmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(ddl)m*Z*eddgdZ+e e,ee,eedddZ-ee.dddZ/ee.dddZ0e ee.ee,ddddZ1e+j2ded ed!d"d#d$ed!d%d&ededeefe.e.ee,ee,e d'd(d)Z3e+4d*eefe"e d+d,d-Z5e+4d.eefe#e d+d/d0Z6e+4d1eefe$e d+d2d3Z7e+j2d4ee%d edeefee.e d5d6d7Z8e+j2d8e%d eefe,e d9d:d;Z9e+4d<eefe,e d9d=d>Z:e+j;d8d?d@eefe,e d9dAdBZ4s z)_active_loan_deduction..)sumr(r4r1r1r2_active_loan_deduction3sr=cCstdd|jDS)Ncss|]}|jdkr|jVqdSr5)r6monthly_benefit)r8br1r1r2r:8s z"_active_benefit..)r;r)r<r1r1r2_active_benefit7sr@N)r!r4total_to_applytransaction_date descriptionr$c sdd|jD}|rdkr dStdd|Dt|D]\}}dkr|t|dkrnt|j}qtfdd|dd D}n t|}t||j} | dkrq:|j| } t|j |j ||d| | d } | |_| dkrd |_ | | q:dS) zNDistribute `total_to_apply` across the employee's active loans proportionally.cSsg|]}|jdkr|qS)r-)r6r8lnr1r1r2 Cs z)_apply_loan_deduction..rNcss|] }|jVqdSN)r7rDr1r1r2r:Gsz(_apply_loan_deduction..c3s |]}t|jVqdSrG)roundr7rDZstandard_totalrAr1r2r:Ns)loan_id employee_idrBrCdebitcreditbalancepaid) r(r; enumeratelenrIr7minremaining_balancerr+r6add) r!r4rArBrCZ active_loansiloanshareZ credit_amt new_balancetxr1rJr2_apply_loan_deduction;s:      r\z/preview)response_model.rH )gelei)r_)monthyearr"r#r!cCsz|tttjttjttjtjdk}|rF|tj |k}|rpdd| dD}|tj |}| tj}g}tdd} |D]} t| } t| } | j| | } | j pd}| jr| jjnd}|| |d <| |d d 7<| |d | 7<|t| j | j| jr$| jjnd| j| j| j| | | | j |d qdd| D}tdd|D}t|||dS)NTcSsg|]}|r|qSr1)strip)r8rWr1r1r2rFysz#preview_payroll..,cSs ddddS)NZUnknownr)r/counttotalr1r1r1r1r2z!preview_payroll..nonez No Companyr/rerHrf) rMr/ bank_nameaccount_number account_name base_salarybenefitloan_deduction net_salaryr" company_namecSs,g|]$\}}t||d|d|ddqS)r/rerf)r"rqemployee_count total_net)r)r8cidvr1r1r2rFscss|] }|jVqdSrG)rs)r8sr1r1r2r:sz"preview_payroll..)rowssummary grand_total)r%rr&r r'r(r)r*r-r"splitr+r,r.r/r0r r=r@rmappendrrjrkrlitemsr;r)rarbr"r#r!r%Zids employeesrwZcompany_totalsr4loan_dedrnnetrtcnameZ summariesryr1r1r2preview_payrollhs\    rz /export-csv)bodyr!cCs~t|t|jd}t||j|j}|jp,t}dd|D}t|||}d|j |j dd|j d}t |dd d |id S) NrcSs&g|]}|j|j|jt|dqSrkr/amount)rkrlrmr=r8er1r1r2rFs  z&export_payroll_csv..payroll_02d_.csvtext/csvContent-Dispositionattachment; filename=content media_typeheaders) rrr"r3r# transfer_dater todayrrbrar/rrr!companyr}transferrw csv_contentfilenamer1r1r2export_payroll_csvs  rz /export-thrcCsvt|t|jd}t||j|j}|jp,t}dd|D}t|||}d|j d|j d}t |ddd |id S) NrcSsg|]}|j|j|jdqSr)rkrlrmrr1r1r2rFs z"export_thr_csv..thr_rrrrrr) rrr"r3r#rr rrrbr/rrr1r1r2export_thr_csvs  rz/exportcCst|t|jd}t||j|j}dd|jD}dd|jD}t}g}g}|D]} |j dkr| | j d} d} d} | j | } | dk}nDd} t | }t| } | j |kr|| j } d}n|} d}| j | | } |j dkrtd| n| }|dkr|| j| j|d || j | j| j| j| j | | | ||d qTtd d |D}t|j|j|j |j|j|t|dd }||||D]$}tfd|j i|}||q|t|||}|j dkrd|jd|jd}nd|j|jdd|jd}t|ddd|idS)a Unified payroll/THR export endpoint. - export_type "regular": base_salary minus loan deductions (with per-employee overrides) - export_type "thr": base_salary only, no deductions - Transfer date = today (date of actual export) - Loan transactions are NOT recorded here; use POST /runs/{id}/run to record them. Always creates a PayrollRun record for audit history. rcSsi|]}|j|jqSr1)rMro)r8or1r1r2 sz"export_unified..cSsi|]}|j|jqSr1)rM cash_advance)r8car1r1r2rsthrrTFr) rM employee_namerlrkrmrnror net_amount has_overridecss|]}|dVqdS)rNr1)r8rr1r1r2r:"sz!export_unified..)rarb export_typer"rq total_amountrrrecorded_transactionsrun_idrrrrrrrrr)rrr"r3r# overridesthr_cash_advancesr rrgetr+rmr=r@maxr{rkrlr/r;rrarbrSrVflushrcommitrr)rr!rr}Z override_mapZcash_advance_maprcsv_rowsZrun_items_datar4Zemp_cash_advancer~rnrZ is_overrideZ standard_dedrrrunZ item_dataitemrrr1r1r2export_unifieds            rz/runs)rbr!cCs>|tttj}|r*|tj|k}|tj  S)zJReturn all payroll export runs, optionally filtered by year, newest first.) r%rr&r r|r*rbr.run_datedescr0)rbr!r%r1r1r2list_payroll_runsEsrz/runs/{run_id})rr!cCs:|tttjtj|k}|s6tddd|S)z/Return a single payroll run with all its items.Payroll run not found status_codedetail) r%rr&r r|r*r+firstrrr!rr1r1r2get_payroll_runQs  rz/runs/{run_id}/runc Cs|tttjtj|k}|s6tddd|j rHtddd|j }|j dkr|jD]z}|j rb|jdkrxqb|tttjtj|j k}|sqb|jrdnd }t|||j|d |jd |jd |qbd |jd |jd }n d|j}d|_ |d|iS)a" Record loan deduction transactions for a previously exported payroll run. Uses the stored loan_deduction per item and distributes it across the employee's current active loans proportionally. Can only be called once (no-op is prevented via recorded_transactions flag). rrrz4Transactions have already been recorded for this runZregularrz (adjusted)zPayroll deduction /rzLoan transactions recorded for zTHR marked as executed for Tmessage)r%rr&r r|r*r+rrrrr rrMrorr(rr\rbrar)rr!rZtx_daterr4suffixrr1r1r2run_payroll_transactions_sH         r)rcCsR|ttj|k}|s*tddd|jr.rrrrrrrrrr)r%rr&r r|r*r+rrrr"rr rrrbr/rar)rr!rrrrrrr1r1r2download_payroll_runs0       r)>typingrrfastapirrrrfastapi.responsesrsqlalchemy.ormr r datetimer collectionsr app.core.databaser app.models.employeerapp.models.companyrapp.models.loan_transactionrapp.models.payroll_runrrapp.models.benefitrapp.schemas.payrollrrrrrrrrapp.helpers.crudrZapp.helpers.csv_exportrrouterstrr3intr=r@r\rrpostrrrrrrrrrr1r1r1r2s~        (    -   Bg  6