ÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂ U S¤ðWˆ4ã @súdZddlZddlTddlmZddlmZddlmZddlmZddlmZdd lmZdd lm Z ddlm Z ddlmZdd lmZzddl mZe ¡Wnek r¸YnXddlmZddlmZddlmZddlmZzddlmZWndZYnXejdddd„ƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGd d!„d!eƒZGd"d#„d#eƒZ d$d%„Z!ee 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(Gd0d1„d1e"e(ƒZ)Gd2d3„d3e"e*ƒZ+d4d5„Z,ej-d6d7d8d9d:d;dd?d@dAdBdCdDdEGdFdG„dGe ƒZ.GdHdI„dIe/ƒZ0GdJdK„dKeƒZ1e0 2dLdMdNdOdPdQœ¡e0 3ej4dRej5dSej6dTej7dUej8dVej9dWej:dRej;dXejdZej?dUej@dSejAdTejBdDi¡d[d\„ZCdad_d`„ZDdS)bzr 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)ÚJsonF)ÚclonecCst|tjt|ƒƒS©N)rrÚCASTÚSQL©ÚselfZas_type©rúI/opt/alt/python38/lib64/python3.8/site-packages/playhouse/postgres_ext.pyÚcast#srcs,eZdZ‡fdd„Zdd„Zdd„Z‡ZS)Ú_LookupNodecs||_||_tt|ƒ ¡dSr)ÚnodeÚpartsÚsuperrÚ__init__)rrr©Ú __class__rrr (sz_LookupNode.__init__cCst|ƒ|jt|jƒƒSr)ÚtyperÚlistr©rrrrÚ clone_base-sz_LookupNode.clone_basecCstt|ddtjt|ƒƒS)NT)Zparens)rÚClauserrrrrrrr0sz_LookupNode.cast)Ú__name__Ú __module__Ú__qualname__r r&rÚ __classcell__rrr!rr'srcsLeZdZd‡fdd„ Zdd„Zeddd„ƒZd d „Zdd„Zd d„Z ‡Z S)Ú_JsonLookupBaseFcstt|ƒ ||¡||_dSr)rr,r Ú_as_json)rrrÚas_jsonr!rrr 4sz_JsonLookupBase.__init__cCst|ƒ|jt|jƒ|jƒSr)r#rr$rr-r%rrrr&8sz_JsonLookupBase.clone_baseTcCs ||_dSr)r-)rr.rrrr.;sz_JsonLookupBase.as_jsoncCs8| d¡}t|ttfƒr*t|tjt|ƒƒSt|tj|ƒS©NT) r.Ú isinstancer$ÚdictrrÚJSONB_CONTAINSrÚJSONB_EXISTS)rÚotherrrrrÚcontains?s z_JsonLookupBase.containscGst| d¡tjtt|ƒƒƒSr/)rr.rÚJSONB_CONTAINS_ANY_KEYrr$©rÚkeysrrrÚcontains_anyEs ýz_JsonLookupBase.contains_anycGst| d¡tjtt|ƒƒƒSr/)rr.rÚJSONB_CONTAINS_ALL_KEYSrr$r7rrrÚcontains_allKs ýz_JsonLookupBase.contains_all)F)T)r(r)r*r r&r r.r5r9r;r+rrr!rr,3sr,c@seZdZdZdd„ZdS)Ú JsonLookupÚjson_lookupcCst|j|j|g|jƒSr)r<rrr-©rÚvaluerrrÚ__getitem__TszJsonLookup.__getitem__N)r(r)r*Ú _node_typer@rrrrr<Qsr<c@seZdZdZdS)ÚJsonPathÚ json_pathN)r(r)r*rArrrrrBWsrBc@s$eZdZdZedd„ƒZdd„ZdS)ÚObjectSliceÚobject_slicecCsLt|tƒr |jpd|jpdg}n"t|tƒr2|g}ntt| d¡ƒ}|||ƒS)Nrú:)r0ÚsliceÚstartÚstopÚintÚmapÚsplit)Úclsrr?rrrrÚcreate]s zObjectSlice.createcCst ||¡Sr©rDrNr>rrrr@gszObjectSlice.__getitem__N)r(r)r*rAÚclassmethodrNr@rrrrrDZs rDc@seZdZdd„ZdS)Ú_ArraycCs||_||_dSr)ÚfieldÚitems)rrRrSrrrr ksz_Array.__init__N)r(r)r*r rrrrrQjsrQcCsD|jjjj ¡}t|jƒ}| |¡td||j ¡d|jj fƒS)Nz%s::%s%sú[])rRÚmodel_classZ_metaÚdatabaseÚget_connrrSZpreparerZget_column_typeÚ dimensions)ZarrÚconnrSrrrÚadapt_arrayos ýrZcs"eZdZdZd‡fdd„ Z‡ZS)ÚIndexedFieldMixinZGiSTNcs<| dd¡tt|ƒj||Ž|jr2|p,|j|_nd|_dS)NÚindexT)Ú setdefaultrr[r r\Údefault_index_typeÚ index_type)rr_ÚargsÚkwargsr!rrr }s zIndexedFieldMixin.__init__)N)r(r)r*r^r r+rrr!rr[zsr[csNeZdZdZedf‡fdd„ Zdd„Zdd„Zd d „Zdd„Z d d„Z ‡ZS)Ú ArrayFieldÚGINécs4|||Ž|_||_|j ¡|_tt|ƒj||ŽdSr)Ú_ArrayField__fieldrXZget_db_fieldÚdb_fieldrrbr )rZfield_classrXr`rar!rrr ‰szArrayField.__init__cCs$|j |¡}|jd|j7_|S)NrT)reÚ__ddl_column__r?rX)rZcolumn_typeÚsqlrrrrgszArrayField.__ddl_column__cCs,|dkrdSt|ttfƒs"t|ƒ}t||ƒSr)r0r$rQr>rrrÚdb_value•s zArrayField.db_valuecCst ||¡SrrOr>rrrr@œszArrayField.__getitem__cGst|tjt|ƒƒSr)rrÚ ACONTAINSr©rrSrrrr5ŸszArrayField.containscGst|tjt|ƒƒSr)rrÚ ACONTAINS_ANYrrkrrrr9¢szArrayField.contains_any)r(r)r*r^ZIntegerFieldr rgrir@r5r9r+rrr!rrb†srbc@seZdZdZdS)ÚDateTimeTZFieldÚdatetime_tzN)r(r)r*rfrrrrrm¦srmc@sheZdZdZdd„Zdd„Zdd„Zdd „Zd d„Zdd „Z dd„Z dd„Zdd„Zdd„Z dd„ZdS)ÚHStoreFieldÚhashcCst|tjt|ƒƒSr)rrÚHKEYr©rÚkeyrrrr@szHStoreField.__getitem__cCs t |¡Sr)ÚfnZakeysr%rrrr8°szHStoreField.keyscCs t |¡Sr)rtZavalsr%rrrÚvalues³szHStoreField.valuescCs t |¡Sr)rtZhstore_to_matrixr%rrrrS¶szHStoreField.itemscGst |tt|ƒƒ¡Sr)rtrGrr$)rr`rrrrG¹szHStoreField.slicecCst ||¡Sr)rtZexistrrrrrÚexists¼szHStoreField.existscCst ||¡Sr)rtÚdefinedrrrrrrw¿szHStoreField.definedcKst|tj|ƒSr)rrÚHUPDATE)rÚdatarrrÚupdateÂszHStoreField.updatecGst |tt|ƒƒ¡Sr)rtÚdeleterr$r7rrrr{ÅszHStoreField.deletecCsJt|tƒrt|tjt|ƒƒSt|ttfƒrrrrr5Ès zHStoreField.containscGst|tjtt|ƒƒƒSr)rrÚHCONTAINS_ANY_KEYrr$r7rrrr9ÏszHStoreField.contains_anyN)r(r)r*rfr@r8rurSrGrvrwrzr{r5r9rrrrroªsrocs:eZdZdZd‡fdd„ Zdd„Zdd„Zd d „Z‡ZS)Ú JSONFieldÚjsonNcs,tdkrtdƒ‚||_tt|ƒj||ŽdS)Nz/Your version of psycopg2 does not support JSON.)rÚ ExceptionÚdumpsrrr )rr„r`rar!rrr ÖszJSONField.__init__cCs(|dkr|St|tƒs$t||jdS|S)N)r„)r0rr„r>rrrriÝs zJSONField.db_valuecCst||gƒSr)r<r>rrrr@äszJSONField.__getitem__cGs t||ƒSr)rBr7rrrÚpathçszJSONField.path)N) r(r)r*rfr rir@r…r+rrr!rrÓs rc@s4eZdZdZdZdd„Zdd„Zdd„Zd d „ZdS)ÚBinaryJSONFieldÚjsonbrccCs2t|ttfƒr t|tjt|ƒƒSt|tjt|ƒƒSr) r0r$r1rrr2rr3r©rr4rrrr5ïszBinaryJSONField.containscCst|tjt|ƒƒSr)rrÚJSONB_CONTAINED_BYrrˆrrrÚcontained_byôszBinaryJSONField.contained_bycGst|tjtt|ƒƒƒSr)rrr6rr$rkrrrr9÷s ýzBinaryJSONField.contains_anycGst|tjtt|ƒƒƒSr)rrr:rr$rkrrrr;ýs ýzBinaryJSONField.contains_allN) r(r)r*rfr^r5rŠr9r;rrrrr†ësr†c@seZdZdZdZdd„ZdS)Ú TSVectorFieldÚtsvectorrccCst|tjt |¡ƒSr)rrÚTS_MATCHrtÚ to_tsquery)rÚqueryrrrÚmatchszTSVectorField.matchN)r(r)r*rfr^rrrrrr‹sr‹cCstt |¡tjt |¡ƒSr)rrtZto_tsvectorrrrŽ)rRrrrrÚMatchsr‘rszH@>zH?&zH?zH?|zH||zA@>zA||zT@@zJB@>zJB<@zJB?|zJB?&zJB?z::)rqrxr|r~rr€rjrlrr2r‰r6r:r3rcsBeZdZd‡fdd„ Zdd„Zdd„Zdd „Z‡fd d„Z‡ZS) ÚPostgresqlExtCompilerFcsPtt|ƒ |||¡}d}|D]}t|tƒr|j}q|rL|j dtd|ƒ¡|S)NéÿÿÿÿzUSING %s) rr’Ú _create_indexr0r[r_ZnodesÚinsertr)rrUZfieldsÚuniqueZclauser_rRr!rrr”$s ÿ z#PostgresqlExtCompiler._create_indexcCs>| |j||¡\}}dd„|jDƒ}d|d |¡f}||fS)NcSsg|]}t|dƒ‘qS)rd)Ústr)Ú.0ÚpartrrrÚ 4sz=PostgresqlExtCompiler._parse_object_slice..z%s[%s]rF)Ú parse_noderrÚjoin)rrÚ alias_mapÚconvrhÚparamsrrrrÚ_parse_object_slice1sz)PostgresqlExtCompiler._parse_object_slicecCs| |j||¡\}}|g}|jD]*}| |||¡\}} | |¡| | ¡q |jr^d |¡}n*|dd…|d} }d d | ¡|f¡}||fS)Nú->r“z->>)r›rrÚappendÚextendr-rœ)rrrržrhrŸZlookupsr™Zpart_sqlZpart_paramsÚheadÚtailrrrÚ_parse_json_lookup8s ÿ z(PostgresqlExtCompiler._parse_json_lookupcCsT| |j||¡\}}|jr d}nd}| dd tt|jƒ¡¡| ||jf¡|fS)Nz#>z#>>z{%s}ú,) r›rr-r¢rœrKr—rZ interpolation)rrrržrhrŸZoperandrrrÚ_parse_json_pathJsz&PostgresqlExtCompiler._parse_json_pathcs(tt|ƒ ¡}|j|j|j|jd|S)N)rEr=rC)rr’Ú get_parse_maprzr r¦r¨)rZ parse_mapr!rrr©Ssýz#PostgresqlExtCompiler.get_parse_map)F) r(r)r*r”r r¦r¨r©r+rrr!rr’#s r’cs@eZdZeZ‡fdd„Zddd„Zd dd „Z‡fd d„Z‡Z S)ÚPostgresqlExtDatabasecs2| dd¡|_| dd¡|_tt|ƒj||ŽdS)NÚserver_side_cursorsFrT)Úpopr«rrrªr )rr`rar!rrr _szPostgresqlExtDatabase.__init__NcCs |r| ¡j|dS| ¡ ¡S)N©Úname)rWÚcursor)rr®rrrÚ get_cursordsz PostgresqlExtDatabase.get_cursorTFcCsÈt ||f¡|p$|jo$| ¡ d¡}| ¡|rN|jtt ¡ƒd}d}n| ¡}z| ||pdd¡Wn:tk r¤}z| ¡r’|j r’| ¡‚W5d}~XYnX|rº| ¡rº| ¡W5QRX|S)NÚselectrFr)rÚdebugr«ÚlowerÚ startswithZexception_wrapperr°r—ÚuuidZuuid1ÚexecuterƒZget_autocommitZautorollbackZrollbackZcommit)rrhrŸÚrequire_commitÚnamed_cursorZuse_named_cursorr¯ÚexcrrrÚexecute_sqlis&þ z!PostgresqlExtDatabase.execute_sqlcs*tt|ƒj|f|Ž}|jr&t|dd|S)NT)Zglobally)rrªÚ_connectr)rrVrarYr!rrr»€szPostgresqlExtDatabase._connect)N)NTF) r(r)r*r’Zcompiler_classr r°rºr»r+rrr!rrª\s ÿ rªc@s eZdZedd„ƒZdd„ZdS)ÚServerSideSelectQuerycCst|jƒ}| |¡Sr)r¼rUZ_clone_attributes)rMrrrrrÚclone_from_queryˆs z&ServerSideSelectQuery.clone_from_querycCs | ¡\}}|jj||dddS)NFT)r·r¸)rhrVrº)rrhrŸrrrÚ_executesÿzServerSideSelectQuery._executeN)r(r)r*rPr½r¾rrrrr¼‡s r¼ztimestamp with time zoneZhstorer‚r‡rŒ)rnrpr‚r‡rŒz@>z?&ú?z?|r¡z||z&&z@@z<@c csFt |¡}|j ¡(| ¡}||_| ¡D] }|Vq,W5QRXdSr)r¼r½rVZtransactionr¶Z_qrÚiterator)Zselect_queryrZquery_resultÚobjrrrÚ ServerSide¬s rÂÚLEFTTcCst|td|ƒ|td|ƒƒS)Nz%s JOIN LATERALzON %s)r'r)ZlhsZrhsZ join_typeZ conditionrrrÚLateralJoin»s ürÄ)rÃT)EÚ__doc__rµZpeeweerrrrrrr r rrZpsycopg2cffir ÚregisterÚImportErrorZpsycopg2.extensionsrrrZpsycopg2.extrasrrr£rrr,r<rBrDÚobjectrQrZr[ZFieldrbZ DateTimeFieldrmrorr†Z TextFieldr‹r‘rzr’ZPostgresqlDatabaserªr¼Zregister_fieldsZregister_opsr|r~rr€rqrxrjrlrr2r‰r6r:r3rrÂrÄrrrrÚsÐ )ñ9+ûñ