U -Phm@spddlmZddlmZddlmZddlmZmZddl m Z ddl m Z ddl Z e eZGdd d ZdS) )Jobbing)Strategy) BrokerAccount)getUniqueStruct getItemByKey) Strategis)HistoricalDataNc@sXeZdZdZdddddZdddZdd Zdd d Zdd dZdddZ ddZ dS)StrategyManagerz Strategy Manager is an overall central component that manages all the strategy for a user it contains all the objects relevant objeccts viz. all strategy objects with setup, all borker account and broker conneciton to perform its task. N)returncCs||_||_dS)z initialize manager with optional strategy code. strategy coe is used to instantiate object for specific strategy and perform operations without affecting other strategy in place. N)Usercode)selfuserr rM/home/worker/MarketYantra/Yantra/yantra/trading/Strategies/strategyManager.py__init__szStrategyManager.__init__c Csg}g}tj}|jr&|j|jd}|D]*}t|j|j|}|r*|jr*| |q*|D]"}|j D]}||krd| |qdqZ|r||_ |}g}|D]}|d|kr| |q||_ ||_ |S)zP getting all strategy objects and relavant setups as dictionary )r exchange) robjectsallr filterrZgetStrategyInstancer isActiveappendZ ActiveScriptsrActiveStrategies) r rZlstActiveStrategiesZlstActiveScriptsZ strategiesZstratstrategyscriptZlstTemprrrgetActiveStrategies"s0     z#StrategyManager.getActiveStrategiesc Cstd|jd|jd|j}t|}|jD]2}|jD]&}|j |}| |j t |d|q:q0td|jd|jddS)a8 subscribe to the websocket for getting messages for Order update or/and price changes this will subscribe to the sockets depending on the respective strategies settings. The sockets read the input and publish it over the redis pubsub. user broker object QueueId to subscribe to zstart: Websocket subscription z strategy [] accountIdzEnd: Websocket subscription N) loggerdebugr r TicksForCallbackrrZActiveAccounts brokerAccountgetBrokerObjectsubscribeWebSocketsZrequireOrderCallbackr)r lstTicksZ lstUniquerrZ brokerObjrrrr#Js   z#StrategyManager.subscribeWebSocketsTc Cszg}g}||t|j}|||_|jD]8}||}t d||D]}| |qZq2||_ |r~| Wn6t k r} zt dt| W5d} ~ XYnXdS)NzTicks zinitiate Error )rrr Connectr!rinitiateZgetScriptForTickCallbackrrrr r# Exceptionstr) r r subscriber$ZlstTickForCallbackZ brokerAccrtickstickerrrr&[s$    zStrategyManager.initiatecCs|jD]}||j|qdSN)rstartr!r rrrrrr.us zStrategyManager.startcCs|jD]}||j|qdSr-)rresyncStrategyValuesr!r/rrrr0}s z$StrategyManager.resyncStrategyValuesc Cs&t|j|}|||||d||} | S)Nr)rr!ZgetHistoricCandleData) r rexchSegsymboltokenfromdatetodateintervalZhistZhistDatarrrgetHistoricalDatas z!StrategyManager.getHistoricalData)N)N)NT)N)N) __name__ __module__ __qualname____doc__rrr#r&r.r0r7rrrrr s (   r )trading.Strategies.jobbingrZtrading.Strategies.strategyr-trading.Entities.BrokerAccounts.brokerAccountrZ utils.viewsrrtrading.modelsrZ&trading.Entities.Report.historicalDatarlogging getLoggerr8rr rrrrs