a

��bvn�@sbdd
lm
Z

dZdZdZdZdZdZdZd	Z	d
Z
dZdZd
Z
dZdZdZdZdZGdd�d�ZdS)�)
�	DataError�TS.ADDzTS.ALTERz
TS.CREATERULEz	TS.CREATEz	TS.DECRBYz
TS.DELETERULEzTS.DELzTS.GETz	TS.INCRBYzTS.INFOzTS.MADDzTS.MGETz	TS.MRANGEzTS.MREVRANGEz
TS.QUERYINDEXzTS.RANGEzTS.REVRANGEc@sP
eZ
dZd
Zdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	dd�Z
dd�Zdd�Zdd�Z
dEdd�
ZdFdd�
Zdd�ZdGdd �
ZdHd!d"�
Zd#d$�ZdId%d&�
Zd'd(�Zd)d*�Zed+d,��
ZedJd-d.�
�
Zed/d0��
Zed1d2��
Zed3d4��
Zed5d6��
Zed7d8��
Zed9d:��
Zed;d<��
Z ed=d>��
Z!ed?d@��
Z"edAdB��
Z#edCdD��
Z$dS)K�TimeSeriesCommandszRedisTimeSeries Commands.c	Ks�|�d
d�}|�dd�}|�di�}|�dd�}|�dd�}|
g
}|�
||�
|�||�
|�||�
|�|t|�
|�||�
|jtg
|�
R�S)af
        Create a new time-series.

        Args:

        key:
            time-series key
        retention_msecs:
            Maximum age for samples compared to last event time (in milliseconds).
            If None or 0 is passed then  the series is not trimmed at all.
        uncompressed:
            Since RedisTimeSeries v1.2, both timestamps and values are
            compressed by default.
            Adding this flag will keep data in an uncompressed form.
            Compression not only saves
            memory but usually improve performance due to lower number
            of memory accesses.
        labels:
            Set of label-value pairs that represent metadata labels of the key.
        chunk_size:
            Each time-serie uses chunks of memory of fixed size for
            time series samples.
            You can alter the default TSDB chunk size by passing the
            chunk_size argument (in Bytes).
        duplicate_policy:
            Since RedisTimeSeries v1.4 you can specify the duplicate sample policy
            ( Configure what to do on duplicate sample. )
            Can be one of:
            - 'block': an error will occur for any out of order sample.
            - 'first': ignore the new value.
            - 'last': override with latest value.
            - 'min': only override if the value is lower than the existing value.
            - 'max': only override if the value is higher than the existing value.
            When this is not set, the server-wide default will be used.

        For more information: https://oss.redis.com/redistimeseries/commands/#tscreate
        �retention_msecsN�uncompressedF�labels�
chunk_size�duplicate_policy)�get�_append_retention�_append_uncompressed�_append_chunk_size�_append_duplicate_policy�
CREATE_CMD�_append_labels�execute_command)	�self�key�kwargsrrrrr	�params�r�F/usr/lib/python3.9/site-packages/redis/commands/timeseries/commands.py�creates&









zTimeSeriesCommands.createcKsb|�d
d�}|�di�}|�dd�}|
g
}|�
||�
|�|t|�
|�||�
|jtg
|�
R�S)z�
        Update the retention, labels of an existing key.
        For more information see

        The parameters are the same as TS.CREATE.

        For more information: https://oss.redis.com/redistimeseries/commands/#tsalter
        rNrr	)r
rr�	ALTER_CMDrr)rrrrrr	rrrr�alterMs	





