File: //opt/alt/python38/lib64/python3.8/site-packages/playhouse/__pycache__/gfk.cpython-38.pyc
U
S��W� � @ s� d Z ddlT ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ e � Z
i Z G dd � d e�ZG d
d � d eee��Zd d
� Z G dd� de
�ZG dd� de
�ZG dd� de
�ZdS )a
Provide a "Generic ForeignKey", similar to Django. A "GFK" is composed of two
columns: an object ID and an object type identifier. The object types are
collected in a global registry (all_models), so all you need to do is subclass
``gfk.Model`` and your model will be added to the registry.
Example:
class Tag(Model):
tag = CharField()
object_type = CharField(null=True)
object_id = IntegerField(null=True)
object = GFKField('object_type', 'object_id')
class Blog(Model):
tags = ReverseGFK(Tag, 'object_type', 'object_id')
class Photo(Model):
tags = ReverseGFK(Tag, 'object_type', 'object_id')
tag.object -> a blog or photo
blog.tags -> select query of tags for ``blog`` instance
Blog.tags -> select query of all tags for Blog instances
� )�*)� BaseModel)�Model)� SelectQuery)� UpdateQuery)�with_metaclassc s e Zd Z� fdd�Z� ZS )r c s, t t| ��| |||�} |dkr(t�| � | S )N)�_metaclass_helper_r )�superr �__new__�
all_models�add)�cls�name�bases�attrs�� __class__� �@/opt/alt/python38/lib64/python3.8/site-packages/playhouse/gfk.pyr
' s
zBaseModel.__new__)�__name__�
__module__� __qualname__r
�
__classcell__r r r r r & s r c @ s e Zd ZdS )r N)r r r r r r r r - s r c C s4 | t kr*tD ] }|jj| kr |t | < q*q t �| �S �N)� table_cacher �_meta�db_table�get)�tbl_name�modelr r r � get_model0 s r c @ s@ e Zd ZdZdd� Zedd� �Zedd� �Zdd � Zd
d � Z d S )
�
BoundGFKField�� model_class� gfk_fieldc C s || _ || _d S r r" )�selfr# r$ r r r �__init__; s zBoundGFKField.__init__c C sB | j jj}t| jj| jjf�}|D ]\}}|t|� s" dS q"dS �NTF)r# r �indexes�setr$ �model_type_field�model_id_field)r% r( �fieldsZindexed_columns� is_uniquer r r �unique? s
� zBoundGFKField.uniquec C s> | j jj}t|t�r:t| jj| jjf�}|t|j � s:dS dS r' )
r# r � primary_key�
isinstance� CompositeKeyr) r$ r* r+ � field_names)r% �pkr, r r r r/ I s
�zBoundGFKField.primary_keyc C s< | j j}|j| jj }|j| jj }||jjk||�� k@ S r )r# r r, r$ r* r+ r �
_get_pk_value)r% �other�meta�
type_field�id_fieldr r r �__eq__S s
��zBoundGFKField.__eq__c C s@ t |�}|jj| jj }|jj| jj }||jjk||�� k@ S r )�typer r, r$ r* r+ r r4 )r% r5 Z other_clsr7 r8 r r r �__ne__[ s
��zBoundGFKField.__ne__N)
r r r � __slots__r& �propertyr. r/ r9 r; r r r r r! 8 s
r! c @ s0 e Zd Zd dd�Zdd� Zd
dd �Zd
d � ZdS )�GFKField� object_type� object_idc C s$ || _ || _d�| j | jf�| _d S )N�.)r* r+ �join�att_name)r% r* r+ r r r r& e s zGFKField.__init__c C sh |j }|�| j�rd|�| j�rd|| j }t|�}|s@td| ��|�� �|jj || j k�}|�� S d S )Nz-Model for table "%s" not found in GFK lookup.)
�_datar r* r+ r �AttributeError�select�wherer r/ )r% �instance�datar r# �queryr r r �get_objk s
��zGFKField.get_objNc C sB |r8| j |jkr*| �|�}|r*||j| j < |j�| j �S t|| �S r )rC �
_obj_cacherK r r! )r% rH �
instance_typeZrel_objr r r �__get__w s
zGFKField.__get__c C s0 ||j | j< |jj|j| j< |�� |j| j< d S r )rL rC r r rD r* r4 r+ )r% rH �valuer r r �__set__� s zGFKField.__set__)r? r@ )N)r r r r& rK rN rP r r r r r> d s �
r> c @ s( e Zd Zd
dd�Zd dd�Zdd � ZdS ) �
ReverseGFKr? r@ c C s& || _ |jj| | _|jj| | _d S r )r# r r, r* r+ )r% r r* r+ r r r r& � s zReverseGFK.__init__Nc C sJ |r,| j �� �| j|jjk| j|�� k@ �S | j �� �| j|jjk�S d S r )r# rF rG r* r r r+ r4 )r% rH rM r r r rN � s
��
�zReverseGFK.__get__c s� |j j}|�� }t|t�rL|j� jkrLt� j� j|� j|i�� |j
�� � nVt t
� fdd�|��r�|D ],}t|� jj|� t|� jj|� |�� qfn td| ��d S )Nc s t | � j�S r )r0 r# )�i�r% r r �<lambda>� � z$ReverseGFK.__set__.<locals>.<lambda>z&ReverseGFK field unable to handle "%s")r r r4 r0 r r# r r* r+ rG �_where�execute�all�map�setattrr �save�
ValueError)r% rH rO ZmtvZmiv�objr rS r rP � s&
� ��
zReverseGFK.__set__)r? r@ )N)r r r r& rN rP r r r r rQ � s
�
rQ N)�__doc__�peeweer Z
_BaseModelr Z_Modelr r r r) r r r �objectr! r> rQ r r r r �<module> s ,"