iPdZddlmZddlZddlmZmZddlmZerddl m Z ddl Z ddl mZdd lmZdd lmZeGd d Zy) zCSymbol-Exchange-Interval Set (Seis) for tracking live data streams.) annotationsN) dataclassfield) TYPE_CHECKING)CallableTvDatafeedLive)Consumer)IntervalcleZdZUdZded<ded<ded<eddd Zd ed <eedd Zd ed<eddd Z ded<ddZ ddZ e d dZ e jd!dZ e jd"dZ d# d$dZ d# d%dZd&dZd&dZd'dZ d( d)dZd#d*dZd+dZy),SeisaSymbol, exchange, and interval data set for live streaming. Holds a unique combination of symbol, exchange, and interval values, along with associated consumers for this data stream. Args: symbol: Ticker symbol string exchange: Exchange where symbol is listed interval: Chart time interval Example: >>> seis = Seis('ETHUSDT', 'BINANCE', Interval.in_1_hour) >>> seis.new_consumer(my_callback) strsymbolexchanger intervalNF)defaultreprinitTvDatafeedLive | None _tvdatafeed)default_factoryrrlist[Consumer] _consumerszdatetime.datetime | None_updatedct|tstS|j|jk(xr4|j|jk(xr|j |j k(S)zCompare two Seis instances for equality. Instances are equal if symbol, exchange, and interval match. Args: other: Object to compare with Returns: True if equal, False otherwise ) isinstancerNotImplementedrrr)selfothers H/home/work/apex_v16/venv/lib/python3.12/site-packages/tvDatafeed/seis.py__eq__z Seis.__eq__+sU%&! ! KK5<< ' , MMU^^ + , MMU^^ + cjd|jd|jd|jjdS)zReturn string representation.zsymbol='z ', exchange='z ', interval='')rrrnamers r!__str__z Seis.__str__?s4$++mDMM?-PTP]P]PbPbOccdeer#c|jS)z'Get associated TvDatafeedLive instance.rr's r! tvdatafeedzSeis.tvdatafeedCsr#czddlm}|j tdt ||s t d||_y)zSet TvDatafeedLive instance. Args: value: TvDatafeedLive instance Raises: AttributeError: If already set (cannot overwrite) TypeError: If value is not TvDatafeedLive instance rr Nz-Cannot overwrite tvdatafeed - delete it firstz%Value must be TvDatafeedLive instance)datafeedr rAttributeErrorr TypeError)rvalueTvDatafeedLiveClasss r!r+zSeis.tvdatafeedHs? D    ' !PQ Q%!45CD D r#cd|_y)z Delete TvDatafeedLive reference.Nr*r's r!r+zSeis.tvdatafeed^s  r#cj|j td|jj|||S)aCreate new consumer and add to this Seis. Args: callback: Function(seis, data) to call with new data timeout: Maximum wait time in seconds (-1 for blocking) Returns: Consumer instance, or False if timeout Raises: RuntimeError: If no TvDatafeedLive reference set Example: >>> def my_callback(seis, data): ... print(f"New data: {data}") >>> consumer = seis.new_consumer(my_callback) 4No TvDatafeedLive instance associated with this Seis)r RuntimeError new_consumer)rcallbacktimeouts r!r6zSeis.new_consumercs7,    #UV V,,T8WEEr#ch|j td|jj||S)a>Remove consumer from this Seis. Args: consumer: Consumer instance to remove timeout: Maximum wait time in seconds (-1 for blocking) Returns: True if successful, False if timeout Raises: RuntimeError: If no TvDatafeedLive reference set r4)rr5 del_consumer)rconsumerr8s r!r:zSeis.del_consumer~s5"    #UV V,,Xw??r#c:|jj|y)zAdd consumer to internal list. Internal method used by TvDatafeedLive. Not for direct use. Args: consumer: Consumer instance to add N)rappendrr;s r! add_consumerzSeis.add_consumers x(r#cl||jvr td|jj|y)zRemove consumer from internal list. Internal method used by TvDatafeedLive. Not for direct use. Args: consumer: Consumer instance to remove Raises: ValueError: If consumer not in list z&Consumer not registered with this SeisN)r ValueErrorremover>s r! pop_consumerzSeis.pop_consumers. 4?? *EF F x(r#cl|jjd}|j|k7r||_yy)zCheck if data contains a new bar (different datetime). Args: data: DataFrame with datetime index Returns: True if new data, False if duplicate rTF)index to_pydatetimer)rdata current_dts r! is_new_datazSeis.is_new_datas4ZZ--/2 ==J &&DMr#c|j td|jj|j|j|j ||S)aGet historical data for this Seis. Args: n_bars: Number of bars to retrieve timeout: Maximum wait time in seconds (-1 for blocking) Returns: DataFrame with OHLCV data, or False if timeout Raises: RuntimeError: If no TvDatafeedLive reference set Example: >>> data = seis.get_hist(n_bars=100) r4)rrrn_barsr8)rr5get_histrrr)rrKr8s r!rLz Seis.get_histsU(    #UV V((;;]]]] )  r#ch|j td|jj||S)aRemove this Seis from TvDatafeedLive. Args: timeout: Maximum wait time in seconds (-1 for blocking) Returns: True if successful, False if timeout Raises: RuntimeError: If no TvDatafeedLive reference set r4)rr5del_seis)rr8s r!rNz Seis.del_seiss5    #UV V((w77r#c|jS)zjGet list of all consumers for this Seis. Returns: List of Consumer instances )rr's r! get_consumerszSeis.get_consumerss r#)r objectreturnbool)rRr)rRr)r0r rRNone)rRrT))r7z$Callable[[Seis, pd.DataFrame], None]r8intrRzConsumer | bool)r;r r8rVrRrS)r;r rRrT)rGz pd.DataFramerRrS) rU)rKrVr8rVrRzpd.DataFrame | bool)r8rVrRrS)rRr)__name__ __module__ __qualname____doc____annotations__rrlistrrr"r(propertyr+setterdeleterr6r:r?rCrIrLrNrPr#r!rrs?  KM*/t%e)TK&T!&t%e!TJT).t%e)TH&T (f  !!*  F6FF  F<@@@  @,) )&      >8"r#r)r[ __future__rdatetime dataclassesrrtypingrcollections.abcrpandaspdr-r r;r mainr rrar#r!rjsBI"( ((" bb br#