zTimeSeriesCommands.altercKs�|�d
d�}|�dd�}|�di�}|�dd�}|�dd�}	|
||g}
|�
|
|�
|�|
|�
|�|
|�
|�|
t|	�
|�|
|�
|jtg
|
�
R�S)a!
        Append (or create and append) a new sample to the series.
        For more information see

        Args:

        key:
            time-series key
        timestamp:
            Timestamp of the sample. * can be used for automatic timestamp (using the system clock).
        value:
            Numeric data value of the sample
        retention_msecs:
            Maximum age for samples compared to last event time (in milliseconds).
            If None or 0 is passed then  the series is not trimmed at all.
        uncompressed:
            Since RedisTimeSeries v1.2, both timestamps and values are compressed by default.
            Adding this flag will keep data in an uncompressed form. Compression not only saves
            memory but usually improve performance due to lower number of memory accesses.
        labels:
            Set of label-value pairs that represent metadata labels of the key.
        chunk_size:
            Each time-serie uses chunks of memory of fixed size for time series samples.
            You can alter the default TSDB chunk size by passing the chunk_size argument (in Bytes).
        duplicate_policy:
            Since RedisTimeSeries v1.4 you can specify the duplicate sample policy
            (Configure what to do on duplicate sample).
            Can be one of:
            - 'block': an error will occur for any out of order sample.
            - 'first': ignore the new value.
            - 'last': override with latest value.
            - 'min': only override if the value is lower than the existing value.
            - 'max': only override if the value is higher than the existing value.
            When this is not set, the server-wide default will be used.

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsadd
        rNrFrrr	)r
