thv(LddlZddlZddlmZmZddlmZmZdZdZdZ dZ y)N)datetime timedelta) current_appurl_forc tjdt_tjs t dddlm}|jj|j}|st d||dvr |jn |j}d|} tjj|}d|d |}tj&j)|j*d } d} | j,D]C} | j.j1d |k(s"| j.j1d|k(sA| } n| sGtj&j!|j*t3|dzd|dvrdnddd|||d} tjdd} tjdd} tj4j6j!| | dgdt9|| j*ddg|||d }|j:|j*d!S#tjj$r?tjj!d |j"|j$d |i }YwxYw)"a# Create a Stripe checkout session for subscription Args: user_id (int): User ID plan_id (str): Plan ID (basic, professional, enterprise) billing_cycle (str): Billing cycle (monthly or yearly) Returns: dict: Checkout session data STRIPE_SECRET_KEY6Stripe API key is not set in application configurationr)SubscriptionPlan)plan_idzInvalid plan ID: yearlyannualz sqlgenai-z SQLGenAI r )name descriptionmetadata-T)productactiveN billing_cycledusdyearmonth)intervalinterval_count)r rprice_lookup_id)r unit_amountcurrency recurringr FRONTEND_URLz6/subscription/success?session_id={CHECKOUT_SESSION_ID}z/subscription/cancelcard subscription)pricequantity)user_idr r) success_url cancel_urlpayment_method_typesmodeclient_reference_id line_itemsr) checkout_url checkout_id)rconfigstripeapi_key ValueErrorapp.auth.modelsr query filter_byfirst annual_price monthly_priceProductretrieveerrorInvalidRequestErrorcreaterrPricelistiddatargetintcheckoutSessionstrurl)r&r rr plan price_amount product_idrrexisting_pricesr$pr'r(checkout_sessions '/var/www/html/app/utils/stripe_utils.pycreate_checkout_sessionrO s!''(;>QRR0  ! ! + +G + < B B DD ,WI677)69M(M4$$SWSeSeLWI&J  ..))*5"'!M?;Oll'' 4'HO E ! ! JJNN9 % 0 JJNN? +} <E    ##JJL3./&37K&KFQX"# #!.#2$   !''788pqK&&~677KLJ..55$X L  * 6(),,'**  << + + ..''TYYK(((7(  sHAI65I6c B tjdt_tjs tjj dyddlm}ddlm }m }m }|d}|dd }|d k(rt|jd ijd }|jd ijd }|jd ijd} |r|r| s-tjj d|jy|jd} | s-tjj d|jytjj| } d} |j j#|j%} | r|| _| dv| _d| _| | _t/j0| j2| _t/j0| j6| _| } |j:j=n|||| dvd| t/j0| j2t/j0| j6}|j:j?||j:j=|j:jA||} ||| j|jdddz |jddjCt/j0|jdd|jdd}|j:j?||j:j=y |d!k(r|jd"} | sy|j j#| #j%}|sy|jd$d|_t/j0|jd%|_t/j0|jd&|_|j:j=y |d'k(rc|jd"} | sy|j j#| #j%}|syd(|_|j:j=y y))z Handle Stripe subscription events Args: event: Stripe event object Returns: bool: True if event was handled successfully rr Fr)db) SubscriptionPaymentHistoryUsertyperAobjectzcheckout.session.completedrr&r rz&Missing metadata in checkout session: r#z(No subscription ID in checkout session: N)r&r r)r&r is_annualstatussubscription_id start_dateend_date amount_totalrrrcreated credit_cardpayment_intentsuccess)r&rYamountr payment_datepayment_methodtransaction_idrXTzcustomer.subscription.updatedr@)rYrXcurrent_period_startcurrent_period_endzcustomer.subscription.deleted cancelled)"rr/r0r1loggerr;app.extensionsrQr3rRrSrTrCrBr@r:r4r5r6r rWrXrYr fromtimestamprerZrfr[sessioncommitaddrefreshupper)eventrQrRrSrT event_type data_objectr&r rrYstripe_subscriptionsubscription_objexisting_subscriptionnew_subscriptionpaymentr#s rNhandle_subscription_eventrxms!''(;>  !YZ!BBvJ-)K11kooj"599)DE//*b155i@# B7;;OL g]    $ $'MknnM]%^ _&//.9    $ $'OP[P^P^O_%` a%11::?K !- 2 2 < 15; __Z7==?!//  0JK(&??+;<   w  6 6%//$/$))33O3TZZ\ *oohA "*"8"8I_9`"a  ( 6 6{G[7\ ]   6 6%//$/$))33O3TZZ\ *   cttjdt_tjs tjj dy tj j|y#tj j$r,}tjj d|Yd}~yd}~wwxYw)z Cancel a Stripe subscription Args: subscription_id (str): Stripe subscription ID Returns: bool: True if subscription was cancelled successfully rr FTz&Error cancelling Stripe subscription: N) rr/r0r1rhr;rRdelete StripeError)rYes rNcancel_stripe_subscriptionr~s!''(;>  !YZ""?3 << # #  #I!!MNsA..B7 "B22B7cvtjdt_tjs tjj dy tj j|}|S#tj j$r,}tjj d|Yd}~yd}~wwxYw)z Get subscription data from Stripe Args: subscription_id (str): Stripe subscription ID Returns: dict: Subscription data rr Nz&Error retrieving Stripe subscription: ) rr/r0r1rhr;rRr:r|)rYr#r}s rNget_subscription_datars!''(;>  !YZ**33OD  << # #  #I!!MNs A//B8 "B33B8) r0osrrflaskrrrOrxr~rryrNrs) (& aF@D.ry