R6W @ 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_modelsadd)clsnamebasesZattrs) __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 _metadb_tableget)tbl_namemodelr 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 indexessetr model_type_fieldmodel_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 othermeta
type_fieldid_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% joinatt_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 AttributeErrorselectwherer r( )r instancedatar 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_whereZexecuteallmapsetattrr save
ValueError)r r? rE ZmtvZmivobjr )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
,"