Your IP : 216.73.216.213
R6�W� � @ s� d Z d d l Td d l m Z d d l m Z d d l m Z d d l m Z d d l m Z e � Z
i Z Gd d � d e � Z Gd
d � d e e e � � Z d d
� Z Gd d � d e
� Z Gd d � d e
� Z Gd d � d e
� Z d S)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 Z d Z � f d d � Z � S)r c s> t t | � j | | | | � } | d k r: t j | � | S)N�_metaclass_helper_r )r r )�superr �__new__�
all_models�add)�cls�name�bases�attrs)� __class__� �@/opt/alt/python35/lib64/python3.5/site-packages/playhouse/gfk.pyr
' s !
zBaseModel.__new__)�__name__�
__module__� __qualname__r
r r )r r r & s r c @ s e Z d Z d S)r N)r r r r r r r r - s r c C sG | t k r: x+ t D]# } | j j | k r | t | <Pq Wt j | � S)N)� table_cacher �_meta�db_table�get)�tbl_name�modelr r r � get_model0 s
r c @ s^ e Z d Z d Z d d � Z e d d � � Z e d d � � Z d d
� Z d d � Z d
S)�
BoundGFKField� model_class� gfk_fieldc C s | | _ | | _ d S)N)r r )�selfr r r r r �__init__; s zBoundGFKField.__init__c C s\ | j j j } t | j j | j j f � } x( | D] \ } } | t | � s4 d Sq4 Wd S)NTF)r r �indexes�setr �model_type_field�model_id_field)r! r# �fieldsZindexed_columnsZ is_uniquer r r �unique? s zBoundGFKField.uniquec C sW | j j j } t | t � rS t | j j | j j f � } | t | j � sS d Sd S)NTF)
r r � primary_key�
isinstanceZ CompositeKeyr$ r r% r& � field_names)r! Zpkr' r r r r) I s zBoundGFKField.primary_keyc C sR | j j } | j | j j } | j | j j } | | j j k | | j � k @S)N)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 sX t | � } | j j | j j } | j j | j j } | | j j k | | j � k @S)N)�typer r' r r% r& r r, )r! r- Z other_clsr/ r0 r r r �__ne__[ s
zBoundGFKField.__ne__N)r r )
r r r � __slots__r" �propertyr( r) r1 r3 r r r r r 8 s
r c @ sI e Z d Z d d d d � Z d d � Z d d d � Z d
d � Z d S) �GFKField� object_type� object_idc C s4 | | _ | | _ d j | j | j f � | _ d S)N�.)r% r& �join�att_name)r! r% r& r r r r" e s zGFKField.__init__c C s� | j } | j | j � r� | j | j � r� | | j } t | � } | s\ t d | � � | j � j | j j | | j k � } | j � Sd 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 s] | rP | j | j k r= | j | � } | r= | | j | j <| j j | j � St | | � S)N)r; �
_obj_cacherB r r )r! r@ �
instance_typeZrel_objr r r �__get__w s zGFKField.__get__c C s@ | | j | j <| j j | j | j <| j � | j | j <d S)N)rC r; r r r< r% r, r& )r! r@ �valuer r r �__set__� s zGFKField.__set__)r r r r" rB rE rG r r r r r6 d s
r6 c @ s= e Z d Z d d d d � Z d d d � Z d d � Z d S)
�
ReverseGFKr7 r8 c C s3 | | _ | j j | | _ | j j | | _ d S)N)r r r' r% r& )r! r r% r& r r r r" � s zReverseGFK.__init__Nc C sg | r> | j j � j | j | j j k | j | j � k @� S| j j � j | j | j j k � Sd S)N)r r> r? r% r r r& r, )r! r@ rD r r r rE � s zReverseGFK.__get__c s� | j j } | j � } t | t � rp | j � j k rp t � j � j | � j | i � j | j
� j � n{ t t
� f d d � | � � r� xW | D]< } t | � j j | � t | � j j | � | j � q� Wn t d | � � d S)Nc s t | � j � S)N)r* r )�i)r! r r �<lambda>� s z$ReverseGFK.__set__.<locals>.<lambda>z&ReverseGFK field unable to handle "%s")r r r, r* r r r r% r& r? Z_where�execute�all�map�setattrr Zsave�
ValueError)r! r@ rF ZmtvZmiv�objr )r! r rG � s !
zReverseGFK.__set__)r r r r" rE rG r r r r rH � s rH N)�__doc__Zpeeweer Z
_BaseModelr Z_Modelr r r r$ r r r �objectr r6 rH r r r r �<module> s
,"