o Hi<@sddlmZmZmZmZddlmZddlmZddl Z ddl m Z m Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZddlZddl m Z ddl!m"Z"ddl#Z#ddlZeZ$e%e&Z'Gddde(e"Z)ddZ*ddZ+de,de(de(fddZ-de,de,fddZ.eej/eefdefddZ0e$j1dd d!eej/eeee0fd"efd#d$Z2e$1d%eej/eeee0fd&e(d"efd'd(Z3e$4d)eej/eefd*e,fd+d,Z5e$4d-d*e,fd.d/Z6e$1d0eej/eefd1d2Z7e$1d3eej/eeee0ee)j8d4d5fd"ed6e)fd7d8Z9e$j1d9d:d!ed;dd5eej/eeee0fd6e(d?e:fd@dAZ;e$4dBeej/fdCe(d*e,fdDdEZ%sz"sanitize_numpy..cSsg|]}t|qSrr(r*irrr 'sz"sanitize_numpy..) isinstancedictitemslistnpintegerint32int64intfloatingfloat32float64floatrstr)objrrrr)#s   r) signal_data timeframereturncs~|dis dSgd}fdd|DtfddttdDr*d Stfd dttdDr=d Sd S) z1 Get EMA status for a specific timeframe ema_stackz No EMA data) 7dcsg|] }d|dqS)ema_N)get)r*period)rCrrr0:sz"get_ema_status..c3s$|] }||dkVqdSNrr. ema_valuesrr <"z!get_ema_status..rPBullishc3s$|] }||dkVqdSrOrr.rQrrrS>rTBearishMixed)rMallrangelen)r@rA ema_periodsr)rCrRrget_ema_status0s ""r\c Cs|d|d|d|ddit|dd|ddit|dd|ddit|dd|dd it|d dd |d d |d d |d d |d d dd|dd|ddS)z'Enhance signal data for UI presentationsymboldecisionconfidence_scorepivots30min)pivot ema_statusdayweekmonth)radailyweeklymonthly yearly_highr yearly_low)upperlower)rjrkcircuit_limitsreasonr timestamp)r]r^ confidence timeframes key_levelsrorp)rMr\)r@rrrformat_ui_signalCs2     rtcCsVt|d}|dd|i}t||stdddt|d|d|d d d S) N_idzerodha_settingsuser_idz#Zerodha settings not found for user status_codedetailr$r%r&rr#)r>rMfind_oneprintrr )db current_userrwsettingsrrrget_zerodha_client_from_dbgs  rz/signalsz#Get signals for all user portfolios)summaryzerodhac sz;t|d}t|}t||}|dd|i}g}|D]}|d} | r2|| } || qd|t||dWStyP} zt dt| dd} ~ ww) Nru portfoliorwr]success)statusrwsignals_generatedsignalsry) r>rMr r findanalyze_symbolappendrZ Exceptionr) r~rrrw db_managerengine portfoliosrrr]signalerrrget_all_portfolio_signalsys,    rz/signals/{portfolio}rc sz5t|d}t|}t||}td|d||}||}td|d|d||t|dWStyJ} ztdt| d d} ~ ww) NruzFetching signal for portfolio: z for user: z Signal for : r)rrr]rrry) r>rMr r r}rr)rr) rr~rrrwrrr]rrrrrget_single_portfolio_signals$  rz /callbackpayloadcCs|d}t|d}|stddd|dd|i}|s%tdd dt|d |d }||}|dd|id d |d iiddiS)N request_tokenruizMissing request tokenryrvrwrxzZerodha settings not foundr$r%$setr&r connected)rMr>rr|r generate_session update_one)rr~rrrwrclientsessionrrr post_callbacks    rz /postbackc sTztd|td|ddiWSty)}ztdt|dd}~ww)NzZerodha postback received: %szerodha_postbacksrreceivedrry)logginginfor get_mongo_db insert_onerrr>)rrrrr kite_postbacks  rz/profilecCs||ddt|di}|r|dstdddt|d|d |d}z |}d |d WSty=tdd dw) Nrvrwrur&izNot authenticated with Zerodharyr$r%ok)rprofilez$Invalid/expired Zerodha access token)r|r>rMrr get_profiler)r~rrrrrrrget_zerodha_profiles    rz/livez:Which top movers to run the strategy on: gainers or losers) descriptionmovercs|d}td|d||tjkrt|}nt|}|s,tdddgdSzt|j|j |j |d}| |}g}|D]9} | t || d| d | d | d | d | d | dd| d| d| dd| d| dddddd qAg} |r|d|} | j} g} t| |D]>\} } | t| | d| d| d | d | d | d | d| d| d| d | d| d| dd | d| dddqdt|| dWSty}ztjd|d |d!d"td#d$d%d}~ww)&Nruz Running z live strategy for user: zNo valid symbols foundrrmessageresults)r$r%r&r~r] volume_dropema5_gapdivergence_confirmeddivergence_bullnear_resistance near_support)rrbear_divergencebull_divergencerr bear_score bull_scorer^)rrr^ entry_price target_price)entryexit) rprwr]featuresmetricstargetsoutcome prediction user_feedbackstreaminstrument_token volume_rise ema5_gap_bull) rema_gap divergencerrr ema_gap_bullrrrr^)idr]rrrrscannedrz#Strategy execution failed for user rTexc_inforzStrategy execution failedry)rMloggerrrrrrrr$r%r& run_strategyrrutcnow insert_many inserted_idszipr>rZrerrorr)r~rrrrwsymbolsstrategyrawanalytics_docsridsinsertedridxrrrrrun_live_strategys          rz /shortsellsz Run intraday short-sell strategyrz!Run on 'gainers' or 'losers' listgj@uBase capital in ₹capitalcs|dkr t|nt|}|sddgdSt|j|j|j||d}||}g}t|ddD]\} } zpt| d} t| d } t | d } t | | | d }| d d }t | dd}t| dd}t| dd}t| dd}|||t | ddt | dd||d}| | | d| d| | | ||| dgd Wq.ty}ztjd| dd|ddWYd}~q.d}~wwdt||dS) NrrzNo symbols to scanr)r$r%r&r~rrP)startrrquantity pivot_levelrnear_upper_circuitFvolume_drop_pctgmomentum_drop_pct ema5_gap_pctrsi_divergence rising_wedge)r momentum_droprrwedgerrr]r skip_reasons) rankr]r sell_entry cover_targetrexpected_profitrrzError shaping result for rTrr)rrrr$r%r&r enumerater=r9roundrMboolrrrrrZ)rrr~rrrrrrrrrrqty expected_pnlnearestupper_okrrrrrrrrrun_intraday_shortsQsn            rz/stream/feedback/{id}rcCs2|d}|ddt|idd|iiddiS)Nfeedbackrrurrrr)rMrr)rrr~fbrrrrs   r)=fastapirrrrfastapi.encodersrbsonrnumpyr5typingrr app.dbr app.v1.services.zerodha.clientr app.v1.services.zerodha.enginer "app.v1.services.zerodha.db_managerr app.v1.dependencies.authr!app.v1.services.zerolive.strategyr app.v1.services.zerolive.bearishrapp.v1.services.zerolive.listrrrrenumrr!router getLoggerrrr>rr'r)r2r\rtrrrMrrpostrrrrrr=rrrrrrs             %       i   O"