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�basesZattrs)� __class__� �/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. r/ r r r �__ne__[ s
zBoundGFKField.__ne__N)r r )
r r r � __slots__r! �propertyr' r( r0 r2 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_cacherA 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)rB r: r r r; r$ r+ r% )r r? �valuer r r �__set__� s zGFKField.__set__)r r r r! rA rD rF r r r r r5 d s
r5 c @ s= e Z d Z d d d d � Z d d d � Z d d � Z d S)
�
ReverseGFKr6 r7 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? rC r r r rD � 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_whereZexecute�all�map�setattrr �save�
ValueError)r r? rE ZmtvZmiv�objr )r r rF � s !
zReverseGFK.__set__)r r r r! rD rF r r r r rG � s rG N)�__doc__Zpeeweer Z
_BaseModelr Z_Modelr r r r# r r r �objectr r5 rG r r r r �<module> s
,"