o 譐^@sldZddlmZddlmZddlmZddlmZddlmZmZddl m Z ddl Z Gd d d e Z dS) z#Helper class for InfluxDB for v0.8.)absolute_import)division)print_function)unicode_literals) namedtuple defaultdict)warnNcsReZdZdZdZfddZddZeddd Zed d Z ed d Z Z S) SeriesHelperaDefine the SeriesHelper object for InfluxDB v0.8. Subclassing this helper eases writing data points in bulk. All data points are immutable, ensuring they do not get overwritten. Each subclass can write to its own database. The time series names can also be based on one or more defined fields. Annotated example:: class MySeriesHelper(SeriesHelper): class Meta: # Meta class stores time series helper configuration. series_name = 'events.stats.{server_name}' # Series name must be a string, curly brackets for dynamic use. fields = ['time', 'server_name'] # Defines all the fields in this time series. ### Following attributes are optional. ### client = TestSeriesHelper.client # Client should be an instance of InfluxDBClient. :warning: Only used if autocommit is True. bulk_size = 5 # Defines the number of data points to write simultaneously. # Only applicable if autocommit is True. autocommit = True # If True and no bulk_size, then will set bulk_size to 1. Fc sF|jsd|_zt|d}Wntytd|jwdD]}z t|d|t||Wqty>td||jwt|dd|_t|d d |_|jr[|js[td |jzt|d |_|jd kru|jrut d|jd |_Wn tyd|_Yn w|jst d|jt t |_ t |j|j|_tt||S)zInitialize class attributes for subsequent constructor calls. :note: *args and **kwargs are not explicitly used in this function, but needed for Python 2 compatibility. TZMetazMissing Meta class in {0}.) series_nameZfields_zMissing {0} in {1} Meta class.Z autocommitFclientNz8In {0}, autocommit is set to True, but no client is set.Z bulk_sizezusz(SeriesHelper.__init__..Nr$)r#sortedrkeys NameErrorrrZ _series_nameappendrrsumvaluesrcommit)selfkwrr$r$r%__init__bs * zSeriesHelper.__init__NcCs$|s|j}||}||S)aCommit everything from datapoints via the client. :param client: InfluxDBClient instance for writing points to InfluxDB. :attention: any provided client will supersede the class client. :return: result of client.write_points. )rZ write_points _json_body__reset_)rr Zrtnr$r$r%r0ys zSeriesHelper.commitcsNg}js tjD]\}}||jfdd|Ddq|S)z[Return JSON body of the datapoints. :return: JSON body of the datapoints. cs g|] fddjDqS)csg|]}t|qSr$)r)r(kZpointr$r% sz7SeriesHelper._json_body_...)r)r(rr7r%r8sz,SeriesHelper._json_body_..)namecolumnsZpoints)rr5sixZ iteritemsrr-r)rZjsonr datar$r9r%r4s  zSeriesHelper._json_body_cCstt|_dS)zReset data storage.N)rrrr9r$r$r%r5szSeriesHelper._reset_r&) r __module__ __qualname____doc__rrr3 classmethodr0r4r5 __classcell__r$r$r"r%r s 4  r )r@Z __future__rrrr collectionsrrwarningsrr<objectr r$r$r$r%s