File: //lib/python3.9/site-packages/redis/commands/search/__pycache__/querystring.cpython-39.pyc
a
��bO � @ s, d d� Z d5dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zd6dd�ZG dd� d�ZG dd� de�Z G dd� de�Z
G dd� de�Z G dd� de�Z G d d � d �Z
G d d � d e
�ZG d d!� d!e
�ZG d"d#� d#e
�ZG d$d%� d%e�ZG d&d'� d'e�ZG d(d)� d)e�Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4S )7c G s | s t d��t| � S )zx
Indicate that the values should be matched to a tag field
### Parameters
- **t**: Tags to search for
z"At least one tag must be specified)�
ValueError�TagValue)�t� r �E/usr/lib/python3.9/site-packages/redis/commands/search/querystring.py�tags s r Tc C s t | |||d�S )z0
Indicate that value is a numeric range
)�
inclusive_min�
inclusive_max)�
RangeValue)�a�br r r r r �between s r c C s
t | | �S )z
Match a numeric value
�r ��nr r r �equal s r c C s t d| dd�S )z%
Match any value less than n
NF�r r
r r r r �lt s r c C s t d| dd�S )z,
Match any value less or equal to n
NTr r
r r r r �le# s r c C s t | ddd�S )z(
Match any value greater than n
NF�r r
r r r r �gt* s r c C s t | ddd�S )z/
Match any value greater or equal to n
NTr r
r r r r �ge1 s r �kmc C s t | |||�S )z-
Indicate that value is a geo region
)�GeoValue)�lat�lon�radius�unitr r r �geo8 s r c @ s4 e Zd Zedd� �Zedd� �Zdd� Zdd� Zd S )
�Valuec C s dS )z�
Whether this type of value may be combined with other values
for the same field. This makes the filter potentially more efficient
Fr ��selfr r r �
combinable@ s zValue.combinablec C s t | t�r| S t| �S )zL
Convert an object to a value, if it is not a value already
)�
isinstancer � ScalarValue��vr r r �
make_valueH s
zValue.make_valuec C s
t � �d S �N��NotImplementedErrorr r r r � to_stringQ s zValue.to_stringc C s | � � S r' �r* r r r r �__str__T s z
Value.__str__N) �__name__�
__module__� __qualname__�propertyr! � staticmethodr& r* r, r r r r r ? s
r c @ s" e Zd ZdZddd�Zdd� ZdS )r Fc C s: |d u r d}|d u rd}t |�t |�g| _|| _|| _d S )Nz-inf�inf)�str�ranger r )r r
r r r r r r �__init__[ s zRangeValue.__init__c C s$ d� | j| jsdnd| js dnd�S )Nz[{1}{0[0]} {2}{0[1]}]�(� )�formatr4 r r r r r r r* d s
�zRangeValue.to_stringN)FF�r- r. r/ r! r5 r* r r r r r X s
r c @ s e Zd ZdZdd� Zdd� ZdS )r# Tc C s t |�| _d S r' )r3 r% )r r% r r r r5 o s zScalarValue.__init__c C s | j S r' r$ r r r r r* r s zScalarValue.to_stringNr9 r r r r r# l s r# c @ s e Zd ZdZdd� Zdd� ZdS )r Fc G s
|| _ d S r' )r )r r r r r r5 y s zTagValue.__init__c C s dd� dd� | jD �� d S )N�{z | c s s | ]}t |�V qd S r' )r3 )�.0r r r r � <genexpr>} � z%TagValue.to_string.<locals>.<genexpr>�})�joinr r r r r r* | s zTagValue.to_stringNr9 r r r r r v s r c @ s e Zd Zddd�ZdS )r r c C s || _ || _|| _|| _d S r' )r r r r )r r r r r r r r r5 � s zGeoValue.__init__N)r )r- r. r/ r5 r r r r r � s r c @ sN e Zd Zdd� Zdd� Zedd� �Zedd� �Zdd
d �Z d d
� Z
dd� Z d S )�Nodec O s� g | _ i }|�� D ]`\}}|�|g �}t|tttf�rH|�t� |�� qt|t�r^|�|� q|�
dd� |D �� q| j dd� |D �7 _ |�� D ] \}}| j �
| � ||�� q�dS )a�
Create a node
### Parameters
- **children**: One or more sub-conditions. These can be additional
`intersect`, `disjunct`, `union`, `optional`, or any other `Node`
type.
The semantics of multiple conditions are dependent on the type of
query. For an `intersection` node, this amounts to a logical AND,
for a `union` node, this amounts to a logical `OR`.
- **kwparams**: key-value parameters. Each key is the name of a field,
and the value should be a field value. This can be one of the
following:
- Simple string (for text field matches)
- value returned by one of the helper functions
- list of either a string or a value
### Examples
Field `num` should be between 1 and 10
```
intersect(num=between(1, 10)
```
Name can either be `bob` or `john`
```
union(name=("bob", "john"))
```
Don't select countries in Israel, Japan, or US
```
disjunct_union(country=("il", "jp", "us"))
```
c s s | ]}t �|�V qd S r' )r r& )r; Zsubvr r r r< � r= z Node.__init__.<locals>.<genexpr>c S s g | ]}t �|��qS r )r@ �to_node)r; �pr r r �
<listcomp>� r= z!Node.__init__.<locals>.<listcomp>N) �params�items�
setdefaultr" r3 �int�float�appendr r& �extend� join_fields)r ZchildrenZkwparamsZkvparams�kr% Zcurvalsr r r r5 � s +
z
Node.__init__c st t |�dkr*td� � d|d �� � ��gS |d jsF� fdd�|D �S td� � d| j�dd � |D ��� d
��}|gS ) N� �@�:� c s$ g | ] }t d � � d|�� � ���qS )rN rO )�BaseNoder* �r; r% ��keyr r rC � r= z$Node.join_fields.<locals>.<listcomp>z:(c s s | ]}|� � V qd S r' r+ rR r r r r< � r= z#Node.join_fields.<locals>.<genexpr>�))�lenrQ r* r! �JOINSTRr? )r rT Zvals�sr rS r rK � s
(zNode.join_fieldsc C s t |t�r|S t|�S r' )r" r@ rQ )�cls�objr r r rA � s
z Node.to_nodec C s
t � �d S r' r( r r r r rW � s z Node.JOINSTRNc C s> | � |�}|rdnd\}}|� | j�dd� | jD ��� |� �S )N)r6 rU )r7 r7 c s s | ]}|� � V qd S r' r+ )r; r r r r r< � r= z!Node.to_string.<locals>.<genexpr>)�_should_use_parenrW r? rD )r � with_parensZpreZpostr r r r* � s
zNode.to_stringc C s |d ur |S t | j�dkS )NrM )rV rD )r Zoptvalr r r r[ � s zNode._should_use_parenc C s | � � S r' r+ r r r r r, � s z Node.__str__)N) r- r. r/ r5 rK � classmethodrA r0 rW r* r[ r, r r r r r@ � s <
r@ c s& e Zd Z� fdd�Zddd�Z� ZS )rQ c s t � �� t|�| _d S r' )�superr5 r3 rX )r rX �� __class__r r r5 � s
zBaseNode.__init__Nc C s | j S r' )rX )r r\ r r r r* � s zBaseNode.to_string)N)r- r. r/ r5 r* �
__classcell__r r r_ r rQ � s rQ c @ s e Zd ZdZdZdS )�
IntersectNodezw
Create an intersection node. All children need to be satisfied in order for
this node to evaluate as true
� N�r- r. r/ �__doc__rW r r r r rb � s rb c @ s e Zd ZdZdZdS )� UnionNodezv
Create a union node. Any of the children need to be satisfied in order for
this node to evaluate as true
�|Nrd r r r r rf � s rf c s" e Zd ZdZd� fdd� Z� ZS )� DisjunctNodezs
Create a disjunct node. In order for this node to be true, all of its
children must evaluate to false
Nc s4 | � |�}t� jdd�}|r(d| d S d| S d S )NF�r\ z(-rU �-�r[ r^ r* �r r\ �retr_ r r r* s
zDisjunctNode.to_string)N�r- r. r/ re r* ra r r r_ r rh s rh c @ s e Zd ZdZdZdS )�DistjunctUnionz~
This node is true if *all* of its children are false. This is equivalent to
```
disjunct(union(...))
```
rg Nrd r r r r ro s ro c s" e Zd ZdZd� fdd� Z� ZS )� OptionalNodez}
Create an optional node. If this nodes evaluates to true, then the document
will be rated higher in score/rank.
Nc s4 | � |�}t� jdd�}|r(d| d S d| S d S )NFri z(~rU �~rk rl r_ r r r* s
zOptionalNode.to_string)Nrn r r r_ r rp s rp c O s t | i |��S r' )rb ��args�kwargsr r r � intersect) s ru c O s t | i |��S r' )rf rr r r r �union- s rv c O s t | i |��S r' )rh rr r r r �disjunct1 s rw c O s t | i |��S r' )ro rr r r r �disjunct_union5 s rx c O s t | i |���� S r' )ru r* rr r r r � querystring9 s ry N)TT)r )r r r r r r r r r r r# r r r@ rQ rb rf rh ro rp ru rv rw rx ry r r r r �<module> s0
]