Current 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)Z_metaclass_helper_r )�superr �__new__�
all_models�add)�cls�name�basesZattrs�� __class__� �/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_columnsZ 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�
isinstanceZ CompositeKeyr' r" r( r) � field_names)r# Zpkr* 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 r/ )r# r0 Z other_clsr2 r3 r r r �__ne__[ s
��zBoundGFKField.__ne__N)
r r r � __slots__r$ �propertyr+ r, r4 r6 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! Zqueryr r r �get_objk s
��zGFKField.get_objNc C sB |r8| j |jkr*| �|�}|r*||j| j < |j�| j �S t|| �S r )r> �
_obj_cacherE r r )r# rC �
instance_typeZrel_objr r r �__get__w s
zGFKField.__get__c C s0 ||j | j< |jj|j| j< |�� |j| j< d S r )rF r> r r r? r( r/ r) )r# rC �valuer r r �__set__� s zGFKField.__set__)r: r; )N)r r r r$ rE rH rJ r r r r r9 d s �
r9 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! rA rB r( r r r) r/ )r# rC rG r r r rH � 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 )r- r! )�i�r# r r �<lambda>� � z$ReverseGFK.__set__.<locals>.<lambda>z&ReverseGFK field unable to handle "%s")r r r/ r- r r! r r( r) rB Z_whereZexecute�all�map�setattrr
Zsave�
ValueError)r# rC rI ZmtvZmiv�objr rM r rJ � s&
� ��
zReverseGFK.__set__)r: r; )N)r r r r$ rH rJ r r r r rK � s
�
rK N)�__doc__Zpeeweer Z
_BaseModelr Z_Modelr r r r' r
r r �objectr r9 rK r r r r �<module> s ,"
Mini Shell