o gj@sFdZddlZddlZddlZddlmZddlmZGdddZ dS)z This file contains functions to create a client class that accesses the Schwab api Coded by Tyler Bowers Github: https://github.com/tylerebowers/Schwab-API-Python N)Stream)Tokensc%@sJeZdZdkddZdefd d Zdld dZdeeBd BfddZ dZ de j fddZ dmdede j fddZdmdedede j fddZdndedejeBdejeBdedede j f d d!Zded"ede j fd#d$Zded%eeBde j fd&d'Zded%eeBde j fd(d)Zded%eeBd"ede j fd*d+ZdndejeBdejeBdedede j f d,d-Z dmded.ejeBd/ejeBd0ed1ede j f d2d3Zded4eeBde j fd5d6Zde j fd7d8Z dod9eeeBded:ede j fd;d<Zdmd=edede j fd>d?Z dpd1ed@edAedBedCedDedEedFedGejeBdHejeBdIedJedKedLedMedNedOede j f$dPdQZd1ede j fdRdSZ dqd1edTedUedVedWed.ejeBd/edXedYede j fdZd[Z!dnd1ed\edWede j fd]d^Z"dmd9eed_ejeBde j fd`daZ#dmdbed_ejeBde j fdcddZ$d1edeede j fdfdgZ%dheeBde j fdidjZ&d S)rClienthttps://127.0.0.1 tokens.jsonFTcCsX|dkrtdd|_||_||_t||||||d|_t||_|jr*tddSdS)ah Initialize a client to access the Schwab API. :param app_key: app key credentials :type app_key: str :param app_secret: app secret credentials :type app_secret: str :param callback_url: url for callback :type callback_url: str :param tokens_file: path to tokens file :type tokens_file: str :param timeout: request timeout :type timeout: int :param verbose: print extra information :type verbose: bool :param update_tokens_auto: update tokens automatically :type update_tokens_auto: bool rzJTimeout must be greater than 0 and is recommended to be 5 seconds or more.z2.3.0)update_tokens_autoz*[Schwabdev] Client Initialization CompleteN) Exceptionversiontimeoutverbosertokensrstreamprint)selfapp_key app_secret callback_url tokens_filer r r r//var/www/html/StreammingVPS/schwabdev/client.py__init__s  zClient.__init__paramscCs(t|D] }||dur||=q|S)z Removes None (null) values :param params: params to remove None values from :type params: dict :return: params without None values :rtype: dict N)listkeys)rrkeyrrr_params_parser0szClient._params_parserN8601cCs|dus t|tr |S|dkr|dddddS|dkr)t|S|dkr5t|d S|d kr>|d S|S) ad Convert time to the correct format, passthrough if a string, preserve None if None for params parser :param dt: datetime.pyi object to convert :type dt: datetime.pyi | str | None :param form: what to convert input to :type form: str :return: converted time or passthrough :rtype: str | None Nr+rZepochepoch_msi YYYY-MM-DDz%Y-%m-%d) isinstancestr isoformatsplitint timestampstrftime)rdtformrrr _time_convert<s    zClient._time_convertlcCs&|durdSt|turd|S|S)z Convert python list to string or passthough if already a string i.e ["a", "b"] -> "a,b" :param l: list to convert :type l: list | str | None :return: converted string or passthrough :rtype: str | None N,)typerjoin)rr/rrr _format_listSs   zClient._format_listzhttps://api.schwabapi.comreturncC(tj|jddd|jji|jdS)a Account numbers in plain text cannot be used outside of headers or request/response bodies. As the first step consumers must invoke this service to retrieve the list of plain text/encrypted value pairs, and use encrypted account values for all subsequent calls for any accountNumber request. :return: All linked account numbers and hashes :rtype: request.Response z"/trader/v1/accounts/accountNumbers AuthorizationBearer headersr requestsget _base_api_urlr access_tokenr rrrraccount_linkedhzClient.account_linkedfieldscC4tj|jddd|jji|d|i|jdS)a All the linked account information for the user logged in. The balances on these accounts are displayed by default however the positions on these accounts will be displayed based on the "positions" flag. :param fields: fields to return (options: "positions") :type fields: str | None :return: details for all linked accounts :rtype: request.Response /trader/v1/accounts/r6r7rBr9rr r;r<r=rr>rr )rrBrrraccount_details_allr  zClient.account_details_all accountHashcCs8tj|jd|dd|jji|d|i|jdS)a Specific account information with balances and positions. The balance information on these accounts is displayed by default but Positions will be returned based on the "positions" flag. :param accountHash: account hash from account_linked() :type accountHash: str :param fields: fields to return :type fields: str | None :return: details for one linked account :rtype: request.Response rDr6r7rBrErF)rrIrBrrraccount_detailss   zClient.account_detailsfromEnteredTime toEnteredTime maxResultsstatusc CsRtj|jd|ddd|jjd||||d||d|d|jdS) a All orders for a specific account. Orders retrieved can be filtered based on input parameters below. Maximum date range is 1 year. :param accountHash: account hash from account_linked() :type accountHash: str :param fromEnteredTime: from entered time :type fromEnteredTime: datetime.pyi | str :param toEnteredTime: to entered time :type toEnteredTime: datetime.pyi | str :param maxResults: maximum number of results :type maxResults: int| None :param status: status ("AWAITING_PARENT_ORDER"|"AWAITING_CONDITION"|"AWAITING_STOP_CONDITION"|"AWAITING_MANUAL_REVIEW"|"ACCEPTED"|"AWAITING_UR_OUT"|"PENDING_ACTIVATION"|"QUEUED"|"WORKING"|"REJECTED"|"PENDING_CANCEL"|"CANCELED"|"PENDING_REPLACE"|"REPLACED"|"FILLED"|"EXPIRED"|"NEW"|"AWAITING_RELEASE_TIME"|"PENDING_ACKNOWLEDGEMENT"|"PENDING_RECALL"|"UNKNOWN") :type status: str| None :return: orders for one linked account hash :rtype: request.Response rD/ordersapplication/jsonr7Acceptr6rrMrKrLrNrEr;r<r=rr>rr.r )rrIrKrLrMrNrrraccount_orderss  zClient.account_ordersordercCs4tj|jd|ddd|jjdd||jdS)ay Place an order for a specific account. :param accountHash: account hash from account_linked() :type accountHash: str :param order: order dictionary, examples in Schwab docs :type order: dict :return: order number in response header (if immediately filled then order number not returned) :rtype: request.Response rDrOrPr7rRr6z Content-Typer9jsonr )r;postr=rr>r )rrIrVrrr order_places zClient.order_placeorderIdcC2tj|jd|d|dd|jji|jdS)a( Get a specific order by its ID, for a specific account :param accountHash: account hash from account_linked() :type accountHash: str :param orderId: order id :type orderId: int | str :return: order details :rtype: request.Response rD/orders/r6r7r8r:rrIr\rrr order_details zClient.order_detailscCr])a) Cancel a specific order by its ID, for a specific account :param accountHash: account hash from account_linked() :type accountHash: str :param orderId: order id :type orderId: str|int :return: response code :rtype: request.Response rDr^r6r7r8)r;deleter=rr>r r_rrr order_cancelrazClient.order_cancelcCs8tj|jd|d|dd|jjdd||jdS)a Replace an existing order for an account. The existing order will be replaced by the new order. Once replaced, the old order will be canceled and a new order will be created. :param accountHash: account hash from account_linked() :type accountHash: str :param orderId: order id :type orderId: str|int :param order: order dictionary, examples in Schwab docs :type order: dict :return: response code :rtype: request.Response rDr^rPr7rWrX)r;putr=rr>r )rrIr\rVrrr order_replaces zClient.order_replacec CsLtj|jddd|jjd||||d||d|d|jdS)a+ Get all orders for all accounts :param fromEnteredTime: start date :type fromEnteredTime: datetime.pyi | str :param toEnteredTime: end date :type toEnteredTime: datetime.pyi | str :param maxResults: maximum number of results (set to None for default 3000) :type maxResults: int | None :param status: status ("AWAITING_PARENT_ORDER"|"AWAITING_CONDITION"|"AWAITING_STOP_CONDITION"|"AWAITING_MANUAL_REVIEW"|"ACCEPTED"|"AWAITING_UR_OUT"|"PENDING_ACTIVATION"|"QUEUED"|"WORKING"|"REJECTED"|"PENDING_CANCEL"|"CANCELED"|"PENDING_REPLACE"|"REPLACED"|"FILLED"|"EXPIRED"|"NEW"|"AWAITING_RELEASE_TIME"|"PENDING_ACKNOWLEDGEMENT"|"PENDING_RECALL"|"UNKNOWN") :type status: str | None :return: all orders :rtype: request.Response z/trader/v1/ordersrPr7rQrrSrErT)rrKrLrMrNrrraccount_orders_alls  zClient.account_orders_all startDateendDatetypessymbolc CsRtj|jd|ddd|jji||||d||d||d|jdS)a  All transactions for a specific account. Maximum number of transactions in response is 3000. Maximum date range is 1 year. :param accountHash: account hash number :type accountHash: str :param startDate: start date :type startDate: datetime.pyi | str :param endDate: end date :type endDate: datetime.pyi | str :param types: transaction type ("TRADE, RECEIVE_AND_DELIVER, DIVIDEND_OR_INTEREST, ACH_RECEIPT, ACH_DISBURSEMENT, CASH_RECEIPT, CASH_DISBURSEMENT, ELECTRONIC_FUND, WIRE_OUT, WIRE_IN, JOURNAL, MEMORANDUM, MARGIN_CALL, MONEY_MARKET, SMA_ADJUSTMENT") :type types: str :param symbol: symbol :return: list of transactions for a specific account :rtype: request.Response rDz /transactionsr6r7r) accountNumberrgrhrjrirErT)rrIrgrhrirjrrr transactionss zClient.transactions transactionIdcCs:tj|jd|d|dd|jji||d|jdS)aX Get specific transaction information for a specific account :param accountHash: account hash number :type accountHash: str :param transactionId: transaction id :type transactionId: str|int :return: transaction details of transaction id using accountHash :rtype: request.Response rDz/transactions/r6r7)rkrmrEr:)rrIrmrrrtransaction_detailss  zClient.transaction_detailscCr5)z Get user preference information for the logged in user. :return: User Preferences and Streaming Info :rtype: request.Response z/trader/v1/userPreferencer6r7r8r:r?rrr preferences%rAzClient.preferencessymbols indicativec Cs>tj|jddd|jji|||||d|jdS)a Get quotes for a list of tickers :param symbols: list of symbols strings (e.g. "AMD,INTC" or ["AMD", "INTC"]) :type symbols: [str] | str :param fields: string of fields to get ("all", "quote", "fundamental") :type fields: str | None :param indicative: whether to get indicative quotes (True/False) :type indicative: boolean | None :return: list of quotes :rtype: request.Response z/marketdata/v1/quotesr6r7)rprBrqrE)r;r<r=rr>rr3r )rrprBrqrrrquotes3s z Client.quotes symbol_idcCsBtj|jdtj|ddd|jji|d|i|j dS)aP Get quote for a single symbol :param symbol_id: ticker symbol :type symbol_id: str (e.g. "AAPL", "/ES", "USD/EUR") :param fields: string of fields to get ("all", "quote", "fundamental") :type fields: str | None :return: quote for a single symbol :rtype: request.Response z/marketdata/v1/z/quotesr6r7rBrE) r;r<r=urllibparse quote_plusrr>rr )rrsrBrrrquoteEs   z Client.quote contractType strikeCountincludeUnderlyingQuotestrategyintervalstrikerangefromDatetoDate volatilityunderlyingPrice interestRatedaysToExpirationexpMonth optionType entitlementc Cstj|jddd|jji|id|d|d|d|d|d |d |d |d || d d|| d d| d| d| d|d|d|d||jdS)a Get Option Chain including information on options contracts associated with each expiration for a ticker. :param symbol: ticker symbol :type symbol: str :param contractType: contract type ("CALL"|"PUT"|"ALL") :type contractType: str :param strikeCount: strike count :type strikeCount: int :param includeUnderlyingQuote: include underlying quote (True|False) :type includeUnderlyingQuote: boolean :param strategy: strategy ("SINGLE"|"ANALYTICAL"|"COVERED"|"VERTICAL"|"CALENDAR"|"STRANGLE"|"STRADDLE"|"BUTTERFLY"|"CONDOR"|"DIAGONAL"|"COLLAR"|"ROLL) :type strategy: str :param interval: Strike interval :type interval: str :param strike: Strike price :type strike: float :param range: range ("ITM"|"NTM"|"OTM"...) :type range: str :param fromDate: from date :type fromDate: datetime.pyi | str :param toDate: to date :type toDate: datetime.pyi | str :param volatility: volatility :type volatility: float :param underlyingPrice: underlying price :type underlyingPrice: float :param interestRate: interest rate :type interestRate: float :param daysToExpiration: days to expiration :type daysToExpiration: int :param expMonth: expiration month ("JAN"|"FEB"|"MAR"|"APR"|"MAY"|"JUN"|"JUL"|"AUG"|"SEP"|"OCT"|"NOV"|"DEC"|"ALL") :type expMonth: str :param optionType: option type ("CALL"|"PUT") :type optionType: str :param entitlement: entitlement ("PN"|"NP"|"PP") :type entitlement: str :return: list of option chains :rtype: request.Response z/marketdata/v1/chainsr6r7rjrxryrzr{r|r}r~rr$rrrrrrrrrErT)rrjrxryrzr{r|r}r~rrrrrrrrrrrr option_chainsTsF*  zClient.option_chainscCrC)z Get an option expiration chain for a ticker :param symbol: ticker symbol :type symbol: str :return: option expiration chain :rtype: request.Response z/marketdata/v1/expirationchainr6r7rjrErF)rrjrrroption_expiration_chainrHzClient.option_expiration_chain periodTypeperiod frequencyType frequencyneedExtendedHoursDataneedPreviousClosec CsTtj|jddd|jji||||||||d||d|| d |jdS)a Get price history for a ticker :param symbol: ticker symbol :type symbol: str :param periodType: period type ("day"|"month"|"year"|"ytd") :type periodType: str :param period: period :type period: int :param frequencyType: frequency type ("minute"|"daily"|"weekly"|"monthly") :type frequencyType: str :param frequency: frequency (1|5|10|15|30) :type frequency: int :param startDate: start date :type startDate: datetime.pyi | str :param endDate: end date :type endDate: datetime.pyi | str :param needExtendedHoursData: need extended hours data (True|False) :type needExtendedHoursData: boolean :param needPreviousClose: need previous close (True|False) :type needPreviousClose: boolean :return: dictionary of containing candle history :rtype: request.Response z/marketdata/v1/pricehistoryr6r7r#) rjrrrrrgrhrrrErT) rrjrrrrrgrhrrrrr price_historys   zClient.price_historysortcCs<tj|jd|dd|jjd|||d|jdS)a Get movers in a specific index and direction :param symbol: symbol ("$DJI"|"$COMPX"|"$SPX"|"NYSE"|"NASDAQ"|"OTCBB"|"INDEX_ALL"|"EQUITY_ALL"|"OPTION_ALL"|"OPTION_PUT"|"OPTION_CALL") :type symbol: str :param sort: sort ("VOLUME"|"TRADES"|"PERCENT_CHANGE_UP"|"PERCENT_CHANGE_DOWN") :type sort: str :param frequency: frequency (0|1|5|10|30|60) :type frequency: int :return: movers :rtype: request.Response z/marketdata/v1/movers/rPr7)acceptr6)rrrErF)rrjrrrrrmoverss  z Client.moversdatec Cs>tj|jddd|jji||||dd|jdS)aO Get Market Hours for dates in the future across different markets. :param symbols: list of market symbols ("equity", "option", "bond", "future", "forex") :type symbols: list :param date: date :type date: datetime.pyi | str :return: market hours :rtype: request.Response z/marketdata/v1/marketsr6r7r$)marketsrrErT)rrprrrr market_hourss  zClient.market_hours market_idc Cs@tj|jd|dd|jji|d||di|jdS)a< Get Market Hours for dates in the future for a single market. :param market_id: market id ("equity"|"option"|"bond"|"future"|"forex") :type market_id: str :param date: date :type date: datetime.pyi | str :return: market hours :rtype: request.Response z/marketdata/v1/markets/r6r7rr$rErT)rrrrrr market_hours  zClient.market_hour projectioncCs0tj|jddd|jji||d|jdS)aD Get instruments for a list of symbols :param symbol: symbol :type symbol: str :param projection: projection ("symbol-search"|"symbol-regex"|"desc-search"|"desc-regex"|"search"|"fundamental") :type projection: str :return: instruments :rtype: request.Response z/marketdata/v1/instrumentsr6r7)rjrrEr:)rrjrrrr instrumentss  zClient.instrumentscusip_idcCs,tj|jd|dd|jji|jdS)z Get instrument for a single cusip :param cusip_id: cusip id :type cusip_id: str|int :return: instrument :rtype: request.Response z/marketdata/v1/instruments/r6r7r8r:)rrrrrinstrument_cusipszClient.instrument_cusip)rrrFT)Nr)N)NN)NF)NNNNNNNNNNNNNNNN)NNNNNNNN)'__name__ __module__ __qualname__rdictrr.rr&r3r=r;Responser@rGrJdatetimer)rUr[r`rcrerfrlrnroboolrrrwanyrrrrrrrrrrrrrs    2 .2 & 5 " #$ r) __doc__rr; urllib.parsertrrrrrrrrrs