B 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\02 Python 3.7 IBridgePy_Win_Anaconda37_64\broker_service_factory\BrokerService.py__init__szBrokerService.__init__cCs|jjS)N)r brokerClient)rrrr _brokerClient'szBrokerService._brokerClientcCsd|jt||j|jfS)Nz0{name=%s id=%s brokerClient=%s timeGenerator=%s})nameidr#r)rrrr __str__+szBrokerService.__str__cCst|jdS)zB Name of the broker :return: string name N)NotImplementedErrorr$)rrrr r$.szBrokerService.namecCst|jdS)zB Name of the broker :return: string name N)r'r$)rrrr brokerName7szBrokerService.brokerNamecCs|jjS)N)r# versionNumber)rrrr r)@szBrokerService.versionNumbercCs|jS)N)r#)rrrr getBrokerClientDszBrokerService.getBrokerClientcCs d|_dS)NT)r)rrrr setAsDataProviderServiceGsz&BrokerService.setAsDataProviderServicevaluecCstdS)N)r')r accountCodetagmetarrr _get_account_info_one_tagJsz'BrokerService._get_account_info_one_tagcCsV|jtd||f|j||d}|dkr8d}|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)rr5rrr4)rr6r7rrr '_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_bar)rr6rrr /_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)rrr5rr3r get_positionr()rr-r6Z adj_securityrrr _get_position^szBrokerService._get_positioncCsD|j|j|}i}x*|D]"}||j}t|}||||<qW|S)N)rget_all_positionsr(contractr )rr-Z allPositionsr8Z str_securityrAr6rrr _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)rr5rrget_all_ordersr()rr-rrr _get_all_orderstszBrokerService._get_all_orderscCsP|jtd|f|jtjkr4|j|n|j ||j |dS)Nz7::add_exchange_primaryExchange_to_security: security=%s) rr5rr3 symbolStatusr SUPER_SYMBOLr#Zappend_security_infoZadd_primaryExchange_to_securityZadd_exchange_to_security)rr6rrr (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)rr5rsubmit_requestsr )r ibpyOrderIdrrr cancel_orderszBrokerService.cancel_ordercCs|jtd|jS)Nz ::connect)rr5rr#ZconnectWrapper)rrrr connectszBrokerService.connectcCs |jtd|jdS)Nz::disconnect: )rr5rr#disconnectWrapper)rrrr disconnectszBrokerService.disconnectcCs|jtd|jS)Nz::get_next_time)rrrr get_next_time)rrrr rNszBrokerService.get_next_timecCsh|jtd||ft|tr0||||St|trdg}x |D]}|||||qDW|SdS)Nz*::get_account_info: accountCode=%s tags=%s)rr5r isinstancestrr0listappend)rr-tagsr/r8r.rrr 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)rr8rrr 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)rr5rrZget_all_active_accountCodesr()rrrr get_active_accountCodessz%BrokerService.get_active_accountCodescCsj|jtd|f||}g}x@|D]8}||j}|tjtjtjtj tj gkr*| ||q*W|S)Nz%::get_all_open_orders: accountCode=%s) rr5rrCstatusr APIPENDING PENDINGSUBMIT PENDINGCANCEL PRESUBMITTED SUBMITTEDrR)rr-Z allOrdersr8rIrWrrr get_all_open_orderss   z!BrokerService.get_all_open_orderscCst|jdS)N)r'r$)rr-rrr rCszBrokerService.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 N)r'r$)rr-rrr r@szBrokerService.get_all_positionscCst|jdS)N)r'r$)rr6fieldwaitForFeedbackInSecondsrrr get_contract_detailssz"BrokerService.get_contract_detailscCs |jS)N)r#get_heart_beats)rrrr rbszBrokerService.get_heart_beatscCs |jS)zO :return: int, number of days of IBridgePy passcode to expiry )r#Zget_authed_expiry)rrrr get_ibpy_expiry_in_dayssz%BrokerService.get_ibpy_expiry_in_dayscCsddddddddd d d d d ddddddddg}i}x |D]}||kr8||||<q8W|t||d}|t|d|j|dS)NZ numberOfRowsZ instrumentZ locationCodeZscanCodeZ abovePriceZ belowPriceZ aboveVolumeZmarketCapAboveZmarketCapBelowZmoodyRatingAboveZmoodyRatingBelowZ spRatingAboveZ spRatingBelowZmaturityDateAboveZmaturityDateBelowZcouponRateAboveZcouponRateBelowZexcludeConvertibleZaverageOptionVolumeAboveZscannerSettingPairsZstockTypeFilter)r`r)rHrrr#get_submit_requests_result)rr`kwargstagListZ subscriptionct reqIdListrrr get_scanner_resultss  z!BrokerService.get_scanner_resultscCs |jS)N)r#"get_TD_access_token_expiry_in_days)rrrr rjsz0BrokerService.get_TD_access_token_expiry_in_dayscCs |jS)N)r#get_new_TD_refresh_token)rrrr rksz&BrokerService.get_new_TD_refresh_tokenNTc Cs|jtd|||||||f|dk rt|dkrt|jdkrX|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=%sNrlz>::request_historical_data: EXIT, endTime=%s must have timezone)tzz%Y%m%d %H:%M:%S %Z)followUpr)rr5rtzinfoerrorr astimezonepytzutcdtdatetimestrftimerZsecTyperHr r#rd) rr6barSizeZgoBackendTime whatToShowZuseRTHr`roZ orderIdListrrr get_historical_datas"  z!BrokerService.get_historical_datacCst|jdS)zj :param ibpyOrderId: string :return: broker_factory::records_def::IBridgePyOrder N)r'r$)rrIrrr get_orderszBrokerService.get_ordercCst|jdS)N)r'r$)rr-r6rrr r>szBrokerService.get_positioncCst|jdS)N)r'r$)rr6r7rorrr get_real_time_price"sz!BrokerService.get_real_time_pricecCst|jdS)N)r'r$)rr6r7rorrr r<%sz(BrokerService.get_5_second_real_time_barcCsttd|jfdS)Nz::get_timestamp: %s)r'rr$)rr6r7rrr get_timestamp(szBrokerService.get_timestampcCst|jdS)N)r'r$)rr6r7rorrr get_real_time_size+sz BrokerService.get_real_time_sizecCst|jdS)N)r'r$)rrIZ target_statusZwaitingTimeInSecondsrrr order_status_monitor.sz"BrokerService.order_status_monitorcCs|jtd|t|f|j|s>|t||di}x|D]}|dkrn|j |t j j |||<qH|dkr|j |t j jd||<qH|dkr|j |t j jd||<qHttd||fqHW|S)Nz(::get_option_info: security=%s fields=%s)r`)deltagammaZvegathetaZ impliedVolZoptPriceZundPrice)Zoption_call_open_interestr:)Zoption_put_open_interestz?::get_option_info: EXIT, cannot handle security=%s fieldName=%s)rr5rr3rPr#Zis_real_time_price_requestedrHr rr4rZTickTypeZLAST_OPTION_COMPUTATIONZOPTION_CALL_OPEN_INTERESTZOPTION_PUT_OPEN_INTERESTr)rr6fieldsr`r8 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)rAorderror`z4No order was placed because order.totalQuantity = 0.Z NoOrderPlaced) rr5rZrequestedContractZrequestedOrderrrHr r#rdinfo)rZibridgePyOrderror`rArrhrrr place_orderCs zBrokerService.place_ordercCs |j|S)N)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 rHRszBrokerService.submit_requestscCs |jS)N)r# use_next_id)rrrr r[szBrokerService.use_next_id)r,)r,)r^)Nrlrmr^T)T)T)T)r^)r^)Tr^)1r __module__ __qualname__r!propertyr#r&r$r(r)r*r+r0r9r;r=r?rBrDrGrJrKrMrNrTrUrVr]rCr@rarbrcrirjrkr{r|r>r}r<r~rrrrrrHrrrrr rs\                    r)__doc__rvrusysrdecimalrrsZ IBridgePyrIBridgePy.IbridgepyToolsrrIBridgePy.constantsrrIBridgePy.quantopianr Z&broker_client_factory.BrokerClientDefsr r r r rrobjectrrrrr s