o -7_@sxdZddlmZddlmZddlmZddlmZddlmZmZddl m Z ddl m Z dd l Z Gd d d e Zd S) zHelper class for InfluxDB.)absolute_import)division)print_function)unicode_literals) namedtuple defaultdict)datetime)warnNcs^eZdZdZdZfddZddZeddd Zed d Z ed d Z e ddZ Z S) SeriesHelperaSubclass 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. The field "time" can be specified when creating a point, and may be any of the time types supported by the client (i.e. str, datetime, int). If the time is not specified, the current system time (utc) will be used. 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. retention_policy = 'your_retention_policy' # Specify the retention policy for the data points time_precision = "h"|"m"|s"|"ms"|"u"|"ns" # Default is ns (nanoseconds) # Setting time precision while writing point # You should also make sure time is set in the given precision Fc s|jsd|_zt|d}Wntytd|jwdD]}z t|d|t||Wqty>td||jwt|dd|_t|d d |_gd }|j|vrbtd |jd |t|dd |_ t|dd |_ |jr~|j s~td|jzt|d|_ |j dkr|jrt d|jd|_ Wn tyd|_ Yn w|jst d|jt t|_d|jvr|jdt|jdg|j|j|_dt|j|jj_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_namefieldstags_zMissing {0} in {1} Meta class.Z autocommitFtime_precisionN)hmsmsunsNz8In {}, time_precision is set, but invalid use any of {}.,retention_policyclientz8In {0}, autocommit is set to True, but no client is set.Z bulk_sizezsz(SeriesHelper.__init__..Nr9)r8pop_current_timestampsetr-r+keys NameErrorr r*Z _series_nameappendr.r#sumvaluesr(commit)selfkwr3Z timestampr r r@r9r9r:__init__ys0     zSeriesHelper.__init__NcCs.|s|j}|j||j|jd}||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. )rr)r'Z write_points _json_body_r$r&_reset_)r3rZrtnr9r9r:rEszSeriesHelper.commitc Csg}|js |t|jD]<\}}|D]5}|iit|dd}|jD]}t||}|dur5||d|<q$|jD] }t|||d|<q9||qq|S)zcReturn the JSON body of given datapoints. :return: JSON body of these datapoints. r)Z measurementr r rNr r ) rrJsixZ iteritemsr*rr+r-rB) r3Zjsonr dataZpointZ json_pointZfieldvaluetagr9r9r:rIs(     zSeriesHelper._json_body_cCstt|_dS)zReset data storage.N)rr)r*)r3r9r9r:rJszSeriesHelper._reset_cCstSr)rZutcnowr9r9r9r:r>szSeriesHelper._current_timestampr)r! __module__ __qualname____doc__rr0rH classmethodrErIrJ staticmethodr> __classcell__r9r9r7r:r s$ B   r )rQZ __future__rrrr collectionsrrrwarningsr rKobjectr r9r9r9r:s