a 9-e(B@sdZddlZddlmZddlmZddlZddlm Z ddl m Z m Z ddl mZmZddlmZdd lmZmZmZmZmZmZGd d d eZdS) zC Created on Thu Aug 17 23:50:16 2017 @author: IBridgePy@gmail.com N)exit)Decimal)IBCpp)-stripe_exchange_primaryExchange_from_securitychoose_whatToShow) SymbolStatus OrderStatus)from_contract_to_security) CancelOrderReqHistoricalData PlaceOrder ReqMktDataReqScannerSubscriptionCancelScannerSubscriptionc@seZdZddZeddZddZeddZed d Zed d Z d dZ ddZ daddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zdbd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zdcd7d8Zd9d:Zd;d<Z d=d>Z!d?d@Z"dAdBZ#dddGdHZ$dIdJZ%dKdLZ&dedMdNZ'dfdOdPZ(dQdRZ)dgdSdTZ*dhdUdVZ+didWdXZ,djdYdZZ-d[d\Z.d]d^Z/d_d`Z0dCS)k BrokerServicecCs8||_||_||_||_||_|jtdd|_dS)Nz ::__init__F) _userConfig_log_timeGenerator _singleTrader_dataFromServernotset__name___asDataProviderService)self userConfiglog timeGenerator singleTraderdataFromServerrG:\My Drive\STUDY\EPAT\09 TBP - Trading & Back-testing Platforms\TBP04 - Backtesting & Live Trading\IB IBridgePy API\IBridgePy_Win_Anaconda37_64\broker_service_factory\BrokerService.py__init__szBrokerService.__init__cCs|jjSN)r brokerClientrrrr _brokerClient'szBrokerService._brokerClientcCsd|jt||j|jfS)Nz0{name=%s id=%s brokerClient=%s timeGenerator=%s})nameidr%rr$rrr __str__+szBrokerService.__str__cCst|jdSzB Name of the broker :return: string name NNotImplementedErrorr&r$rrr r&.szBrokerService.namecCst|jdSr)r*r$rrr brokerName7szBrokerService.brokerNamecCs|jjSr")r% versionNumberr$rrr r-@szBrokerService.versionNumbercCs|jSr")r%r$rrr getBrokerClientDszBrokerService.getBrokerClientcCs d|_dS)NT)rr$rrr setAsDataProviderServiceGsz&BrokerService.setAsDataProviderServicevaluecCstdSr")r+)r accountCodetagmetarrr _get_account_info_one_tagJsz'BrokerService._get_account_info_one_tagcCsV|jtd||f|j||d}|dur8d}|jtd|||f|S)NzC::_get_real_time_price_from_dataFromServer: security=%s tickType=%spricezT::_get_real_time_price_from_dataFromServer: security=%s tickType=%s returnedValue=%s)rrr full_printr get_valuedebug)rsecuritytickTypeansrrr (_get_real_time_price_from_dataFromServerMs z6BrokerService._get_real_time_price_from_dataFromServercCs(|jtd||f|j||dS)Nz/::_from_dataFromServer: security=%s tickType=%ssize)rr9rrr8rr:r;rrr '_get_real_time_size_from_dataFromServerVsz5BrokerService._get_real_time_size_from_dataFromServercCs"|jtd||j|S)Nz;::_get_5_second_real_time_bar_from_dataFromServe: security=)rrrrget_5_second_real_time_barrr:rrr /_get_5_second_real_time_bar_from_dataFromServerZsz=BrokerService._get_5_second_real_time_bar_from_dataFromServercCs4t|}|jtd|f|j|j||S)z Guarantee to return a PositionRecord even if there is no current position. :param accountCode: :param security: :return: Guarantee to return a PositionRecord z ::_get_position: adj_security=%s)rrr9rr7r get_positionr,)rr1r:Z adj_securityrrr _get_position^szBrokerService._get_positioncCs@|j|j|}i}|D]"}||j}t|}||||<q|Sr")rget_all_positionsr,contractr )rr1Z allPositionsr<Z str_securityrGr:rrr _get_all_positionsis z BrokerService._get_all_positionscCs&|jtd|f|j|j|S)zx :param accountCode: :return: dict Keyed by ibpyOrderId, value = models::Order::IbridgePyOrder z!::_get_all_orders: accountCode=%s)rr9rrget_all_ordersr,rr1rrr _get_all_orderstszBrokerService._get_all_orderscCsP|jtd|f|jtjkr4|j|n|j ||j |dS)Nz7::add_exchange_primaryExchange_to_security: security=%s) rr9rr7 symbolStatusr SUPER_SYMBOLr%Zappend_security_infoZadd_primaryExchange_to_securityZadd_exchange_to_securityrBrrr (add_exchange_primaryExchange_to_security}s   z6BrokerService.add_exchange_primaryExchange_to_securitycCs(|jtd|f|t|dS)zj Cancel order by its ibpyOrderId :param ibpyOrderId: string :return: None z::cancel_order: ibpyOrderId=%sN)rr9rsubmit_requestsr r ibpyOrderIdrrr cancel_orderszBrokerService.cancel_ordercCs|jtd|jS)Nz ::connect)rr9rr%ZconnectWrapperr$rrr connectszBrokerService.connectcCs |jtd|jdS)Nz::disconnect: )rr9rr%disconnectWrapperr$rrr disconnectszBrokerService.disconnectcCs|jtd|jS)Nz::get_next_time)rrrr get_next_timer$rrr rVszBrokerService.get_next_timecCsd|jtd||ft|tr0||||St|tr`g}|D]}|||||qB|SdS)Nz*::get_account_info: accountCode=%s tags=%s)rr9r isinstancestrr4listappend)rr1tagsr3r<r2rrr get_account_infos  zBrokerService.get_account_infocCs$|j}|jtd|f|S)zN Get IB server time :return: datetime in UTC timezone z::get_datetime=%s)rZget_current_timerrr)rr<rrr get_datetimes zBrokerService.get_datetimecCs|jtd|j|jS)z get a list of accountCodes from IB server to check if the user-input accountCode is acceptable. :return: z::get_active_accountCodes)rr9rrZget_all_active_accountCodesr,r$rrr get_active_accountCodessz%BrokerService.get_active_accountCodescCsf|jtd|f||}g}|D]8}||j}|tjtjtjtj tj fvr(| ||q(|S)Nz%::get_all_open_orders: accountCode=%s) rr9rrIstatusr APIPENDING PENDINGSUBMIT PENDINGCANCEL PRESUBMITTED SUBMITTEDrZ)rr1Z allOrdersr<rQr_rrr get_all_open_orderss  z!BrokerService.get_all_open_orderscCst|jdSr"r*rJrrr rIszBrokerService.get_all_orderscCst|jdS)z Get all of positionRecords associated with the accountCode :param accountCode: :return: dictionary, keyed by Security object with exchange info!!!, value = PositionRecord Nr*rJrrr rFszBrokerService.get_all_positionscCst|jdSr"r*)rr:fieldwaitForFeedbackInSecondsrrr get_contract_detailssz"BrokerService.get_contract_detailscCs |jSr")r%get_heart_beatsr$rrr rjszBrokerService.get_heart_beatscCs |jS)zO :return: int, number of days of IBridgePy passcode to expiry )r%Zget_authed_expiryr$rrr get_ibpy_expiry_in_dayssz%BrokerService.get_ibpy_expiry_in_dayscCs^gd}i}|D]}||vr||||<q|t||d}|t|d|j|dS)N)Z numberOfRowsZ instrumentZ locationCodeZscanCodeZ abovePriceZ belowPriceZ aboveVolumeZmarketCapAboveZmarketCapBelowZmoodyRatingAboveZmoodyRatingBelowZ spRatingAboveZ spRatingBelowZmaturityDateAboveZmaturityDateBelowZcouponRateAboveZcouponRateBelowZexcludeConvertibleZaverageOptionVolumeAboveZscannerSettingPairsZstockTypeFilterrhr)rOrrr%get_submit_requests_result)rrhkwargstagListZ subscriptionct reqIdListrrr get_scanner_resultssz!BrokerService.get_scanner_resultscCs |jSr")r%"get_TD_access_token_expiry_in_daysr$rrr rssz0BrokerService.get_TD_access_token_expiry_in_dayscCs |jSr")r%get_new_TD_refresh_tokenr$rrr rtsz&BrokerService.get_new_TD_refresh_tokenNTc Cs|jtd|||||||f|durt|dkrt|jdurX|jtd|ft|jtjd}t j |d}|dkrt |j }|t||||||||d} |j| dS) a :param followUp: :param security: Security :param barSize: string barSize can be any of the following values(string) 1 sec, 5 secs,15 secs,30 secs,1 min,2 mins,3 mins,5 mins,15 mins,30 mins,1 hour,1 day :param goBack: string :param endTime: default value is '', IB server deems '' as the current server time. If user wants to supply a value, it must be a datetime with timezone :param whatToShow: string whatToShow: see IB documentation for choices TRADES,MIDPOINT,BID,ASK,BID_ASK,HISTORICAL_VOLATILITY,OPTION_IMPLIED_VOLATILITY :param useRTH: int 1=within regular trading hours, 0=ignoring RTH :param waitForFeedbackInSeconds: :return: a dataFrame, keyed by a datetime with timezone UTC, columns = ['open', 'high', 'low', 'close', 'volume'] The latest time record at the bottom of the dateFrame. zv::get_historical_data: security=%s barSize=%s goBack=%s endTime=%s whatToShow=%s useRTH=%s waitForFeedbackInSeconds=%sNruz>::request_historical_data: EXIT, endTime=%s must have timezone)tzz%Y%m%d %H:%M:%S %Z)followUpr)rr9rtzinfoerrorr astimezonepytzutcdtdatetimestrftimerZsecTyperOr r%rm) rr:barSizeZgoBackendTime whatToShowZuseRTHrhrxZ orderIdListrrr get_historical_datas"  z!BrokerService.get_historical_datacCst|jdS)zj :param ibpyOrderId: string :return: broker_factory::records_def::IBridgePyOrder Nr*rPrrr get_orderszBrokerService.get_ordercCst|jdSr"r*)rr1r:rrr rDszBrokerService.get_positioncCst|jdSr"r*rr:r;rxrrr get_real_time_price"sz!BrokerService.get_real_time_pricecCst|jdSr"r*rrrr rA%sz(BrokerService.get_5_second_real_time_barcCsttd|jfdS)Nz::get_timestamp: %s)r+rr&r?rrr get_timestamp(szBrokerService.get_timestampcCst|jdSr"r*rrrr get_real_time_size+sz BrokerService.get_real_time_sizecCst|jdSr"r*)rrQZ target_statusZwaitingTimeInSecondsrrr order_status_monitor.sz"BrokerService.order_status_monitorcCs|jtd|t|f|j|s>|t||di}|D]}|dvrl|j |t j j |||<qF|dvr|j |t j jd||<qF|dvr|j |t j jd||<qFttd||fqF|S)Nz(::get_option_info: security=%s fields=%srl)deltagammaZvegathetaZ impliedVolZoptPriceZundPrice)Zoption_call_open_interestr>)Zoption_put_open_interestz?::get_option_info: EXIT, cannot handle security=%s fieldName=%s)rr9rr7rXr%Zis_real_time_price_requestedrOr rr8rZTickTypeZLAST_OPTION_COMPUTATIONZOPTION_CALL_OPEN_INTERESTZOPTION_PUT_OPEN_INTERESTr)rr:fieldsrhr< fieldNamerrr get_option_info1s  zBrokerService.get_option_infocCsl|jtd|f|j}|j}t|ddkrX|t||||d}|j |dS|j ddSdS)Nz ::place_order: ibridgePyOrder=%sZ totalQuantityr)rGorderrxrhz4No order was placed because order.totalQuantity = 0.Z NoOrderPlaced) rr9rZrequestedContractZrequestedOrderrrOr r%rminfo)rZibridgePyOrderrxrhrGrrqrrr place_orderCs zBrokerService.place_ordercCs |j|Sr")r%ZprocessMessagesWrapper)rZtimeNowrrr processMessagesNszBrokerService.processMessagescGs"|jtd||jj|S)zu :param args: broker_client_factory::BrokerClientDefs::Request :return: a list of reqId !!! z::submit_requests: args=)rrrr%Z request_data)rargsrrr rORszBrokerService.submit_requestscCs |jSr")r% use_next_idr$rrr r[szBrokerService.use_next_id)r0)r0)rf)NrurvrfT)T)T)T)rf)rf)Trf)1r __module__ __qualname__r!propertyr%r(r&r,r-r.r/r4r=r@rCrErHrKrNrRrSrUrVr\r]r^rerIrFrirjrkrrrsrtrrrDrrArrrrrrrOrrrrr rsf               !       r)__doc__rr~sysrdecimalrr|Z IBridgePyrIBridgePy.IbridgepyToolsrrIBridgePy.constantsrrIBridgePy.quantopianr Z&broker_client_factory.BrokerClientDefsr r r r rrobjectrrrrr s