R6W4 - @ sH d Z d d l 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 d d l m Z d d
l m Z d d l m
Z
d d l m Z d d
l m Z y d d l
m Z e j Wn e k
r Yn Xd d l m Z d d l m Z d d l m Z d d l m Z y d d l m Z Wn d Z Yn Xe j d d d d Z Gd d d e Z Gd d d e Z Gd d d e Z Gd d d e Z Gd d! d! e Z Gd" d# d# e Z d$ d% Z! e e e! Gd& d' d' e Z" Gd( d) d) e" e# Z$ Gd* d+ d+ e% Z& Gd, d- d- e" e# Z' Gd. d/ d/ e# Z( Gd0 d1 d1 e" e( Z) Gd2 d3 d3 e" e* Z+ d4 d5 Z, e j- d6 d7 d8 d9 d: d; d< d= d> d? d@ dA dB dC dD dE dF dG dH dI dJ dK dL dM dN dO dP dQ dR dS GdT dU dU e
Z. GdV dW dW e/ Z0 GdX dY dY e Z1 e0 j2 dZ d[ d\ d] d^ d^ d_ d_ d` d` i e0 j3 e j4 da e j5 db e j6 dc e j7 dd e j8 de e j9 df e j: da e j; dg e j< dh e j= da e j> di e j? dd e j@ db e jA dc e jB dS i dj dk ZC dl dm dn do ZD d S)pzr
Collection of postgres-specific extensions, currently including:
* Support for hstore, a key/value type storage
N)*)
Expression)logger)Node)OP)Param) Passthrough)
returns_clone)
QueryCompiler) SelectQuery) UUIDField)compat)adapt)AsIs)register_adapter)register_hstore)JsoncloneFc C s t | t j t | S)N)r r CASTSQL)selfas_type r /postgres_ext.pycast# s r c s: e Z d Z f d d Z d d Z d d Z S) _LookupNodec s) | | _ | | _ t t | j d S)N)nodepartssuperr __init__)r r r ) __class__r r r ( s z_LookupNode.__init__c C s t | | j t | j S)N)typer listr )r r r r
clone_base- s z_LookupNode.clone_basec C s% t t | d d t j t | S)NZparensT)r Clauser r r )r r r r r r 0 s z_LookupNode.cast)__name__
__module__ __qualname__r r# r r r )r r r ' s r c sj e Z d Z d f d d Z d d Z e d d d Z d d
Z d d Z d
d Z S)_JsonLookupBaseFc s& t t | j | | | | _ d S)N)r r( r _as_json)r r r as_json)r r r r 4 s z_JsonLookupBase.__init__c C s% t | | j t | j | j S)N)r! r r" r r) )r r r r r# 8 s z_JsonLookupBase.clone_baseTc C s
| | _ d S)N)r) )r r* r r r r* ; s z_JsonLookupBase.as_jsonc C sP | j d } t | t t f r= t | t j t | St | t j | S)NT) r*
isinstancer" dictr r JSONB_CONTAINSr JSONB_EXISTS)r otherr r r r contains? s z_JsonLookupBase.containsc G s( t | j d t j t t | S)NT)r r* r JSONB_CONTAINS_ANY_KEYr r" )r keysr r r contains_anyE s z _JsonLookupBase.contains_anyc G s( t | j d t j t t | S)NT)r r* r JSONB_CONTAINS_ALL_KEYSr r" )r r2 r r r contains_allK s z _JsonLookupBase.contains_all)
r% r& r' r r# r r* r0 r3 r5 r r )r r r( 3 s r( c @ s" e Z d Z d Z d d Z d S)
JsonLookup json_lookupc C s t | j | j | g | j S)N)r6 r r r) )r valuer r r __getitem__T s zJsonLookup.__getitem__N)r% r& r'
_node_typer9 r r r r r6 Q s r6 c @ s e Z d Z d Z d S)JsonPath json_pathN)r% r& r' r: r r r r r; W s r; c @ s4 e Z d Z d Z e d d Z d d Z d S) ObjectSlice object_slicec C sp t | t r0 | j p d | j p' d g } n3 t | t rK | g } n t t | j d } | | | S)Nr :)r+ slicestartstopintmapsplit)clsr r8 r r r r create] s ! zObjectSlice.createc C s t j | | S)N)r= rG )r r8 r r r r9 g s zObjectSlice.__getitem__N)r% r& r' r: classmethodrG r9 r r r r r= Z s
r= c @ s e Z d Z d d Z d S)_Arrayc C s | | _ | | _ d S)N)fielditems)r rJ rK r r r r k s z_Array.__init__N)r% r& r' r r r r r rI j s rI c C s^ | j j j j j } t | j } | j | t d | | j j d | j j
f S)Nz%s::%s%sz[]) rJ model_classZ_metadatabaseget_connr rK Zpreparer Zget_column_type
dimensions)ZarrconnrK r r r adapt_arrayo s
rQ c s+ e Z d Z d Z d f d d Z S)IndexedFieldMixinZGiSTNc sT | j d d t t | j | | | j rG | p> | j | _ n d | _ d S)NindexT)
setdefaultr rR r rS default_index_type
index_type)r rV argskwargs)r r r r } s
zIndexedFieldMixin.__init__)r% r& r' rU r r r )r r rR z s rR c sj e Z d Z d Z e d f d d Z d d Z d d Z d d
Z d d Z d
d Z
S)
ArrayFieldGIN c sJ | | | | _ | | _ | j j | _ t t | j | | d S)N)_ArrayField__fieldrO Z get_db_fielddb_fieldr rY r )r Z field_classrO rW rX )r r r r s zArrayField.__init__c C s, | j j | } | j d | j 7_ | S)Nz[])r\ __ddl_column__r8 rO )r Z column_typesqlr r r r^ s zArrayField.__ddl_column__c C s> | d k r d St | t t f s1 t | } t | | S)N)r+ r" rI )r r8 r r r db_value s
zArrayField.db_valuec C s t j | | S)N)r= rG )r r8 r r r r9 s zArrayField.__getitem__c G s t | t j t | S)N)r r ACONTAINSr )r rK r r r r0 s zArrayField.containsc G s t | t j t | S)N)r r
ACONTAINS_ANYr )r rK r r r r3 s zArrayField.contains_any) r% r& r' rU Z IntegerFieldr r^ r` r9 r0 r3 r r )r r rY s rY c @ s e Z d Z d Z d S)DateTimeTZField datetime_tzN)r% r& r' r] r r r r rc s rc c @ s e Z d Z d Z d d Z d d Z d d Z d d Z d
d Z d d
Z d d Z
d d Z d d Z d d Z
d d Z d S) HStoreFieldhashc C s t | t j t | S)N)r r HKEYr )r keyr r r r9 s zHStoreField.__getitem__c C s
t j | S)N)fnZakeys)r r r r r2 s zHStoreField.keysc C s
t j | S)N)ri Zavals)r r r r values s zHStoreField.valuesc C s
t j | S)N)ri Zhstore_to_matrix)r r r r rK s zHStoreField.itemsc G s t j | t t | S)N)ri r@ r r" )r rW r r r r@ s zHStoreField.slicec C s t j | | S)N)ri Zexist)r rh r r r exists s zHStoreField.existsc C s t j | | S)N)ri defined)r rh r r r rl s zHStoreField.definedc K s t | t j | S)N)r r HUPDATE)r datar r r update s zHStoreField.updatec G s t j | t t | S)N)ri deleter r" )r r2 r r r rp s zHStoreField.deletec C si t | t r( t | t j t | St | t t f rV t | t j t | St | t j | S)N)
r+ r, r r HCONTAINS_DICTr r" tupleHCONTAINS_KEYS
HCONTAINS_KEY)r r8 r r r r0 s
zHStoreField.containsc G s t | t j t t | S)N)r r HCONTAINS_ANY_KEYr r" )r r2 r r r r3 s zHStoreField.contains_anyN)r% r& r' r] r9 r2 rj rK r@ rk rl ro rp r0 r3 r r r r re s re c sO e Z d Z d Z d f d d Z d d Z d d Z d d
Z S) JSONFieldjsonNc s> t d k r t d | | _ t t | j | | d S)Nz/Your version of psycopg2 does not support JSON.)r Exceptiondumpsr rv r )r ry rW rX )r r r r s zJSONField.__init__c C s6 | d k r | St | t s2 t | d | j S| S)Nry )r+ r ry )r r8 r r r r` s
zJSONField.db_valuec C s t | | g S)N)r6 )r r8 r r r r9 s zJSONField.__getitem__c G s
t | | S)N)r; )r r2 r r r path s zJSONField.path)r% r& r' r] r r` r9 rz r r )r r rv s
rv c @ sL e Z d Z d Z d Z d d Z d d Z d d Z d d
Z d S) BinaryJSONFieldjsonbrZ c C sG t | t t f r. t | t j t | St | t j t | S)N) r+ r" r, r r r- r r. r )r r/ r r r r0 s zBinaryJSONField.containsc C s t | t j t | S)N)r r JSONB_CONTAINED_BYr )r r/ r r r contained_by s z BinaryJSONField.contained_byc G s t | t j t t | S)N)r r r1 r r" )r rK r r r r3 s z BinaryJSONField.contains_anyc G s t | t j t t | S)N)r r r4 r r" )r rK r r r r5 s z BinaryJSONField.contains_allN) r% r& r' r] rU r0 r~ r3 r5 r r r r r{ s r{ c @ s( e Z d Z d Z d Z d d Z d S)
TSVectorFieldtsvectorrZ c C s t | t j t j | S)N)r r TS_MATCHri
to_tsquery)r queryr r r match s zTSVectorField.matchN)r% r& r' r] rU r r r r r r s r c C s% t t j | t j t j | S)N)r ri Z to_tsvectorr r r )rJ r r r r Match s r rg rh rm zH@>rq zH?&rs zH?rt zH?|ru zH||ra zA@>rb zA||r zT@@r- zJB@>r} zJB<@r1 zJB?|r4 zJB?&r. zJB?r z::c s[ e Z d Z d f d d Z d d Z d d Z d d Z f d
d Z S) PostgresqlExtCompilerFc st t t | j | | | } d } x& | D] } t | t r+ | j } q+ W| rp | j j d t d | | S)Nr[ zUSING %s) r r
_create_indexr+ rR rV Znodesinsertr )r rL ZfieldsuniqueZclauserV rJ )r r r r $ s
z#PostgresqlExtCompiler._create_indexc C sW | j | j | | \ } } d d | j D } d | d j | f } | | f S)Nc S s g | ] } t | d q S)r[ )str).0partr r r
<listcomp>4 s z=PostgresqlExtCompiler._parse_object_slice.<locals>.<listcomp>z%s[%s]r? )
parse_noder r join)r r alias_mapconvr_ paramsr r r r _parse_object_slice1 s z)PostgresqlExtCompiler._parse_object_slicec C s | j | j | | \ } } | g } xF | j D]; } | j | | | \ } } | j | | j | q1 W| j r d j | } n9 | d d | d }
} d j d j |
| f } | | f S)Nz->r[ z->>r r )r r r appendextendr) r ) r r r r r_ r Zlookupsr Zpart_sqlZ part_paramsheadtailr r r _parse_json_lookup8 s
z(PostgresqlExtCompiler._parse_json_lookupc C sx | j | j | | \ } } | j r0 d } n d } | j d d j t t | j | j | | j f | f S)Nz#>z#>>z{%s},) r r r) r r rD r r Z
interpolation)r r r r r_ r Zoperandr r r _parse_json_pathJ s &z&PostgresqlExtCompiler._parse_json_pathc s> t t | j } | j d | j d | j d | j | S)Nr> r7 r< )r r
get_parse_mapro r r r )r Z parse_map)r r r r S s
z#PostgresqlExtCompiler.get_parse_map)r% r& r' r r r r r r r )r r r # s
r c s^ e Z d Z e Z f d d Z d d d Z d d d d d Z f d
d Z S) PostgresqlExtDatabasec sG | j d d | _ | j d d | _ t t | j | | d S)Nserver_side_cursorsFr T)popr r r r r )r rW rX )r r r r _ s z PostgresqlExtDatabase.__init__Nc C s, | r | j j d | S| j j S)Nname)rN cursor)r r r r r
get_cursord s z PostgresqlExtDatabase.get_cursorTFc C s t j | | f | p4 | j o4 | j j d } | j | rq | j d t t j } d } n | j } y | j
| | p f WnG t k
r } z' | j r | j
r | j WYd d } ~ Xn X| r | j r | j Wd QRX| S)Nselectr F)r debugr lower
startswithZexception_wrapperr r uuidZuuid1executerx Zget_autocommitZ autorollbackZrollbackZcommit)r r_ r require_commit named_cursorZuse_named_cursorr excr r r execute_sqli s$
z!PostgresqlExtDatabase.execute_sqlc s8 t t | j | | } | j r4 t | d d | S)NZgloballyT)r r _connectr )r rM rX rP )r r r r s z PostgresqlExtDatabase._connect) r% r& r' r Zcompiler_classr r r r r r )r r r \ s r c @ s. e Z d Z e d d Z d d Z d S)ServerSideSelectQueryc C s t | j } | j | S)N)r rL Z_clone_attributes)rF r r r r r clone_from_query s z&ServerSideSelectQuery.clone_from_queryc C s1 | j \ } } | j j | | d d d d S)Nr Fr T)r_ rM r )r r_ r r r r _execute s z ServerSideSelectQuery._executeN)r% r& r' rH r r r r r r r s r rd ztimestamp with time zonerf Zhstorerw r| r z@>z?&?z?|z->z||z&&z@@z<@c c s[ t j | } | j j 6 | j } | | _ x | j D] } | VqA WWd QRXd S)N)r r rM Z transactionr Z_qriterator)Z select_queryr Z query_resultobjr r r
ServerSide s r ZLEFTTc C s& t | t d | | t d | S)Nz%s JOIN LATERALzON %s)r$ r )ZlhsZrhsZ join_typeZ conditionr r r LateralJoin s
r )E__doc__r Zpeeweer r r r r r r r
r r Z psycopg2cffir
register ImportErrorZpsycopg2.extensionsr r r Zpsycopg2.extrasr r r r r r( r6 r; r= objectrI rQ rR ZFieldrY Z
DateTimeFieldrc re rv r{ Z TextFieldr r ro r ZPostgresqlDatabaser r Zregister_fieldsZ register_opsrq rs rt ru rg rm ra rb r r- r} r1 r4 r. r r r r r r r <module> s
) 9+