o /.]d@sbdZddlmZddlmZddlmZddlmZddlZddlmZe Z Gdd d e Z dS) z Module to prepare the resultset.)absolute_import)division)print_function)unicode_literalsN)InfluxDBClientErrorc@seZdZdZd!ddZeddZejddZedd Zd d Z d"d dZ ddZ ddZ e ddZddZddZddZddZddZe dd Zd S)# ResultSetz0A wrapper around a single InfluxDB query result.TcCs:||_|jdd|_|jdur|durt|jdSdS)zInitialize the ResultSet.errorNT)_rawget_errorrr)selfseriesZ raise_errorsr4/usr/lib/python3/dist-packages/influxdb/resultset.py__init__s  zResultSet.__init__cC|jS)zRaw JSON from InfluxDB.r r rrrrawz ResultSet.rawcCs ||_dS)Nr)r valuerrrr s cCr)zError returned by InfluxDB.)r rrrrr$rzResultSet.errorcCs~tdtt|tr+t|dkrtd|d}|d}t|ts*|dur*tdnt|tr5d}|}n|}d}|||S)aRetrieve the series name or specific set based on key. :param key: Either a series name, or a tags_dict, or a 2-tuple(series_name, tags_dict). If the series name is None (or not given) then any serie matching the eventual given tags will be given its points one after the other. To get the points of every series in this resultset then you have to provide None as key. :return: A generator yielding `Point`s matching the given key. NB: The order in which the points are yielded is actually undefined but it might change.. zQResultSet's ``__getitem__`` method will be deprecated. Use``get_points`` instead.zonly 2-tuples allowedrNztags should be a dict) warningswarnDeprecationWarning isinstancetuplelen TypeErrordict get_points)r keynametagsrrr __getitem__)s$    zResultSet.__getitem__Nccst|ttdtdfstd|D]G}|d|dd}|dur7|dur6||D]}|Vq0q|d|fvr^|di}||D]}|dusZ|||sZ|||r]|VqHqdS)aReturn a generator for all the points that match the given filters. :param measurement: The measurement name :type measurement: str :param tags: Tags to look for :type tags: dict :return: Points generator Nz"measurement must be an str or None measurementr#resultsr$) rbytestypedecoder _get_seriesr _get_points_for_series _tag_matches)r r'r$r Z series_nameitemZ series_tagsrrrr!Ps2       zResultSet.get_pointscCs>g}|D]}|d|dt|dfqdd|S)z#Representation of ResultSet object.z'%s': %srrzResultSet({%s})z, )itemsappendlistjoin)r r0r/rrr__repr__us  zResultSet.__repr__ccs$|D] }t||VqdS)z*Yield one dict instance per series result.N)keysr2r%)r r"rrr__iter__~s zResultSet.__iter__cCs0|D]\}}||t}||krdSqdS)z0Check if all key/values in filter match in tags.FT)r0r _sentinel)r$filterZtag_nameZ tag_valueZseries_tag_valuerrrr.s  zResultSet._tag_matchescCs|jdgS)zReturn all series.r )rr rrrrr,szResultSet._get_seriescCs t|S)z,Return the len of the keys in the ResultSet.)rr5rrrr__len__s zResultSet.__len__c Cs<g}|D]}||d|dd|ddfq|S)zvReturn the list of keys in the ResultSet. :return: List of keys. Keys are tuples (series_name, tags) r'r#r(r$N)r,r1r )r r5r rrrr5s   zResultSet.keyscCsJg}|D]}|d|dd|ddf}||||fq|S)zgReturn the set of items from the ResultSet. :return: List of tuples, (key, generator) r'r#r(r$N)r,r r1r-)r r0r Z series_keyrrrr0s    zResultSet.itemsccs*|dgD] }||d|VqdS)zReturn generator of dict from columns and values of a series. :param series: One series :return: Generator of dicts valuescolumnsN)r point_from_cols_vals)r r pointrrrr-sz ResultSet._get_points_for_seriescCs&i}t|D] \}}||||<q|S)zCreate a dict from columns and values lists. :param cols: List of columns :param vals: List of values :return: Dict where keys are columns. ) enumerate)ZcolsZvalsr=Z col_indexZcol_namerrrr<szResultSet.point_from_cols_vals)T)NN)__name__ __module__ __qualname____doc__rpropertyrsetterrr%r!r4r6 staticmethodr.r,r9r5r0r-r<rrrrrs,     '%    r) rBZ __future__rrrrrZinfluxdb.exceptionsrobjectr7rrrrrs