File: //home/arjun/projects/buyercall/buyercall/lib/__pycache__/util_sqlalchemy.cpython-310.pyc
o
�we3 � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ G d d
� d
e�ZG d d � d e
�ZG d
d� de�ZdS )� N)�inspect)�DateTime)�
TypeDecorator)�UUID)�SerializerMixin)�tzware_datetime)�dbc @ s. e Zd ZdZedd�ZdZdd� Zdd� Zd S )
�
AwareDateTimez�
A DateTime type which can only store tz-aware DateTimes.
Source:
https://gist.github.com/inklesspen/90b554c864b99340747e
T)�timezoneFc C s( t |tj�r|jd u rtd�|���|S )Nz{!r} must be TZ-aware)�
isinstance�datetime�tzinfo�
ValueError�format)�self�value�dialect� r �?/home/arjun/projects/buyercall/buyercall/lib/util_sqlalchemy.py�process_bind_param s z AwareDateTime.process_bind_paramc C s dS )NzAwareDateTime()r �r r r r �__repr__ s zAwareDateTime.__repr__N) �__name__�
__module__� __qualname__�__doc__r �impl�cache_okr r r r r r r s
r c @ s e Zd Zedd� ffZdS )�CustomSerializerMixinc C s t | �S �N��str)�xr r r �<lambda> s z CustomSerializerMixin.<lambda>N)r r r r �serialize_typesr r r r r s
�r c @ s� e Zd Zeje� ded�Zeje� deed�Ze dd� �Z
e g dfdd��Z e d d
� �Z d d � Z
d
d� Zdd� Ze dd� �Ze dd� �Ze dd� �Ze dd� �Ze dd� �Zd dd �Zd S ) �
ResourceMixinF)�nullable�default)r& r' �onupdatec C s$ || j jvrd}|dvrd}||fS )z�
Sort a resource by a specific field and direction.
:param field: Field name
:type field: str
:param direction: Direction
:type direction: str
:return: tuple
�
created_on)�asc�descr* )� __table__�columns)�cls�field� directionr r r �sort_by. s
zResourceMixin.sort_by� c sX t tt� ��� |dkr | j�| j��| �|��}dd� |D �}� r*� fdd�|D �}|S )a�
Determine which IDs are to be modified.
:param scope: Affect all or only a subset of items
:type scope: str
:param ids: List of ids to be modified
:type ids: list
:param omit_ids: Remove 1 or more IDs from the list
:type omit_ids: list
:param query: Search query (if applicable)
:type query: str
:return: list
�all_search_resultsc S s g | ]}t |d ��qS )r r )�.0�itemr r r �
<listcomp>Y � z5ResourceMixin.get_bulk_action_ids.<locals>.<listcomp>c s g | ]}|� vr|�qS r r )r4 �id��omit_idsr r r6 _ r7 )�list�mapr! �query�
with_entitiesr8 �filter�search)r. �scope�idsr: r= r r9 r �get_bulk_action_idsB s z!ResourceMixin.get_bulk_action_idsc C s* | j �| j�|��jdd�}tj�� |S )z�
Delete 1 or more model instances.
:param ids: List of ids to be deleted
:type ids: list
:return: Number of deleted instances
F)�synchronize_session)r= r? r8 �in_�deleter �session�commit)r. rB � delete_countr r r � bulk_deletec s
�
zResourceMixin.bulk_deletec C s t j�| � t j�� | S )z?
Save a model instance.
:return: self
)r rG �addrH r r r r �saver s
zResourceMixin.savec C s t j�| � t j�� S )zY
Delete a model instance.
:return: db.session.commit()'s result
)r rG rF rH r r r r rF } s
zResourceMixin.deletec sB t t� ��}� jj�� }d�� fdd�|D ��}d|� jj|f S )Nz, c 3 s$ � | ]
}d |t � |d�f V qdS )z%s=%rN)�getattr)r4 �nr r r � <genexpr>� s �" z(ResourceMixin.__str__.<locals>.<genexpr>z <%s %s(%s)>)�hexr8 r, �c�keys�join� __class__r )r �obj_idr- �valuesr r r �__str__� s zResourceMixin.__str__c C �< d }|r z| j �| j|k��� }|j}W |S Y |S |S r )r= r? r8 �first�sid)r. r8 rZ �objr r r �get_sid_from_id� � �z ResourceMixin.get_sid_from_idc C rX r )r= r? rZ rY r8 )r. rZ r8 r[ r r r �get_id_from_sid� r] z ResourceMixin.get_id_from_sidc K s | di |��}|� � |S )Nr )rL )r. �paramsr[ r r r �create� s zResourceMixin.createc C � |r
| j �| j|k��� S d S r )r= r? r8 rY )r. �id_r r r � get_by_id� � zResourceMixin.get_by_idc C ra r )r= r? rZ rY )r. rZ r r r �
get_by_sid� rd zResourceMixin.get_by_sidc
K sl z t � }|�� D ]
\}}t| ||� q| �� |r | W S dW S ty5 } z t|� W Y d }~dS d }~ww )NTF)�locals�items�setattrrL � Exception�print)r �
return_object�kwargsr_ �k�v�er r r �put� s ��zResourceMixin.putN)F)r r r r �Columnr r r) �
updated_on� classmethodr1 rC rJ rL rF rW r\ r^ r` rc re rp r r r r r% $ s@ ��
�
r% )r �sqlalchemy.inspectionr �
sqlalchemyr �sqlalchemy.typesr �uuidr �sqlalchemy_serializerr �buyercall.lib.util_datetimer �buyercall.extensionsr r r r% r r r r �<module> s