o
�D�i� � @ s� d Z ddlmZmZ ddlmZ ddlmZmZ G dd� de�Z eG dd� d��Z
eG d d
� d
��Z G d d � d �Z G d
d� d�Z
dS )z Advanced tile query builder.� )� dataclass�field)�Enum)�Optional�Callablec @ s e Zd ZdZdZdS )� SortOrder�asc�descN)�__name__�
__module__� __qualname__�ASC�DESC� r r �)/tmp/py-qry/src/plato_tile_query/query.pyr s r c @ s& e Zd ZU eed< eed< eed< dS )�Filterr �operator�valueN)r
r r �str�__annotations__�anyr r r r r
s
r c @ s: e Zd ZU ee ed< eed< eed< eed< eed< dS )� QueryResult�items�total�page�per_page�filtered_totalN)r
r r �list�dictr �intr r r r r s
r c @ s* e Zd Zd1dee fdd�Zdee dd fdd�Zded ed
edd fd d �Z d
edd fdd�Z
de dd fdd�Z de dd fdd�Z
dedd fdd�Zdedd fdd�Zd2ded edd fd d �Zd edd fd!d"�Zd#edd fd$d%�Zd&edd fd'd(�Zdefd)d*�Zd+ed,edefd-d.�Zdefd/d0�ZdS )3� QueryBuilderN�
collectionc C s: |pg | _ g | _d| _tj| _d| _d| _d| _g | _ d S )N� � � )
� _collection�_filters� _sort_fieldr r � _sort_order�_page� _per_page�_search�_fields��selfr! r r r �__init__ s
zQueryBuilder.__init__�returnc C �
|| _ | S �N)r% r- r r r �from_collection# � z QueryBuilder.from_collectionr �opr c C s | j �t|||d�� | S )N)r r r )r& �appendr )r. r r5 r r r r �where' s zQueryBuilder.where�domainc C � | � dd|�S )Nr8 �eq�r7 )r. r8 r r r r8 + � zQueryBuilder.domain�min_confc C r9 )N�
confidence�gter; )r. r= r r r �confidence_min. r< zQueryBuilder.confidence_min�max_confc C r9 )Nr> �lter; )r. rA r r r �confidence_max1 r< zQueryBuilder.confidence_max�priorityc C r9 )NrD r: r; )r. rD r r r rD 4 r< zQueryBuilder.priority�queryc C r1 r2 )r+ )r. rE r r r �search7 r4 zQueryBuilder.searchr �orderc C s || _ t|�| _| S r2 )r' r r( )r. r rG r r r �sort_by; s
zQueryBuilder.sort_by�numc C r1 r2 )r) )r. rI r r r r @ r4 zQueryBuilder.page�nc C r1 r2 )r* )r. rJ r r r r D r4 zQueryBuilder.per_page�fieldsc G s t |�| _| S r2 )r r, )r. rK r r r �selectH s
zQueryBuilder.selectc s� t �j�}�jD ] � � �fdd�|D �}q�jr&�j�� ��fdd�|D �}�jr:�jtjk}|j �fdd�|d� �j
rF�fdd�|D �}t |�}�j d �j
}|�j
}|||� }t|t �j��j �j
|d �S )
Nc s g | ]
}�� |� �r|�qS r )�_match��.0�item)�fr. r r �
<listcomp>P s z(QueryBuilder.execute.<locals>.<listcomp>c sX g | ](}� t |�d d���� v s(� t |�dd���� v s(� t |�dd���� v r|�qS )�contentr" r8 �id)r �get�lowerrN )�qr r rR T s �c s | � � jd�S )Nr )rU r' )�x�r. r r �<lambda>[ s z&QueryBuilder.execute.<locals>.<lambda>)�key�reversec s g | ] � � fd d��j D ��qS )c s i | ] }|� v r|� � |��qS r )rU )rO �k�rP r r �
<dictcomp>^ � z3QueryBuilder.execute.<locals>.<listcomp>.<dictcomp>)r, )rO rY r^ r rR ^ r` r# )r r r r r )r r% r& r+ rV r' r( r r �sortr, �lenr) r* r )r. �resultsr\ r �start�end�
page_itemsr )rQ rW r. r �executeL s&
�zQueryBuilder.executerP rQ c C s� |� |j�}|d u r dS dd� dd� dd� dd� dd� dd� d d� d
d� d �}|� |j�}|rCz|||j�W S ttfyB Y dS w dS ) NFc S s | |kS r2 r ��a�br r r rZ l � z%QueryBuilder._match.<locals>.<lambda>c S s | |kS r2 r rh r r r rZ l rk c S s | |kS r2 r rh r r r rZ m rk c S s | |k S r2 r rh r r r rZ m rk c S s | |kS r2 r rh r r r rZ n rk c S s | |kS r2 r rh r r r rZ n rk c S s | r|t | �v S dS )NF)r rh r r r rZ o s c S s | |v S r2 r rh r r r rZ p rk )r: �ne�gt�ltr? rB �contains�in)rU r r r � TypeError�
ValueError)r. rP rQ �val�ops�fnr r r rM h s" � �zQueryBuilder._matchc C s t | �� j�S r2 )rb rg r rY r r r �county r< zQueryBuilder.countr2 )r )r
r r r r r/ r3 r r r7 r8 �floatr@ rC rD rF rH r r r rL r rg r �boolrM rv r r r r r s
r c @ sp e Zd ZdZddee fdd�Z dd
ed ed ed
edede dee fdd�Z
dde dedee fdd�Z dS )� TileQueryz High-level tile query interface.N�tilesc C s |pg | _ t| j �| _d S r2 )�_tilesr �_builder)r. rz r r r r/ ~ s
zTileQuery.__init__r" � r> r$ rE r8 r= rD ra �limitr0 c C sn t | j�}|r |�|� |r|�|� |dkr |�|� |r#|�|� |r+|�|d� |�|� |�� }|j S )Nr r )
r r{ rF r8 r@ rD rH r rg r ) r. rE r8 r= rD ra r~ rj �resultr r r �find� s
zTileQuery.find� rJ r c C s | j d||d�S )Nr} )r= ra r~ )r� )r. rJ r r r r �top_n� s zTileQuery.top_nr2 )r" r" r} r" r> r$ )r� r> ) r
r r �__doc__r r r/ r rw r r� r� r r r r ry | s ����
� ry N)r� � dataclassesr r �enumr �typingr r r r r r ry r r r r �<module> s d