o Tj@s<ddlZddlZddlZddlZddlZddlmZddlmZddlm Z ddl Tddl m Z ddl m Z ddl m Z dd l mZzdd lmZWn eyUdZYnwzdd lmZWn eyidZYnwegd Zid ededededededejdejdejdedededededejdejdeeeeejej dZ!Gdd 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(Gd+d,d,e%Z)d-d.Z*ed/d0Z+Gd1d2d2e+Z,Gd3d4d4e"Z-dAd5d6Z.dAd7d8Z/dBd;d<Z0d=d>Z1d?d@Z2dS)CN) OrderedDict) namedtuple)isclass)*) _StringField)_query_val_transform) CommaNodeList)make_snake_case) postgres_ext)CockroachDatabase)andasassertbreakclasscontinuedefdelelifelseexceptexecfinallyforfromglobalifimportinislambdanotorpassprintraisereturntrywhilewithyieldAUTOBIGAUTOBIGINTBLOBBOOLCHARDATEDATETIMEDECIMALDOUBLEFLOATINTSMALLINTTEXTTIMEUUIDUUIDB)VARCHARJSONJSONB TIMESTAMPTZINTERVALc@s eZdZdS) UnknownFieldN)__name__ __module__ __qualname__rErEW/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/playhouse/reflection.pyrA:srAc@sdeZdZdZeefZ  dddZddZdd Z d d Z d d Z ddZ  dddZ ddZdS)Columnz1 Store metadata about a database column. FNc CsR||_||_||_||_||_||_||_||_| |_| |_ d|_ d|_ d|_ dSN) name field_classraw_column_typenullable primary_key column_nameindexuniquedefaultextra_parameters rel_model related_nameto_field) selfrIrJrKrLrMrNrOrPrQrRrErErF__init__Ds zColumn.__init__cs.gd}dfdd|D}dj|fS)N)rJrKrLrMrN, c3s"|] }d|t|fVqdSz%s=%sN)getattr).0attrrVrErF ^s  z"Column.__repr__..zColumn(%s, %s))joinrI)rVattrs keyword_argsrEr]rF__repr__Ws zColumn.__repr__cCs6i}|jdur ||j|jrd|d<|jtus|j|jkr&d|j|d<|jr3t|jt s3d|d<|j durCd|j dd|d <| r|j |d <|jrVd|j|d <|jr`d|j|d <t|d d}dD]}|rrt||dnd}|r|dvrd|||<qh|s|jrd|d<|S|jr| sd|d<|S)NTnull'%s'rNrMz[SQL("DEFAULT %s")]"z\" constraintsmodelfieldbackref foreign_key on_delete on_updatez NO ACTIONRESTRICTTruerPrO)rRupdaterLrJForeignKeyFieldrIrNrM issubclass AutoFieldrQreplaceis_foreign_keyrSrUrTrZis_primary_keyrPrO)rVparamsfkr\valuerErErFget_field_parameterscs@        zColumn.get_field_parameterscCs|jtup|jSrH)rJrtrMr]rErErFrwzColumn.is_primary_keycCs |jtuSrH)rJrrr]rErErFrv zColumn.is_foreign_keycCs|jtuo |jdkSNz'self')rJrrrSr]rErErFis_self_referential_fks zColumn.is_self_referential_fkcCsP||_t|_d|_|j|jkrd|_n||j|_|r|jpd|_|p$d|_ dSr~) rjrrrJrR dest_tabletablerSrIrUrT)rVrj model_namesdestrTrErErFset_foreign_keys  zColumn.set_foreign_keycCsi}|D]\}}t|rt|tr|j}|||<qdddt|D}d|j|j j|f}|j t urAd||j f}|S)NrXcss |] \}}d||fVqdSrYrE)r[kvrErErFr^sz#Column.get_field..z %s = %s(%s)z%s # %s) r{itemsrrsFieldrBr_sortedrIrJrArK)rV field_paramskeyrz param_strrhrErErF get_fields     zColumn.get_field)FNFFNN)NN)rBrCrD__doc__ IntegerFieldrtprimary_key_typesrWrbr{rwrvrrrrErErErFrG>s  ' rGc@speZdZiZdZedZddZddZ ddd Z dd d Z d d Z ddZ dddZdddZdddZdS)Metadataz*^\s*[\w ]+\(\s*(\d+)\s*(?:,\s*(\d+))?\s*\)cCs||_d|_t|j|_dS)NF)databaserequires_extensiondict column_map)rVrrErErFrWszMetadata.__init__cG|j||SrH)r execute_sql)rVsqlrxrErErFexecutezMetadata.executeNc Cstdd|j||D}|||\}}dd|D}t|dkrC|d}||jrC||tur9t||<n ||t urCt ||<t}|D]7\} } || } | | | j } | | } || | }|rpt| pjifi|} t| | | j| j| j| | | d|| <qJ|S)Ncss|]}|j|fVqdSrHrI)r[metadatarErErFr^s  z'Metadata.get_columns..cSg|] \}}|jr|qSrErM)r[rImrErErF sz(Metadata.get_columns..r)rJrKrLrMrNrQrR)rr get_columnsget_column_typesrlenidentityrrtBigIntegerField BigAutoField_clean_defaultrQget_extract_type_paramsrrG data_typercrM)rVrschemar column_types extra_paramspk_namespkcolumnsrI column_datarJrQextra type_paramsrErErFrs>          zMetadata.get_columnscCstrH)NotImplementedErrorrVrrrErErFrszMetadata.get_column_typescCs|j|jpd}|durdS|\}}t|tr&t|dkr&dt|iSt|trB|durDt|t|fdkrFt|t|dSdSdSdS)Nr max_length) ) max_digitsdecimal_places)re_type_paramsmatch full_typegroupsrs CharFieldint DecimalField)rVrJrrp1p2rErErFrs  zMetadata._extract_type_paramscCsR|dus|ttfvs|dkrdSt|tr%t|tr%|ds%d|}|p(dS)Nrc'rdz'')rtrlowerrsr isinstancestr startswith)rVrJrQrErErFrs  zMetadata._clean_defaultcCrrH)rget_foreign_keysrrErErFrrzMetadata.get_foreign_keyscCrrH)rget_primary_keysrrErErFrrzMetadata.get_primary_keyscCrrH)r get_indexesrrErErFr rzMetadata.get_indexesrH)rBrCrDrextension_importrecompilerrWrrrrrrrrrErErErFrs   ( rc seZdZidedededededede de d e d e d e d e d e dedededeZeeeeee e ee e eed ZdZfddZdddZZS)PostgresqlMetadataiiiii:iZii;iii ) iiiiiiii[iii z$from playhouse.postgres_ext import *cstt||tdurT|d}|}|D]0\}}}|dkr'tj|j|<q|dkr2tj|j|<q|dkr=tj |j|<q|dkrGtj |j|<q|j D] }tj |j|<qKdSdS)N9select oid, typname, format_type(oid, NULL) from pg_type;jsonjsonbhstoretsvector) superrrWr rfetchall JSONFieldrBinaryJSONField HStoreField TSVectorField array_types ArrayFieldrVrcursorresultsoidtypnameformatted_type __class__rErFrW3s$   zPostgresqlMetadata.__init__Nc Csi}i}tdurttjtjtjtjtjfnt}|d||}|D]$\}}|j |t ||<|||vr:d|_ ||j vrHd|j |i||<q$||fS)Na8SELECT a.attname, a.atttypid FROM pg_catalog.pg_attribute AS a INNER JOIN pg_catalog.pg_class AS t ON a.attrelid = t.oid INNER JOIN pg_catalog.pg_namespace AS n ON t.relnamespace = n.oid WHERE t.relname = %s AND n.nspname = COALESCE(%s, current_schema()) AND a.attnum > 0 AND NOT a.attisdropped ORDER BY a.attnumTrJ)r setrrrrrrrrrrArr) rVrrrrextension_typesrrIrrErErFrIs0   z#PostgresqlMetadata.get_column_typesrH)rBrCrD BooleanField BlobFieldrSmallIntegerFieldr TextField FloatField DoubleFieldr DateField DateTimeField TimeFieldr UUIDFieldrrrrWr __classcell__rErErrFrsh      rc@s<eZdZejZeed<ejZeed<dZddZ dS)CockroachDBMetadatarrz#from playhouse.cockroachdb import *cCspt||d|_tdur4|d}|}|D]\}}}|dkr'tj|j|<q|jD] }tj |j|<q+dSdS)NTrr) rrWrr rrrrrrrrErErFrWss     zCockroachDBMetadata.__init__N) rBrCrDrrcopyrrrrWrErErErFrjs   rc @seZdZidededededededede d ed e d e d e d ede dede de ee e eeee e eee d ZdddZdS) MySQLMetadatabigintbinaryblobchardatedatetimedecimaldoubleenumfloatrintegerlongbloblongtext mediumblob mediumint mediumtext) numericsmallinttexttime timestamptinyblobtinyinttinytext varbinaryvarcharyearNcCsli}|d||}|D]$\}}}|dkrt||<q |dkr&t||<q |j|t||<q |ifS)NzSELECT column_name, data_type, column_type FROM information_schema.columns WHERE table_name = %s AND table_schema = COALESCE(%s, DATABASE())z tinyint(1)r)rrrrrrrrA)rVrrrrrIr column_typerErErFrs     zMySQLMetadata.get_column_typesrH)rBrCrDrrrrrrrrrrrrrrrErErErFrsb     rc@seZdZidededededededede d e d e d e d e d ede de de de e e e edZdZdZdjeedZdZddZdddZdS)SqliteMetadatarrboolbooleanrrrrrr zinteger unsignedrlongrreal smallintegerr)zsmallint unsignedrrrz (?:["\[\(]+)?z (?:["\]\)]+)?zi(?:FOREIGN KEY\s*)?{begin}(.+?){end}\s+(?:.+\s+)?references\s+{begin}(.+?){end}\s*\(["|\[]?(.+?)["|\]]?\))beginendz&^\s*(?:var)?char\s*\(\s*(\d+)\s*\)\s*$cCsf|}||jvr|j|}|St|j|rt}|Stdd|}|dkr*t}|S|j|t }|S)Nz\(.+\)r) rrrsearch re_varcharrsub BareFieldrrA)rVrrKrJrErErF_map_cols   zSqliteMetadata._map_colNcCs4i}|j|}|D] }||j||j<q |ifSrH)rrr'rrI)rVrrrrcolumnrErErFrs  zSqliteMetadata.get_column_typesrH)rBrCrDrrrrrrrrrrrrr!r"formatre_foreign_keyr$r'rrErErErFrsb     rcCsB|jD] }|dus ||vrdSq|jrtd|jtjrdSdS)NFz \)\s*WHERE\sT)rrrr#I)rO column_namesr(rErErF_is_index_usables r-_DatabaseMetadata)r primary_keys foreign_keysrindexesc@s$eZdZddZddZddZdS)DatabaseMetadatacs`g}jD]&}t|jdkr-t|tjr-fdd|jD}|||jfq|S)Nrcsg|] }j|jqSrE)rrIr[r(rVrrErFr sz9DatabaseMetadata.multi_column_indexes..)r1rrr-rappendrP)rVraccumrO field_namesrEr4rFmulti_column_indexess z%DatabaseMetadata.multi_column_indexescCsJi}|j|D]}t|jdkr"t|t|j|r"|j||jd<q|S)Nrr)r1rrr-rrP)rVrr6rOrErErFcolumn_indexesszDatabaseMetadata.column_indexescsfddjDS)Ncs$g|]}t|tjs|qSrE)r-rr)r[rOr4rErFrs z6DatabaseMetadata.unhandled_indexes..)r1r4rEr4rFunhandled_indexessz"DatabaseMetadata.unhandled_indexesN)rBrCrDr8r9r:rErErErFr2s r2c@seZdZeegZdddZddZedddZ dd Z d d Z d d Z ddZ dddZdddZ  dddZ   dddZdS) IntrospectorNcCs||_||_dSrH)rr)rVrrrErErFrW"s zIntrospector.__init__cCs d|jjS)Nzrrr]rErErFrb& zIntrospector.__repr__cCst|tr|jdurtd|j}trt|trt|}n$t|tr't|}nt|tr1t |}nt|t r;t |}ntd||||dS)Nz)Cannot introspect an uninitialized Proxy.z"Introspection not supported for %rr) rProxyobj ValueErrorr rPostgresqlDatabaser MySQLDatabaserSqliteDatabaser)clsrrrrErErF from_database)s           zIntrospector.from_databasecCs t|jjSrH)typerrr]rErErFget_database_class;r=zIntrospector.get_database_classcCs |jjjSrHr<r]rErErFget_database_name>r}zIntrospector.get_database_namecCs |jjjSrH)rrconnect_paramsr]rErErFget_database_kwargsAr}z Introspector.get_database_kwargscCs|jjr d|jjSdS)N r)rrrr]rErErFget_additional_importsDs z#Introspector.get_additional_importsTcCsL|rt|}tdd|}ddd|dD}|ds$d|}|S)N[^\w]+rcs|]}|VqdSrH)title)r[r%rErErFr^Mz/Introspector.make_model_name.._rT)r rr%r_splitisalpha)rVr snake_caserg model_namerErErFmake_model_nameIs zIntrospector.make_model_nameFcCsp|}|r t|}|}|rtdd|p|}tdd|}|tvr(|d7}t|r6|dr6d|}|S)Nz_id$rrNrRr)stripr rrr%RESERVED_WORDSrisdigit)rVr(rvrVrErErFmake_column_nameRszIntrospector.make_column_namec! s|jjj|jd}|r|jjj|jd}|dd|Dt|}dur/fdd|D}t|}i} i} i} i} i} |D]}|j||j}|j||j}z |j ||j| |<Wnt ys}z g| |<WYd}~nEd}~wwg}| |D]}|j |vr| |qzt d||j|j fqz|| |<dur| |D]}|j |vr| |j ||j q|||| |<tdd|D}tdd| |D}|D]+\}}|rtd d |}n ||||v|}|}|d r||vr|}||_q|D]#}t|jd kr$t|t|r$|jd }|j||_d||_qdd|D| |<|| |<|| |<q?i}dd}|D]?}t}t| ||dD]0}z | ||j}Wn tyeYqOw|j }||vryd||jf||<qO||qOqB|D]6}| |D].}| |j |j}z | |j |j!} Wn tyd} Ynw|j"|| | |#|dqqt$| | | | | S)Nr>cSsg|]}|jqSrEr)r[viewrErErFrjsz+Introspector.introspect..csg|]}|vr|qSrErE)r[r table_namesrErFrnsz6Ignoring foreign-key %s.%s, "%s" was not introspected.csrOrH)r)r[rNrErErFr^sz*Introspector.introspect..css|]}|jVqdSrHr()r[fk_colrErErFr^srNrR_idrrTcSrrEr)r[rIr(rErErFrs  cSs|jSrHr`)rjrErErFsz)Introspector.introspect..)rz %s_%s_set)rjrrrT)%rr get_tablesr get_viewsextendrrrrrArr5warningswarnr(addrXrrr%r\rendswithrIrrr-rPrOrKeyErrorr dest_columnrrr2)!rVr_literal_column_names include_viewsrVtablesviews schema_tables table_setrr/r0rr1r table_indexes table_columnsexckeeprjlower_col_namesfkscol_namer(new_name lower_namerO related_namessort_fnmodels_referencedrsrcrrEr^rF introspectds                  zIntrospector.introspectc sp|||i}GfdddttfddtjD] \}}||vr5||q(|S)NcseZdZGfdddZdS)z/Introspector.generate_models..BaseModelcseZdZjjZjZdS)z4Introspector.generate_models..BaseModel.MetaN)rBrCrDrrrrEr]rErFMetas rN)rBrCrDrrEr]rErF BaseModelsrc s jD] }|j}||vr*|kr*|vr%td|fq ||q gj}|D] \}}|jrC|j q6  }Gfddd}d} s`d|_nt dkrvt fdd|D|_d} d|i} |D]\}}|j} | turrt} n| turt} ||jd } |jdur| tur| |j|jr| r| turt} d| d <n |jr| turd| d <|r|rd | d <n|jj} | |vr|| | d <nt} | | d <|jr|j| d<d|f| d<dD]}t|j|d}|r|dvr|| |<q|jdur!td|j}|g| d<|sC||vr?||r5d| d<n|s>d| d<nd| d<| di| | |j <q~z+z t t!f| |<Wnt"yjshYnwWvrx#dSdSvr#ww)Nz0Possible reference cycle found between %s and %scseZdZZZdS)zAIntrospector.generate_models.._create_model..MetaN)rBrCrDr1 table_namerE)r8rrErFrsrFrcsg|] \}}|vr|jqSrEr)r[colrh)r/rErFrs zGIntrospector.generate_models.._create_model..T)rNrcrMrVrgrel_model_namerhz %s_%s_relrirkrnz DEFAULT %srfrPrOrE)$rir0rrgrhrrrMr5rIr8r9r CompositeKeyrJrrr&rArLrRrqrtrrvrrjDeferredForeignKeyrUrZrQSQLrwrGrrAremove)rmodelsrjrrrNr(r9r composite_keyr` FieldClassrxrr\rz constraint)r _create_model bare_fieldsrpending skip_invalid)r8r/rrFrs                          z3Introspector.generate_models.._create_model)rModelrrrr) rVrr_rmrrnrrrgrE)rrrrrrVrrFgenerate_modelsse zIntrospector.generate_modelsrH)T)FT)NFFT)FNFFF)rBrCrDrtr pk_classesrWrb classmethodrFrHrIrKrMrXr\rrrErErErFr;s(      r;cCstj||d}|S)Nr>)r;rFr)rr introspectorrErErFrisrcKstj||d}|jdi|S)Nr>rE)r;rFr)rroptionsrrErErFrnsrTFc Cst|jj|jjD]@}d|j|jfg}|jr|dn|r1|jr)|dn|jr1|dt |t rC|d|j j |j jftd|q |r|j}|sVdStd|D]c}dg}|jj}|jd d d &|t|j|jdur|d ||j|\}} Wdn1swY|ttt| } || d d|jr|dtd|q\dSdS)Nz %s %sz PKz UNIQUEz INDEXz FK: %s.%srz index(es)z %sz"")paramquotez WHERE re)r$_metarI sorted_fields field_typerMr5rPrOrrrrSrB rel_fieldr_fields_to_indexrget_sql_context scope_valuesrr _expressions_whereliteralquerytuplemaprru_unique) rgr1inline_indexesrhparts index_listrOctxrrxcleanrErErF print_modelssJ             rc Cs|j\}}|jjjdkr||jjjd}td|}| \}}}d dd| dD}d |||f }|t tt|S)Nrz^(.+?\()(.+)(\).*)z, css|]}d|VqdS)z %sNrEr3rErErFr^rQz get_table_sql..rXrL)_schema _create_tablerrrrrurrrr_rTrYrrr) rgrrx match_objcreaterrindentedrrErErF get_table_sqls rcCstt|dSrH)r$r)rgrErErFprint_table_sqlr|rrH)TF)3rrruuidrg collectionsrrinspectrpeeweerrrr playhouser ImportErrorplayhouse.cockroachdbr rrZrbytesrrrDecimalrrr:r timedelta FieldTypeMapobjectrArGrrrrrr-r.r2r;rrrrrrErErErFs               wZ[4:  L  (