T>qj> dZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z mZmZmZmZmZmZddlZ ddlmZej4Z ejdddej8j:Zd Zej@jBZ"eejejfZ#d9d eee$fd Z%d e&fd Z'd e(fdZ)d e&fdZ*d e+fdZ,d e$fdZ-dZ.dZ/d:dZ0dZ1ejdfdZ3ejdfdZ4de5d e6fdZ7de5d e$fdZ8GddZ9ddded ee5fd!Z:d"e#d ejfd#Z;d"e#d$efd%Zd,e#d e6fd-Z?d(e#d)e#d*e5d e ejfd.Z@d,e#d e6fd/ZAd0ZBd1ZCd2ZDdzdf../s*M1+;A+>*Msc34K|]}|jywr)__dict__rs rrzdf..2s*D!1::*Ds_fields)axis)pandasobjectsrlistr DataFrame from_recordsrnamecolumns isinstancetuplegetattrdrop) objsrpdrobjdffieldrlabelexcludes rr-r-"s& Dz1g  ***M*MMB28+>%**>BJ ] +***Dt*DDB**40B c5 !c9d3G$  &(@UE,?5@@ WWW1W % I?AsD, D6Dreturnct|std|dt|Dcic]#}|jt ||j%c}Scc}w)zm Return dataclass values as ``dict``. This is a non-recursive variant of ``dataclasses.asdict``. Object  is not a dataclass)r TypeErrorrr$r()r,r.s rdataclassAsDictr6BsM  '#&9:;;>DSk JUEJJUZZ0 0 JJ Js(AcrtstddtfdtDS)zo Return dataclass values as ``tuple``. This is a non-recursive variant of ``dataclasses.astuple``. r3r4c3JK|]}t|jywr)r(r$)rr.r,s rrz#dataclassAsTuple..SsCeejj)Cs #)rr5r'r)r,s`rrrLs7  '#&9:;; Cvc{C CCc\t|std|dt|Dcgc]}t||j}}t t||Dcic];\}}||j k7r'||k(r"t|tr|gk(s |j|=c}}Scc}wcc}}w)z| For a ``dataclass`` instance get the fields that are different from the default values and return as ``dict``. r3r4) rr5rr(r$zipdefaultr&r!)r,r.valuesvalues rdataclassNonDefaultsr?Vs  '#&9:;;4:3K @5gc5::& @F @.1&+v.F ;*eU EMM ! UNE4(Ub[  E ;;A ;s B#AB(ct|std|d|D]&}|jjt |(|jjdi||S)z Update fields of the given ``dataclass`` object from zero or more ``dataclass`` source objects and/or from keyword arguments. r3r4)rr5rupdater6)r,srcObjskwargssrcObjs rdataclassUpdaterFesc  '#&9:;;5 OF345CLL!&! Jr9ct|}|jj}djd|j D}|d|dS)z Provide a culled representation of the given ``dataclass`` instance, showing only the fields with a non-default value. z, c30K|]\}}|d|yw)=NrA)rkvs rrz dataclassRepr..ys>1!AaU|>s())r? __class__ __qualname__joinitems)r,attrsclsNamerDs r dataclassReprrTrsJ ! %Emm((G YY> > >FYaxq !!r9ct|}|j}t|dk7s |dtk7ryt |dd}t |tsyt d|DS)z0From https://stackoverflow.com/a/2166841/6067848r rFrNc3@K|]}t|tk(ywr)typestr)rns rrz'isnamedtupleinstance..s)!tAw#~)s)rW __bases__lenr'r(r&all)xtbfs risnamedtupleinstancera}sX QA A 1v{adem9d#A a  )q) ))r9c t|ttttt fr|St|t jt jfr|jSt|tr-|jDcic]\}}|t|c}}St|r.|jDcic]}|tt||c}St|t t"t$fr|Dcgc] }t|c}St'|r*|j(j*tt-|iSt |Scc}}wcc}wcc}w)zo Convert object to a tree of lists, dicts and simple values. The result can be serialized to JSON. )r&boolintfloatrXbytesdtdatetime isoformatdictrQtreerarr(r!r'setrrNrOr?)r,rJrKr`is rrlrls #c5#u56 C"''277+ ,}} C '*yy{3tq!47 33 c "25++>Q4Q((>> C$s+ ,!$%AQ%% c  **D1Ec1J,KLL3x4>%sE 8E7Ec ddl}ddlm}ddlm}ddlm}t||jr(|gdjDcgc] }t|} }n|rKt|ddr<|D cgc]0} | j| j| j| jf2} } n;|D cgc]0} | j | j| j| jf2} } |j#\} } | j%|| j'd| j)dd t+| D]\} \}}}}||k\r|}||}}n|}||}}|| | f||f|d }| j-||| | f||f|d }| j-||| d z |fd ||z ||dd}| j/|| j1| Scc}wcc} wcc} w)z{ Create candlestick plot for the given bars. The bars can be given as a DataFrame or as a list of bar objects. rN)Line2D) Rectangle)openhighlowcloseopen_T r )xdataydatacolor linewidthg333333?g333333?g?)xywidthheight edgecolor facecoloralpha antialiased)rmatplotlib.pyplotpyplotmatplotlib.linesrpmatplotlib.patchesrqr&r"r=r'hasattrrvrsrtrurrsubplots set_titlegridset_size_inches enumerateadd_line add_patchautoscale_view)barstitleupColor downColorr+pltrprqrKohlcTupsr_figaxrYrvrsrtrur{bodyHibodyLolinerects rbarplotrs #',$ %"#CDKKME!HMM '$q'7+?CD!QWWaffaeeQWW5DD>BCQVVQVVQUUAGG4CCllnGCLLGGDMA(1((;$ $E4e E>E"EFFE"EFFa&-  Da&.  DC F?  T9< JUMECsG *5G&5Gchtjtjtjy)zAllow Control-C to end program.N)signalSIGINTSIG_DFLrAr9r allowCtrlCrs MM&--0r9cHtj}|jr%tjdj|n|j|tjd}tj |}|j ||j|y)z1Create a log handler that logs to the given file. ib_insync.%(asctime)s %(name)s %(levelname)s %(message)sN)logging getLoggerhandlerssetLevel Formatter FileHandler setFormatter addHandler)pathlevellogger formatterhandlers r logToFilers{    F +&//6!!8:I!!$'G # gr9ctj}|jDcgc];}t|tjur|j t jur|=}}|r%tjdj|y|j|tjd}tj}|j||j|ycc}w)z.Create a log handler that logs to the console.rrN) rrrrW StreamHandlerstreamsysstderrrrrr)rrh stdHandlersrrs r logToConsolers    F??H 7g++ +CJJ0F HKH +&//6 %% <> '')Y''"HsAC)r]c ||k7S)zNot a number test.rA)r]s risNanrs 6Mr9rYcd}|dkr| }|dz }t|tur|dkrt|dz}|S|dkrd}|S|dksJttjtj |}t |d \}}td D]2}d jd |z }||d d |zzzz}|dk7rn |dz }d}4|dzz }|dk7r |d|dzz }|S)zBFormat the integer or float n to 3 significant digits + SI prefix.r-i g^ 9^;z0.00 g ҉Ez%.{}frw1000r zyzafpnum kMGTPEZY) rWrdrXmathfloorlog10divmodrangeformat)rYslogrnj_trytemplvals rformatSIrs A1u B S Aw#~!d( FSL" H! U  H7{{$**TZZ]+,c1~1!H DNN1q5)E1rb1f~-.Cf} FAA   S3Y 6 $QU+ +A Hr9c$eZdZdZddZdZdZy)timeitzContext manager for timing.c||_yr)r)selfrs r__init__ztimeit.__init__ s  r9c6tj|_yr)rit0)rs r __enter__ztimeit.__enter__#s))+r9ct|jdzttj|jz zdzy)Nz took r)printrrrir)r_argss r__exit__ztimeit.__exit__&s1 djj8#htyy{TWW/D&EEKLr9N)Run)__name__ __module__rO__doc__rrrrAr9rrrs%Mr9r)timeout awaitablesrc` t}|s|jry|jd}tjdk\rt j |}nt jj }|r6t j|}|j |j||S|St|dk(r|d}nt j|}|rt j||}t j| fd}tj!| |j } tj%||S#t j$rY|SwxYw#t j$r}tj#xs|d}~wwxYw#tj%|wxYw)a] By default run the event loop forever. When awaitables (like Tasks, Futures or coroutines) are given then run the event loop until each has completed and return their results. An optional timeout (in seconds) can be given that will raise asyncio.TimeoutError if the awaitables are not ready within the timeout period. N)rr rc&jyr)cancel)_tasks ronErrorzrun..onErrorPs KKMr9)getLoop is_running run_foreverr version_infoasyncio all_tasksTaskgatherrrun_until_completeCancelledErrorr[wait_for ensure_futureglobalErrorEventconnectr> disconnect) rrloopresultrr`futurerers @rrunr*s{ 9D  ??      v %))$/I ..0I  *A HHJ ''*. M6M' z?a ]F^^Z0F %%fg6F$$V,    ) 1,,T2F  ' ' 0 M-)) * M- "%% 0"((*/a / 0  ' ' 0s6E E# E E #F6FFFF-rict|tjr>tjj tj j |}|S|}|Sr)r&rgridatetimecombinerhtoday)rir^s r _fillDater^sC$ KK    6 H  Hr9callbackct|}tjj|j}||z j }t }|j||g|S)aY Schedule the callback to be run at the given time with the given arguments. This will return the Event Handle. Args: time: Time to run callback. If given as :py:class:`datetime.time` then use today as date. callback: Callable scheduled to run. args: Arguments for to call callback with. )rrgrnowr total_secondsr call_later)rirargsr^rdelayrs rscheduler gsT $A ++//!(( #C W # # %E 9D 4??5( 2T 22r9secsc@ttj|y)z Wait for the given amount of seconds while everything still keeps processing in the background. Never use time.sleep(). Args: secs (float): Time in seconds to wait. T)rrsleep)rs rrrzs d r9startendstepc#nK|dkDsJtj|}t|}|jrtjj nd}tj j|}||kr ||z }||kr |t|kr$t||||z }|t|kr#yyw)a Iterator that waits periodically until certain time points are reached while yielding those time points. Args: start: Start time, can be specified as datetime.datetime, or as datetime.time in which case today is used as the date end: End time, can be specified as datetime.datetime, or as datetime.time in which case today is used as the date step (float): The number of seconds of each period rsecondsN) rg timedeltarrtimezoneutcrr waitUntilrrrdeltar^tzrs r timeRangers !8O8 LL &E%AHH$B ++//" C c' U  c' y~ !  U  y~ sA?B50B53B5r^ctjj|j}t ||z j }t tj|y)z Wait until the given time t is reached. Args: t: The time t can be specified as datetime.datetime, or as datetime.time in which case today is used as the date. T) rgrrrrr rrrr^rrs rrrsE ++//!(( #C aL3  - - /D d r9cK|dkDsJtj|}t|}|jrtjj nd}tj j|}||kr ||z }||kr |t|kr-t|d{|||z }|t|kr,yy7w)z#Async version of :meth:`timeRange`.rrN) rgrrrrrrrwaitUntilAsyncrs rtimeRangeAsyncr#s !8O8 LL &E%AHH$B ++//" C c' U  c' y~ Q U  y~ sA?CCB>C<CcKtjj|j}t ||z j }t j|d{y7w)z#Async version of :meth:`waitUntil`.NT)rgrrrrr rrr s rr"r"sM ++//!(( #C aL3  - - /D --  sAA( A&!A(c,ddl}|jy)z*Patch asyncio to allow nested event loops.rN) nest_asyncioapply)r&s r patchAsyncior(sr9cFtjjS)z2Get the asyncio event loop for the current thread.)rget_event_loop_policyget_event_looprAr9rrrs  ( ( * 9 9 ;;r9cty)z4Use nested asyncio event loop for Jupyter notebooks.N)r(rAr9r startLoopr-sNr9qtLibperiodcfddvrtdddlm}|dz|dz}|jj xs|jt j atgy) z Run combined Qt5/asyncio event loop. Args: qtLib: Name of Qt library to use: * PyQt5 * PyQt6 * PySide2 * PySide6 period: Period in seconds to poll Qt. cjsXj}j}|jj |j j ||fj\}}|jddk(r|jn|j|jj ||ftjy)NrPyQt6)r  QEventLoopQTimerrrquitappendpoprexecexec_stopqApp processEvents)qlooptimerrr/qcr.qt_stepstacks rr@zuseQt..qt_steps (MMOEIIKE MM ! !%** - LL% (yy{ u A( ekkm  eU^$ r9)PyQt5r2PySide2PySide6zUnknown Qt library: r) import_modulez.QtCorez .QtWidgetsN) RuntimeError importlibrE QApplicationinstancerargvr;r)r.r/rEqwrr?r@rAs`` @@@@ruseQtrLs   <<1%9::' uy( )B u|+ ,B OO $ $ & )sxx( 9DE Ir9c|sd}|St|tjr=|jtjj }|j d}|St|tjrstj|j|j|jdddjtjj }|j d}|S|}|S)z/Format date or datetime to string that IB uses.r)rz%Y%m%d %H:%M:%S UTC;) r&rgr astimezonerrstrftimerhyearmonthday)r^rs rformatIBDatetimerUs  H Ar{{ # LLBKKOOL , JJ, - H Arww  KK FFAGGQUUBB 00: bkkoo 0N JJ, - H  Hr9rct|dk(rCt|dd}t|dd}t|dd}tj|||}|S|j rCtj j t|tjj}|S|jddk\rZd|vrV|jdd\}}}tj j||zd}|jt| }|S|jdd jd d d d }tj j|d}|S)z7Parse string in IB date or datetime format to datetime.rrrxrrz z%Y%m%d%H:%M:%SrrrN)r[rdrgrhisdigitr fromtimestamprrcountsplitstrptimereplacer ) rymdr^s0s1s2sss rparseIBDatetimerfs, 1v{ !AK !AK !AK GGAq!  H  KK % %c!fbkkoo > H  t1}WWS!_ B KK b*: ; IIXb\I * HYYsB  ' 'R 0" 5 KK %5 6 Hr9r)rbluered)g{Gz?)rBg{Gz?)Irrrrgrrrrri dataclassesrrtypingrrrrr r r eventkitevzoneinfor ImportErrorbackports.zoneinfoEventrrrEPOCH UNSET_INTEGER float_infomax UNSET_DOUBLETime_trXr-rkr6r'rr?objectrFrTrarlrrINFOrrrercrrrrrr rrrr#r"r(rr-rLrhrUrfrAr9rrys ,III,! 288:  D!Qr{{7 ~~!! rww # $Xd3i(@KDKDUD ; ;  "#" **5p1 ",, ||#(Ut   # 6 M M<@1Y1%1h F r{{ 363X3&   V& BKK 4  D  &bkk2$Ft <  %%%P bggr{{C=> 3 " s uRWWbkk%9: ,++,sG-- G<;G<