}0=iIdZddlmZddlmZddlmZddlmZmZddl m Z m Z m Z m Z ddlmZmZmZmZe r dd lmZdd lmZed Gd dZGddZy)z&Wallet service for point transactions.) annotations) dataclass)datetime)Decimal ROUND_HALF_UP) TYPE_CHECKINGAnyDictOptional)InsufficientBalanceErrorInvalidParamsErrorInvalidPointTypeErrorWalletOperationError)Model)WalletRepositoryProtocolT)frozenceZdZUdZded<ded<ded<ded<ded <d ed <d ed <ded <ded<ded<ded<y)TransactionRecordz'Represents a wallet transaction record. int | Noneidstrwtypeintiiduidtyperamountbalance trans_typedescrcdateDict[str, Any] extra_dataN)__name__ __module__ __qualname____doc____annotations__W/home/cursorai/projects/telegram-earn/packages/wallet_utils/src/wallet_utils/service.pyrrs?1N J H H I O  J Jr+rcHeZdZdZhdZd dZ d ddZ d ddZddZ ddZ d dd Z dd Z d dd Z e dd Zy) WalletServicez Python port of the legacy WalletPoint PHP class. The service works with a WalletRepositoryProtocol that only needs to define point types. All balance operations and transaction recording are handled automatically. > rrrrr"r!rrrr c||_y)z Initialize the wallet service. Args: repository: WalletRepositoryProtocol instance that defines point types and provides User model N) repository)selfr0s r,__init__zWalletService.__init__>s %r+Ncf|i}|jd|||||d||jj}|j|d} t dt | } |j td| zt} |jj||| d} td|||d | | |||j|jd i } |jj| }|S#t$r} td | | d} ~ wwxYw#t$r} td | | d} ~ wwxYw)a Add points to a user wallet balance. Args: user_id: The user ID point_type: The type of wallet to add to (e.g., "credit_balance") amount: The amount to add (must be positive) remarks: Description/remarks for the transaction trans_type: Transaction type code (integer constant, optional) params: Extra parameters dict with optional 'data' key for additional fields iid: Initiator ID - ID of the user who performed this transaction (-100 for system) Returns: The transaction ID Raises: InvalidPointTypeError: If point type doesn't exist InvalidParamsError: If params are invalid WalletOperationError: If the operation fails NaddFr0.1roundingallow_negativezError adding user point: cdata rrrrrrrr r!r"r$#Error creating transaction record: )_validate_add_deduct_pointr0get_point_typesgetmaxrquantizerrupdate_balance Exceptionrr_get_current_datetimecreate_transaction_record)r1user_id point_typerremarksr paramsr point_typesdecimal_placesamount_rounded balance_aftererecordtransaction_ids r, add_pointzWalletService.add_pointGsZ< >F '' 7JUF oo557 $Q7QN 34 >)ITab O OO::7JP^ot:uM #!!!,,.zz&"-   Y!__FFvNN- O&)B1#'FGQ N O& Y&)LQC'PQWX X Ys0C3D3 D<D  D D0D++D0c |i}|jd||||||||jj} | j|d} t dt | } |j td| zt} |jj||| |\} } | sP|jj||}t|}t|}td |d | d d |d | d || td|||d| | |||j!|jdi } |jj#|}|S#t$r}td||d}~wwxYw#t$r}td||d}~wwxYw)a- Deduct points from a user wallet balance using atomic SQL operation. Uses SQL WHERE clause to prevent race conditions and ensure balance sufficiency. Args: user_id: The user ID point_type: The type of wallet to deduct from amount: The amount to deduct (must be positive) remarks: Description/remarks for the transaction trans_type: Transaction type code (integer constant, optional) allow_negative: Whether to allow negative balance (default: False) params: Extra parameters dict with optional 'data' key for additional fields iid: Initiator ID - ID of the user who performed this transaction (-100 for system) Returns: The transaction ID Raises: InsufficientBalanceError: If insufficient balance and allow_negative=False InvalidPointTypeError: If point type doesn't exist InvalidParamsError: If params are invalid WalletOperationError: If the operation fails Ndeductr5rr6r7r9zError deducting user point: z6Insufficient balance for this transaction. Available: .fz , Requested: ) available requesteddr<r=r>)r?r0r@rArBrrCrrdeduct_balance_atomicrErget_user_balancefloatr rrFrG)r1rHrIrrJr r:rKrrLrMrNsuccessrOrPcurrent_balancerXrYrQrRs r, deduct_pointzWalletService.deduct_pointsF >F '' gz67JX^ oo557 $Q7QN 34 >)ITab R%)__%J%J^N&K& "G]"oo>>w SOo.If I*HSTUcTddeReHfg'.)9':;=##  #!!!,,.zz&"-   Y!__FFvNNG R&)EaS'IJPQ Q R@ Y&)LQC'PQWX X Ys0"E+E( E%E  E%( F1FFcb|jj}||vrtd|d|y)z Validate whether the given point type is valid. Raises: InvalidPointTypeError: If point type doesn't exist No such point ( ) exists!rIN)r0r@r)r1rHrI valid_typess r,_validate_user_point_typez'WalletService._validate_user_point_types>oo557 [ ('!*Y7J  )r+c |j|||dk(rdnd} t|tstd| d| d|vr|drt|dtstd| d| g} |dj D]"} | |j vs| j | $| r%d j| } td| d | d | y y y ) z Validate parameters for add/deduct point operations. Raises: InvalidPointTypeError: If point type doesn't exist InvalidParamsError: If params are invalid r4rSr`zInvalid params for z(), it must be a dict!)methodr<zInvalid params["data"] for z, z-(), some of the data parameters are invalid: !N)rf isinstancedictrkeysRESERVED_FIELDSappendjoin) r1 add_deductrHrIrrJr r:rKrhinvalid_fieldsfield fields_strs r,r?z(WalletService._validate_add_deduct_points$ &&w ; *e 3&$'$':6(BX%Ybhi i V vfVnd3(1&9OPY_ N,,.D000"))%0/!YY~6 (1&9fgqfrrst!!/ r+c |jj} | jj} || j |} |A|jj } || vrt d|d|| j |} || j |} |%|dvrtd |d | j | } || j | } |7t|trtj|}| j | } |7t|trtj|}| j |} | jd} | dkDr| | d} || d|} g} | D]}t|j|j|j |j"|j$|j&|j(|j*|j,t/|j0dr|j0j3nt|j0|j4xsi }| j7|| S)a Retrieve wallet transaction history with various filters. Args: user_id: Filter by user ID (uid field) point_type: Filter by wallet/point type (wtype field) trans_type: Filter by transaction type code (trans_type field, integer constant) transaction_type: Filter by transaction type ('c' for credit, 'd' for debit) (type field) iid: Filter by initiator ID (iid field) - user who performed the transaction (-100 for system) start_date: Filter transactions from this date (inclusive) (cdate field) end_date: Filter transactions until this date (inclusive) (cdate field) limit: Maximum number of records to return offset: Number of records to skip (for pagination) Returns: List of TransactionRecord objects Example: # Get all transactions for a user transactions = service.get_transaction_history(user_id=123) # Get credit transactions for a specific point type transactions = service.get_transaction_history( user_id=123, point_type="credit_balance", transaction_type="c", ) # Get transactions by transaction type code from wallet_utils.transaction_types import WALLET_DEPOSIT transactions = service.get_transaction_history( user_id=123, trans_type=WALLET_DEPOSIT, ) # Get transactions in date range from datetime import datetime, timedelta end = datetime.now() start = end - timedelta(days=30) transactions = service.get_transaction_history( user_id=123, start_date=start, end_date=end, limit=100, ) Nrrbrcrdrr r;rZInvalid transaction_type: &. Must be 'c' (credit) or 'd' (debit).rr cdate__gte cdate__ltez-cdater isoformatr=)r0get_wallet_modelobjectsallfilterr@rrrjrr fromisoformatorder_byrrrrrrrrr r!hasattrr"rr$rn)r1rHrIr transaction_typer start_dateend_datelimitoffset wallet_modelquerysetrerecordsobjrQs r,get_transaction_historyz%WalletService.get_transaction_history$s8t779 ''++-  73H  !//99;K,+%j\;  Z8H  !*=H  'z1(01A0BBhi ,<=H ?3/H  !*c*%33J? *=H  (C(#11(;(;H$$X. A:(H  'HC&66iiGGGGXXzz >>ii/6syy+/Ncii))+TWX[XaXaTb>>/R F NN6 " r+c|jj} |jj|}t |j |j |j|j|j|j|j|j|jt|jdr|jj!nt#|j|j$xsi S#|j&$rYywxYw)z Get a single transaction by its ID. Args: transaction_id: The transaction ID Returns: TransactionRecord if found, None otherwise )rrr=N)r0rrrArrrrrrrrr r!rr"rrr$ DoesNotExist)r1rRrrs r,get_transaction_by_idz#WalletService.get_transaction_by_ids779  &&**n*=C$66iiGGGGXXzz >>ii/6syy+/Ncii))+TWX[XaXaTb>>/R  ((  sCC99D  D c|jj}|jj} || j |} |A|jj } || vrt d|d|| j |} || j |} |%|dvrtd|d | j | } || j | } |7t|trtj|}| j | } |7t|trtj|}| j | } | jS)a Count transactions matching the given filters. Args: user_id: Filter by user ID (uid field) point_type: Filter by wallet/point type (wtype field) trans_type: Filter by transaction type code (trans_type field, integer constant) transaction_type: Filter by transaction type ('c' for credit, 'd' for debit) (type field) iid: Filter by initiator ID (iid field) - user who performed the transaction (-100 for system) start_date: Filter transactions from this date (inclusive) (cdate field) end_date: Filter transactions until this date (inclusive) (cdate field) Returns: Number of transactions matching the filters rurbrcrdrvrwrxryrzr{r|r}r) r0rrrrr@rrrjrrrcount) r1rHrIr rrrrrrres r,count_transactionsz WalletService.count_transactionssd2779 ''++-  73H  !//99;K,+%j\;  Z8H  !*=H  'z1(01A0BBhi ,<=H ?3/H  !*c*%33J? *=H  (C(#11(;(;H~~r+cFtjjS)z*Get current datetime as ISO format string.)rnowrr*r+r,rFz#WalletService._get_current_datetimes||~''))r+)r0z'WalletRepositoryProtocol')NN)rHrrIrrrrJrr rrKOptional[Dict[str, Any]]rrreturnr)NFNr)rHrrIrrrrJrr rr:boolrKrrrrr)rHrrIrrNone)rprrHrrIrrrrJrr rr:rrKr#rr) NNNNNNNNr)rHrrI str | Noner rrrrrrdatetime | str | Nonerrrrrrrzlist[TransactionRecord])rRrrzTransactionRecord | None)NNNNNNN)rHrrIrr rrrrrrrrrrr)rr)r%r&r'r(rmr2rSr`rfr?rrr staticmethodrFr*r+r,r.r.(sq O%"&+/FFF F  F  F)FF F\"&$+/[[[ [  [  [[)[[ [z ))) )  )  )))) )Z#!%!%'+,0*. ||| | % |  |*|(||| !||>#!%!%'+,0*.@ @ @  @ % @  @ *@ (@  @ D**r+r.N)r( __future__r dataclassesrrdecimalrrtypingrr r r exceptionsr rrrdjango.db.modelsrr0rrr.r*r+r,rsY,"!*55&4 $    \*\*r+