ejv( 0dZddlmZddlmZddlmZmZmZddl m Z m Z m Z eGddZ eGd d ZGd d Zed k(rdddddddgZddddddddddddgZddiZdZej+eeeeZedej1D]#\ZZedk\rdndZededededz d%ej9eZed eD]5Zedej>d!ej@d"ejBdz d7y#y#)$z Settlement engine for TripTalley. Computes who owes whom after tallying all expenses in a trip. Uses debt simplification to minimize the number of transfers.  dataclass)Decimal)DictListTuple)Moneyconvert_currency from_minorceZdZUdZeed<eed<eed<edefdZ edefdZ edefdZ de fd Z defd Zy ) Balancez,A user's net balance in a specific currency.user_id amount_minorcurrencyreturnc |jdkDSNrrselfs 3/srv/projects/triptalley/backend/app/core/settle.py is_positivezBalance.is_positive  1$$c |jdkSrrrs r is_negativezBalance.is_negativerrc,t|jSN)absrrs r abs_amountzBalance.abs_amounts4$$%%rcBt|j|jSrr rrrs r to_decimalzBalance.to_decimal $++T]];;rcp|jdk\rdnd}|t|j|jS)Nr+)rr r)rsigns r__repr__zBalance.__repr__#s7''1,s" 4#4#4dmmDEFFrN)__name__ __module__ __qualname____doc__int__annotations__strpropertyboolrrr!rr$r*rrrr s~6 LM %T%%%T%%&C&&?A ArN) r+r,r-r.r/r0r1rr$r*r4rrr6r6(s5-OM2L '(; < z*H 13FGFX% %&7%8z0SettlementEngine.simplify_debt..s QqTrT)keyreversec |dSrZr4r[s rr]z0SettlementEngine.simplify_debt..s 1Q4r)rr SETTLEMENT)r7r8rrr )itemsrJsortlenminr6) rM creditorsdebtorsrbalancepaymentsc_idxd_idx creditor_id creditor_amt debtor_id debtor_amt settle_amounts r simplify_debtzSettlementEngine.simplify_debts$I  ( 0 4 GW{  '7!341'23  4 >48  6 uc)n$W)=(1%(8 %K$+EN !Iz j9Mq !**!.) !!,\M-IJIe 'm)CDGEN"a' u~a A% /c)n$W)=2rri from_currency to_currencyfx_rateDisplayPaymentc ddlm}|Gdd}g}|D]i}t|j|j||}|j ||j |j|j|j||k|S)a Convert canonical settlement payments to a viewer's home currency. Args: payments: Payments in the settlement currency from_currency: The settlement currency to_currency: The viewer's home currency fx_rate: FX rate from settlement to home currency Returns: List of DisplayPayment with both canonical and converted amounts rrcVeZdZUeed<eed<eed<eed<eed<eed<defdZy ) .DisplayPaymentr7r8rramount_display_minordisplay_currencyrc d|jd|jdt|j|jdt|j |j d S)Nr;r<r=z ())r7r8r rrryrzrs rr*zESettlementEngine.convert_for_display..DisplayPayment.__repr__sb 1 12*T__rs "$$66 GG G4  A A  A@ @ H z !!&$   HQt<Qt<Qq9L %cH  00,*=H +$NN,=ls y4&'#+c):;<=  --h7H - ] 'z!,,s1>>RUCUVYBZ[\]Sr