rrr
r�ADD_CMDrr)rr�	timestamp�valuerrrrrr	rrrr�add`s&










zTimeSeriesCommands.addcCs4g}|
D]}|D]}|�|�

qq|j
tg
|�
R�S)
aH

        Append (or create and append) a new `value` to series
        `key` with `timestamp`.
        Expects a list of `tuples` as (`key`,`timestamp`, `value`).
        Return value is an array with timestamps of insertions.

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsmadd
        )�appendr�MADD_CMD)rZ
ktv_tuplesrZktv�itemrrr�madd�s
	


zTimeSeriesCommands.maddc
Ks�|�d
d�}|�dd�}|�dd�}|�di�}|�dd�}|
|g}	|�
|	|�
|�|	|�
|�|	|�
|�|	|�
|�|	|�
|jtg
|	�
R�S)aN
        Increment (or create an time-series and increment) the latest
        sample's of a series.
        This command can be used as a counter or gauge that automatically gets
        history as a time series.

        Args:

        key:
            time-series key
        value:
            Numeric data value of the sample
        timestamp:
            Timestamp of the sample. None can be used for automatic timestamp (using the system clock).
        retention_msecs:
            Maximum age for samples compared to last event time (in milliseconds).
            If None or 0 is passed then  the series is not trimmed at all.
        uncompressed:
            Since RedisTimeSeries v1.2, both timestamps and values are compressed by default.
            Adding this flag will keep data in an uncompressed form. Compression not only saves
            memory but usually improve performance due to lower number of memory accesses.
        labels:
            Set of label-value pairs that represent metadata labels of the key.
        chunk_size:
            Each time-series uses chunks of memory of fixed size for time series samples.
            You can alter the default TSDB chunk size by passing the chunk_size argument (in Bytes).

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsincrbytsdecrby
        rNrrFrr)r
�_append_timestamprrr
rr�
INCRBY_CMD�
rrrrrrrrrrrrr�incrby�s









zTimeSeriesCommands.incrbyc
Ks�|�d
d�}|�dd�}|�dd�}|�di�}|�dd�}|
|g}	|�
|	|�
|�|	|�
|�|	|�
|�|	|�
|�|	|�
|jtg
|	�
R�S)a~
        Decrement (or create an time-series and decrement) the
        latest sample's of a series.
        This command can be used as a counter or gauge that
        automatically gets history as a time series.

        Args:

        key:
            time-series key
        value:
            Numeric data value of the sample
        timestamp:
            Timestamp of the sample. None can be used for automatic
            timestamp (using the system clock).
        retention_msecs:
            Maximum age for samples compared to last event time (in milliseconds).
            If None or 0 is passed then  the series is not trimmed at all.
        uncompressed:
            Since RedisTimeSeries v1.2, both timestamps and values are
            compressed by default.
            Adding this flag will keep data in an uncompressed form.
            Compression not only saves
            memory but usually improve performance due to lower number
            of memory accesses.
        labels:
            Set of label-value pairs that represent metadata labels of the key.
        chunk_size:
            Each time-series uses chunks of memory of fixed size for time series samples.
            You can alter the default TSDB chunk size by passing the chunk_size argument (in Bytes).

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsincrbytsdecrby
        rNrrFrr)r
r#rrr
rr�
DECRBY_CMDr%rrr�decrby�s"









zTimeSeriesCommands.decrbycCs|�t
|
||�S)
ao
        Delete data points for a given timeseries and interval range
        in the form of start and end delete timestamps.
        The given timestamp interval is closed (inclusive), meaning start
        and end data points will also be deleted.
        Return the count for deleted items.
        For more information see

        Args:

        key:
            time-series key.
        from_time:
            Start timestamp for the range deletion.
        to_time:
            End timestamp for the range deletion.

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsdel
        )r�DEL_CMD)rr�	from_time�to_timerrr�delete
szTimeSeriesCommands.deletecCs(|
|g}|�|||�
|j
tg
|�
R�S)
a�

        Create a compaction rule from values added to `source_key` into `dest_key`.
        Aggregating for `bucket_size_msec` where an `aggregation_type` can be
        [`avg`, `sum`, `min`, `max`, `range`, `count`, `first`, `last`,
        `std.p`, `std.s`, `var.p`, `var.s`]

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tscreaterule
        )�_append_aggregationr�CREATERULE_CMD)r�
source_key�dest_key�aggregation_type�bucket_size_msecrrrr�
createrule
s	
zTimeSeriesCommands.createrulecCs|�t
|
|�S)
z�
        Delete a compaction rule.
        For more information see

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsdeleterule
        )r�DELETERULE_CMD)rr/r0rrr�
deleterule$
szTimeSeriesCommands.deleterulecCsN|
||g}|�||�
|�
|||	�
|�||�
|�||
�
|�|||�
|S)
z*Create TS.RANGE and TS.REVRANGE arguments.)�_append_filer_by_ts�_append_filer_by_value�
_append_count�
_append_alignr-�rrr*r+�countr1r2�filter_by_ts�filter_by_min_value�filter_by_max_value�alignrrrrZ__range_params-
s





z!TimeSeriesCommands.__range_paramsNr
cCs.|�|
||||||||	|
�
}|j
tg
|�
R�S)
a
        Query a range in forward direction for a specific time-serie.

        Args:

        key:
            Key name for timeseries.
        from_time:
            Start timestamp for the range query. - can be used to express
            the minimum possible timestamp (0).
        to_time:
            End timestamp for range query, + can be used to express the
            maximum possible timestamp.
        count:
            Optional maximum number of returned results.
        aggregation_type:
            Optional aggregation type. Can be one of
            [`avg`, `sum`, `min`, `max`, `range`, `count`,
            `first`, `last`, `std.p`, `std.s`, `var.p`, `var.s`]
        bucket_size_msec:
            Time bucket for aggregation in milliseconds.
        filter_by_ts:
            List of timestamps to filter the result by specific timestamps.
        filter_by_min_value:
            Filter result by minimum value (must mention also filter
            by_max_value).
        filter_by_max_value:
            Filter result by maximum value (must mention also filter
            by_min_value).
        align:
            Timestamp for alignment control for aggregation.

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsrangetsrevrange
        )�!_TimeSeriesCommands__range_paramsr�	RANGE_CMDr:rrr�rangeD
s/









�zTimeSeriesCommands.rangecCs.|�|
||||||||	|
�
}|j
tg
|�
R�S)
a*
        Query a range in reverse direction for a specific time-series.

        **Note**: This command is only available since RedisTimeSeries >= v1.4

        Args:

        key:
            Key name for timeseries.
        from_time:
            Start timestamp for the range query. - can be used to express the minimum possible timestamp (0).
        to_time:
            End timestamp for range query, + can be used to express the maximum possible timestamp.
        count:
            Optional maximum number of returned results.
        aggregation_type:
            Optional aggregation type. Can be one of [`avg`, `sum`, `min`, `max`, `range`, `count`,
            `first`, `last`, `std.p`, `std.s`, `var.p`, `var.s`]
        bucket_size_msec:
            Time bucket for aggregation in milliseconds.
        filter_by_ts:
            List of timestamps to filter the result by specific timestamps.
        filter_by_min_value:
            Filter result by minimum value (must mention also filter_by_max_value).
        filter_by_max_value:
            Filter result by maximum value (must mention also filter_by_min_value).
        align:
            Timestamp for alignment control for aggregation.

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsrangetsrevrange
        )r@r�REVRANGE_CMDr:rrr�revrange�
s,









�zTimeSeriesCommands.revrangecCs|||g}|�||�
|�
||	|
�
|�||�
|�||�
|�||
|�
|�|||
�
|�d
g
�

||7}|�|||�
|S)z,Create TS.MRANGE and TS.MREVRANGE arguments.�FILTER)r6r7r8r9r-�_append_with_labels�extend�_append_groupby_reduce)rr1r2r;�filtersr*r+�with_labelsr<r=r>�groupby�reduce�
select_labelsr?rrrrZ__mrange_params�
s









z"TimeSeriesCommands.__mrange_paramsFcCs6|�|||||
||||	|
|||
|�}|j
tg
|�
R�S)
a�
        Query a range across multiple time-series by filters in forward direction.

        Args:

        from_time:
            Start timestamp for the range query. `-` can be used to
            express the minimum possible timestamp (0).
        to_time:
            End timestamp for range query, `+` can be used to express
            the maximum possible timestamp.
        filters:
            filter to match the time-series labels.
        count:
            Optional maximum number of returned results.
        aggregation_type:
            Optional aggregation type. Can be one of
            [`avg`, `sum`, `min`, `max`, `range`, `count`,
            `first`, `last`, `std.p`, `std.s`, `var.p`, `var.s`]
        bucket_size_msec:
            Time bucket for aggregation in milliseconds.
        with_labels:
            Include in the reply the label-value pairs that represent metadata
            labels of the time-series.
            If this argument is not set, by default, an empty Array will be
            replied on the labels array position.
        filter_by_ts:
            List of timestamps to filter the result by specific timestamps.
        filter_by_min_value:
            Filter result by minimum value (must mention also
            filter_by_max_value).
        filter_by_max_value:
            Filter result by maximum value (must mention also
            filter_by_min_value).
        groupby:
            Grouping by fields the results (must mention also reduce).
        reduce:
            Applying reducer functions on each group. Can be one
            of [`sum`, `min`, `max`].
        select_labels:
            Include in the reply only a subset of the key-value
            pair labels of a series.
        align:
            Timestamp for alignment control for aggregation.

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsmrangetsmrevrange
        )�"_TimeSeriesCommands__mrange_paramsr�
MRANGE_CMD�rr*r+rIr;r1r2rJr<r=r>rKrLrMr?rrrr�mrange�
s"@













�zTimeSeriesCommands.mrangecCs6|�|||||
||||	|
|||
|�}|j
tg
|�
R�S)
a�
        Query a range across multiple time-series by filters in reverse direction.

        Args:

        from_time:
            Start timestamp for the range query. - can be used to express
            the minimum possible timestamp (0).
        to_time:
            End timestamp for range query, + can be used to express
            the maximum possible timestamp.
        filters:
            Filter to match the time-series labels.
        count:
            Optional maximum number of returned results.
        aggregation_type:
            Optional aggregation type. Can be one of
            [`avg`, `sum`, `min`, `max`, `range`, `count`,
            `first`, `last`, `std.p`, `std.s`, `var.p`, `var.s`]
        bucket_size_msec:
            Time bucket for aggregation in milliseconds.
        with_labels:
            Include in the reply the label-value pairs that represent
            metadata labels
            of the time-series.
            If this argument is not set, by default, an empty Array
            will be replied
            on the labels array position.
        filter_by_ts:
            List of timestamps to filter the result by specific timestamps.
        filter_by_min_value:
            Filter result by minimum value (must mention also filter
            by_max_value).
        filter_by_max_value:
            Filter result by maximum value (must mention also filter
            by_min_value).
        groupby:
            Grouping by fields the results (must mention also reduce).
        reduce:
            Applying reducer functions on each group. Can be one
            of [`sum`, `min`, `max`].
        select_labels:
            Include in the reply only a subset of the key-value pair
            labels of a series.
        align:
            Timestamp for alignment control for aggregation.

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsmrangetsmrevrange
        )rNr�
MREVRANGE_CMDrPrrr�	mrevrange,s"B













�zTimeSeriesCommands.mrevrangecCs|�t
|
�S)
z�# noqa
        Get the last sample of `key`.

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsget
        )r�GET_CMD�rrrrrr
�szTimeSeriesCommands.getcCs6g}|�||�
|�
d
g
�

||
7}|jtg
|�
R�S)z�# noqa
        Get the last samples matching the specific `filter`.

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsmget
        rE)rFrGr�MGET_CMD)rrIrJrrrr�mget�s




zTimeSeriesCommands.mgetcCs|�t
|
�S)
z�# noqa
        Get information of `key`.

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsinfo
        )r�INFO_CMDrUrrr�info�szTimeSeriesCommands.infocCs|jt
g
|
�
R�S)
z�# noqa
        Get all the keys matching the `filter` list.

        For more information: https://oss.redis.com/redistimeseries/master/commands/#tsqueryindex
        )r�QUERYINDEX_CMD)rrIrrr�
queryindex�szTimeSeriesCommands.queryindexcCs|
r|�d
g
�

dS)z"Append UNCOMPRESSED tag to params.ZUNCOMPRESSEDN�
rG)rrrrrr�s
z'TimeSeriesCommands._append_uncompressedcCs8|
r|rtd
�
�
|
r |�
dg
�

|r4|�
dg
|�
�

dS)z!Append labels behavior to params.z:with_labels and select_labels cannot be provided together.Z
WITHLABELSZSELECTED_LABELSN)rrG)rrJrMrrrrF�s

�


z&TimeSeriesCommands._append_with_labelscCs*|
d
u
r&|d
u
r&|�d|
d|�
�g�

d
S)z)Append GROUPBY REDUCE property to params.NZGROUPBYZREDUCE)rG�upper)rrKrLrrrrH�s
z)TimeSeriesCommands._append_groupby_reducecCs|
d
u
r|�d|
g�

d
S)z$Append RETENTION property to params.NZ	RETENTIONr\)rZ	retentionrrrr�s
z$TimeSeriesCommands._append_retentioncCs2|
r.|�d
�

|
�
�D]\}}|�||g�

qdS)z!Append LABELS property to params.ZLABELSN)r�itemsrG)rr�
k�
vrrrr�s



z!TimeSeriesCommands._append_labelscCs|
d
u
r|�d|
g�

d
S)z Append COUNT property to params.NZCOUNTr\)rr;rrrr8�s
z TimeSeriesCommands._append_countcCs|
d
u
r|�d|
g�

d
S)z$Append TIMESTAMP property to params.N�	TIMESTAMPr\)rrrrrr#�s
z$TimeSeriesCommands._append_timestampcCs|
d
u
r|�d|
g�

d
S)z Append ALIGN property to params.NZALIGNr\)rr?rrrr9�s
z TimeSeriesCommands._append_aligncCs$|
d
u
r |�d�

|�
|
|g�

d
S)z&Append AGGREGATION property to params.NZAGGREGATION)rrG)rr1r2rrrr-�s


z&TimeSeriesCommands._append_aggregationcCs|
d
u
r|�d|
g�

d
S)z%Append CHUNK_SIZE property to params.NZ
CHUNK_SIZEr\)rrrrrr
�s
z%TimeSeriesCommands._append_chunk_sizecCs2|d
u
r.|
dkr |�d|g�

n|�d|g�

d
S)z^Append DUPLICATE_POLICY property to params on CREATE
        and ON_DUPLICATE on ADD.
        NrZON_DUPLICATEZDUPLICATE_POLICYr\)rZcommandr	rrrr�s

z+TimeSeriesCommands._append_duplicate_policycCs|
d
u
r|�dg
|
�
�

d
S)z'Append FILTER_BY_TS property to params.NZFILTER_BY_TSr\)rZts_listrrrr6�s
z&TimeSeriesCommands._append_filer_by_tscCs$|
d
u
r |d
u
r |�d|
|g�

d
S)z*Append FILTER_BY_VALUE property to params.NZFILTER_BY_VALUEr\)rZ	min_valueZ	max_valuerrrr7�s
z)TimeSeriesCommands._append_filer_by_value)NNr
NNNN)NNr
NNNN)NNr
FNNNNNNN)NNr
FNNNNNNN)
F)
N)%�__name__�
__module__�__qualname__�__doc__rrrr"r&r(r,r3r5r@rBrDrNrQrSr
rWrYr[�staticmethodrrFrHrrr8r#r9r-r
rr6r7rrrrrs�
44,0	





�
B





�
:#









�
X









�
U

























rN)Zredis.exceptionsrrrr.rr'r4r)rTr$rXr rVrOrRrZrArCrrrrr�<module>
s$