ÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂ 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)Úparens)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|ƒƒƒSr0)rr/rÚJSONB_CONTAINS_ANY_KEYrr$©rÚkeysrrrÚcontains_anyEs ýz_JsonLookupBase.contains_anycGst| d¡tjtt|ƒƒƒSr0)rr/rÚJSONB_CONTAINS_ALL_KEYSrr$r8rrrÚcontains_allKs ýz_JsonLookupBase.contains_all)F)T)r)r*r+r r&r r/r6r:r<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_typerArrrrr=Qsr=c@seZdZdZdS)ÚJsonPathÚ json_pathN)r)r*r+rBrrrrrCWsrCc@s$eZdZdZedd„ƒZdd„ZdS)ÚObjectSliceÚobject_slicecCsLt|tƒr |jpd|jpdg}n"t|tƒr2|g}ntt| d¡ƒ}|||ƒS)Nrú:)r1ÚsliceÚstartÚstopÚintÚmapÚsplit)Úclsrr@rrrrÚcreate]s zObjectSlice.createcCst ||¡Sr©rErOr?rrrrAgszObjectSlice.__getitem__N)r)r*r+rBÚclassmethodrOrArrrrrEZs rEc@seZdZdd„ZdS)Ú_ArraycCs||_||_dSr)ÚfieldÚitems)rrSrTrrrr ksz_Array.__init__N)r)r*r+r rrrrrRjsrRcCsD|jjjj ¡}t|jƒ}| |¡td||j ¡d|jj fƒS)Nz%s::%s%sú[])rSÚmodel_classÚ_metaÚdatabaseÚget_connrrTZpreparerÚget_column_typeÚ dimensions)ZarrÚconnrTrrrÚadapt_arrayos ýr]cs"eZdZdZd‡fdd„ Z‡ZS)ÚIndexedFieldMixinZGiSTNcs<| dd¡tt|ƒj||Ž|jr2|p,|j|_nd|_dS)NÚindexT)Ú setdefaultrr^r r_Údefault_index_typeÚ index_type)rrbÚargsÚkwargsr!rrr }s zIndexedFieldMixin.__init__)N)r)r*r+rar r,rrr!rr^zsr^csNeZdZdZedf‡fdd„ Zdd„Zdd„Zd d „Zdd„Z d d„Z ‡ZS)Ú ArrayFieldÚGINécs4|||Ž|_||_|j ¡|_tt|ƒj||ŽdSr)Ú_ArrayField__fieldr[Úget_db_fieldÚdb_fieldrrer )rZfield_classr[rcrdr!rrr ‰szArrayField.__init__cCs$|j |¡}|jd|j7_|S)NrU)rhÚ__ddl_column__r@r[)rÚcolumn_typeÚsqlrrrrkszArrayField.__ddl_column__cCs,|dkrdSt|ttfƒs"t|ƒ}t||ƒSr)r1r$rRr?rrrÚdb_value•s zArrayField.db_valuecCst ||¡SrrPr?rrrrAœszArrayField.__getitem__cGst|tjt|ƒƒSr)rrÚ ACONTAINSr©rrTrrrr6ŸszArrayField.containscGst|tjt|ƒƒSr)rrÚ ACONTAINS_ANYrrprrrr:¢szArrayField.contains_any)r)r*r+raÚIntegerFieldr rkrnrAr6r:r,rrr!rre†srec@seZdZdZdS)ÚDateTimeTZFieldÚdatetime_tzN)r)r*r+rjrrrrrs¦srsc@sheZdZdZdd„Zdd„Zdd„Zdd „Zd d„Zdd „Z dd„Z dd„Zdd„Zdd„Z dd„ZdS)ÚHStoreFieldÚhashcCst|tjt|ƒƒSr)rrÚHKEYr©rÚkeyrrrrAszHStoreField.__getitem__cCs t |¡Sr)ÚfnZakeysr%rrrr9°szHStoreField.keyscCs t |¡Sr)rzZavalsr%rrrÚvalues³szHStoreField.valuescCs t |¡Sr)rzZhstore_to_matrixr%rrrrT¶szHStoreField.itemscGst |tt|ƒƒ¡Sr)rzrHrr$)rrcrrrrH¹szHStoreField.slicecCst ||¡Sr)rzZexistrxrrrÚexists¼szHStoreField.existscCst ||¡Sr)rzÚdefinedrxrrrr}¿szHStoreField.definedcKst|tj|ƒSr)rrÚHUPDATE)rÚdatarrrÚupdateÂszHStoreField.updatecGst |tt|ƒƒ¡Sr)rzÚdeleterr$r8rrrrÅszHStoreField.deletecCsJt|tƒrt|tjt|ƒƒSt|ttfƒrzH?&zH?zH?|zH||zA@>zA||zT@@zJB@>zJB<@zJB?|zJB?&zJB?z::)rwr~r‚r„r…r†rorqr“r3rr7r;r4rcsBeZdZd‡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_indexr1r^rbÚnodesÚinsertr)rrVÚfieldsÚuniqueÚclauserbrSr!rrrš$s ÿ z#PostgresqlExtCompiler._create_indexcCs>| |j||¡\}}dd„|jDƒ}d|d |¡f}||fS)NcSsg|]}t|dƒ‘qS)rg)Ústr)Ú.0ÚpartrrrÚ 4sz=PostgresqlExtCompiler._parse_object_slice..z%s[%s]rG)Ú parse_noderrÚjoin)rrÚ alias_mapÚconvrmÚ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¥)rrr¦r§rmr¨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¥rLr rÚ interpolation)rrr¦r§rmr¨ZoperandrrrÚ_parse_json_pathJsz&PostgresqlExtCompiler._parse_json_pathcs(tt|ƒ ¡}|j|j|j|jd|S)N)rFr>rD)rr˜Ú get_parse_mapr€r©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 )rrcrdr!rrr _szPostgresqlExtDatabase.__init__NcCs |r| ¡j|dS| ¡ ¡S)N©Úname)rYÚ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Úselectr·Fr)rÚdebugrµÚlowerÚ startswithÚexception_wrapperrºr ÚuuidÚuuid1Úexecuter‰Úget_autocommitÚautorollbackÚrollbackÚcommit)rrmr¨Úrequire_commitÚnamed_cursorZuse_named_cursorr¹ÚexcrrrÚexecute_sqlis&þ z!PostgresqlExtDatabase.execute_sqlcs*tt|ƒj|f|Ž}|jr&t|dd|S)NT)Zglobally)rr´Ú_connectr)rrXrdr\r!rrrË€szPostgresqlExtDatabase._connect)N)NTF) r)r*r+r˜Úcompiler_classr rºrÊrËr,rrr!rr´\s ÿ r´c@s eZdZedd„ƒZdd„ZdS)ÚServerSideSelectQuerycCst|jƒ}| |¡Sr)rÍrVÚ_clone_attributes)rNr•rrrrÚclone_from_queryˆs z&ServerSideSelectQuery.clone_from_querycCs | ¡\}}|jj||dddS)NFT)rÇrÈ)rmrXrÊ)rrmr¨rrrÚ_executesÿzServerSideSelectQuery._executeN)r)r*r+rQrÏrÐrrrrrÍ‡s rÍztimestamp with time zoneZhstorerˆrr’)rtrvrˆrr’z@>z?&ú?z?|rªz||z&&z@@z<@c csFt |¡}|j ¡(| ¡}||_| ¡D] }|Vq,W5QRXdSr)rÍrÏrXÚtransactionrÂÚ_qrÚiterator)Úselect_queryrZquery_resultÚobjrrrÚ ServerSide¬s r×ÚLEFTTcCst|td|ƒ|td|ƒƒS)Nz%s JOIN LATERALzON %s)r(r)ÚlhsÚrhsÚ join_typeZ conditionrrrÚLateralJoin»s ürÜ)rØT)EÚ__doc__rÀÚpeeweerrrrrrr r rrZpsycopg2cffir ÚregisterÚImportErrorZpsycopg2.extensionsrrrZpsycopg2.extrasrrr¬rrr-r=rCrEÚobjectrRr]r^ÚFieldreÚ DateTimeFieldrsrur‡rŒÚ TextFieldr‘r—r€r˜ÚPostgresqlDatabaser´rÍÚregister_fieldsÚregister_opsr‚r„r…r†rwr~rorqr“r3rr7r;r4rr×rÜrrrrÚsÐ )ñ9+ûñ