ÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂ U S¤ðWUUã @sžzddlmZWnek r(eZYnXddlmZddlZddlTzddlmZWn>ek r”zddl mZWnek rŽdZYnXYnXzddl mZWnek r¾dZYnXeddd d ddd ddddddddddddddddddd d!d"d#d$d%gƒZ Gd&d'„d'eƒZGd(d)„d)eƒZGd*d+„d+eƒZGd,d-„d-eƒZGd.d/„d/eƒZGd0d1„d1eƒZed2d3ƒZGd4d5„d5eƒZGd6d7„d7eƒZd:d8d9„ZdS);é)ÚOrderedDict)Ú namedtupleN)Ú*)Ú FIELD_TYPE)Úpostgres_extÚandÚasÚassertÚbreakÚclassÚcontinueÚdefÚdelÚelifÚelseÚexceptÚexecÚfinallyÚforÚfromÚglobalÚifÚimportÚinÚisÚlambdaÚnotÚorÚpassÚprintÚraiseÚreturnÚtryÚwhileÚwithÚyieldc@seZdZdS)ÚUnknownFieldN)Ú__name__Ú __module__Ú__qualname__©r*r*úG/opt/alt/python38/lib64/python3.8/site-packages/playhouse/reflection.pyr&sr&c@s\eZdZdZeefZddd„Zdd„Zdd „Z d d„Z dd „Zdd„Zddd„Z dd„ZdS)ÚColumnz1 Store metadata about a database column. FNc CsF||_||_||_||_||_||_||_||_d|_d|_ d|_ dS©N)ÚnameÚfield_classÚraw_column_typeÚnullableÚprimary_keyÚ db_columnÚindexÚuniqueÚ rel_modelÚrelated_nameÚto_field) Úselfr.r/r0r1r2r3r4r5r*r*r+Ú__init__'szColumn.__init__cs4dddddg}d ‡fdd„|Dƒ¡}d ˆj|fS) Nr/r0r1r2r3ú, c3s |]}d|tˆ|ƒfVqdS©z%s=%sN)Úgetattr)Ú.0Úattr©r9r*r+Ú >sÿz"Column.__repr__..zColumn(%s, %s))Újoinr.)r9ÚattrsZkeyword_argsr*r@r+Ú__repr__7sûþzColumn.__repr__cCsºi}|jrd|d<|jtks(|j|jkr6d|j|d<|jrN|jtk rNd|d<| ¡rˆ|j|d<|j rtd|j |d<|j rˆd|j |d<| ¡s¶|jr d |d <n|j r¶| ¡s¶d |d<|S)NTÚnullz'%s'r3r2r6r8r7ÚTruer5r4)r1r/ÚForeignKeyFieldr.r3r2ÚPrimaryKeyFieldÚis_foreign_keyr6r8r7Úis_primary_keyr5r4)r9Úparamsr*r*r+Úget_field_parametersCs& zColumn.get_field_parameterscCs|jtkp|jSr-)r/rHr2r@r*r*r+rJ_szColumn.is_primary_keycCs |jtkSr-)r/rGr@r*r*r+rIbszColumn.is_foreign_keycCs|jtko|jdkS©Nz'self')r/rGr6r@r*r*r+Úis_self_referential_fkes ÿzColumn.is_self_referential_fkcCsJ||_t|_|j|jkr d|_n||j|_|r6|jp8d|_|pBd|_dSrM) Úforeign_keyrGr/Ú dest_tableÚtabler6r.r8r7)r9rOÚmodel_namesÚdestr7r*r*r+Úset_foreign_keyiszColumn.set_foreign_keycCsT| ¡}d dd„t| ¡ƒDƒ¡}d|j|jj|f}|jtkrPd||jf}|S)Nr;css|]\}}d||fVqdSr<r*)r>ÚkÚvr*r*r+rAwsÿz#Column.get_field..z%s = %s(%s)z%s # %s) rLrBÚsortedÚitemsr.r/r'r&r0)r9Zfield_paramsZ param_strÚfieldr*r*r+Ú get_fieldts ÿý zColumn.get_field)FNFF)NN)r'r(r)Ú__doc__ÚIntegerFieldrHZprimary_key_typesr:rDrLrJrIrNrTrZr*r*r*r+r,!s ÿ ÿ r,c@sVeZdZiZdZdd„Zdd„Zddd„Zdd d „Zddd„Z dd d„Z ddd„ZdS)ÚMetadataÚcCs||_d|_dS)NF)ÚdatabaseÚrequires_extension)r9r_r*r*r+r:ˆszMetadata.__init__cGs|j ||¡Sr-)r_Úexecute_sql)r9ÚsqlrKr*r*r+ÚexecuteŒszMetadata.executeNc Csštdd„|j ||¡Dƒƒ}| ||¡}| ||¡}t|ƒdkr\|d}||tkr\t||<tƒ}| ¡D]*\}} t |||| j | j| j|d||<qj|S)Ncss|]}|j|fVqdSr-©r.)r>Úmetadatar*r*r+rAsÿz'Metadata.get_columns..ér)r/r0r1r2r3) rr_Úget_columnsÚget_column_typesÚget_primary_keysÚlenr\rHrXr,Ú data_typerEr2) r9rQÚschemareÚcolumn_typesZpk_namesÚpkÚcolumnsr.Úcolumn_datar*r*r+rgs(þúzMetadata.get_columnscCst‚dSr-)ÚNotImplementedError©r9rQrlr*r*r+rhªszMetadata.get_column_typescCs|j ||¡Sr-)r_Úget_foreign_keysrrr*r*r+rsszMetadata.get_foreign_keyscCs|j ||¡Sr-)r_rirrr*r*r+ri°szMetadata.get_primary_keyscCs|j ||¡Sr-)r_Úget_indexesrrr*r*r+rt³szMetadata.get_indexes)N)N)N)N)N)r'r(r)Ú column_mapÚextension_importr:rcrgrhrsrirtr*r*r*r+r]„s r]csˆeZdZeeeeeeeee e e eeeee edœZdZ‡fdd„Zdd„Zd‡fdd „ Zd‡fd d„ Zd‡fdd „ Zd‡fdd„ Z‡ZS)ÚPostgresqlMetadata)ééééééi¼i½iii:iZi i;iòi¤i†z$from playhouse.postgres_ext import *cs|tt|ƒ |¡tdk rx| d¡}| ¡}|D]H\}}d|krLtj|j|<q.d|krbtj|j|<q.d|kr.tj |j|<q.dS)Nz!select oid, typname from pg_type;ZjsonZhstoreZtsvector) Úsuperrwr:rrcÚfetchallÚ JSONFieldruÚHStoreFieldÚ TSVectorField)r9r_ÚcursorÚresultsZoidZtypname©Ú __class__r*r+r:Ís zPostgresqlMetadata.__init__cCszi}tdk r ttjtjtjfƒntƒ}d||f}| d|¡}|jD].}|j |j t ¡||j<||j|krFd|_qF|S)Nz "%s"."%s"zSELECT * FROM %s LIMIT 1T) rÚsetr€r‚rrcÚdescriptionruÚgetÚ type_coder&r.r`)r9rQrlrmZextension_typesÚ identifierrƒÚcolumn_descriptionr*r*r+rhÝs&ýýý þ z#PostgresqlMetadata.get_column_typesNcs|pd}tt|ƒ ||¡S©NÚpublic)r~rwrgrrr…r*r+rgòszPostgresqlMetadata.get_columnscs|pd}tt|ƒ ||¡Sr)r~rwrsrrr…r*r+rsösz#PostgresqlMetadata.get_foreign_keyscs|pd}tt|ƒ ||¡Sr)r~rwrirrr…r*r+riúsz#PostgresqlMetadata.get_primary_keyscs|pd}tt|ƒ ||¡Sr)r~rwrtrrr…r*r+rtþszPostgresqlMetadata.get_indexes)N)N)N)N)r'r(r)ÚBooleanFieldÚ BlobFieldÚBigIntegerFieldr\Ú TextFieldÚ FloatFieldÚ CharFieldÚ DateFieldÚ DateTimeFieldÚ TimeFieldÚDecimalFieldrurvr:rhrgrsrirtÚ __classcell__r*r*r…r+rw·s2ïrwc(s°eZdZedkriZn|ejeejeej e ejeej eejeejeejeejeejeejeejeejeejeejeejeejeejeej eej!eiZ‡fdd„Z"ddd„Z#‡Z$S)Ú MySQLMetadataNcs.d|kr| d¡|d<tt|ƒj|f|ŽdS)NÚpasswordÚpasswd)Úpopr~ršr:)r9r_Úkwargsr…r*r+r:szMySQLMetadata.__init__cCsDi}| d|¡}|jD]&}|dd…\}}|j |t¡||<q|S)NzSELECT * FROM `%s` LIMIT 1é)rcrˆrur‰r&)r9rQrlrmrƒrŒr.rŠr*r*r+rh#s zMySQLMetadata.get_column_types)N)%r'r(r)rruÚBLOBr’ÚCHARr”ÚDATEr•ÚDATETIMEr–ÚDECIMALr˜ZDOUBLEr“ÚFLOATZINT24r\Z LONG_BLOBZLONGZLONGLONGr‘ZMEDIUM_BLOBZ NEWDECIMALZSHORTÚSTRINGÚ TIMESTAMPÚTIMEr—Z TINY_BLOBZTINYZ VAR_STRINGr:rhr™r*r*r…r+ršsZìršc@sbeZdZeeeeeeee e e e eee e e ee dœZdZdZdjeedZdZdd„Zdd d„Zd S) ÚSqliteMetadata)ÚbigintÚblobÚboolZbooleanÚcharÚdateÚdatetimeÚdecimalZintegerzinteger unsignedÚintÚlongÚrealZsmallintegerÚsmallintzsmallint unsignedÚtextÚtimez (?:["\[$]+)?z (?:["\]$]+)?zi(?:FOREIGN KEY\s*)?{begin}(.+?){end}\s+(?:.+\s+)?references\s+{begin}(.+?){end}\s*$["|\[]?(.+?)["|\]]?$)ÚbeginÚendz&^\s*(?:var)?char\s*$\s*(\d+)\s*$\s*$cCsR| ¡}||jkr|j|}n0t |j|¡r2t}nt dd|¡}|j |t¡}|S)Nz$.+$r^) ÚlowerruÚreÚsearchÚ re_varcharr”Úsubr‰r&)r9Úcolumn_typer0r/r*r*r+Ú_map_colPs zSqliteMetadata._map_colNcCs0i}|j |¡}|D]}| |j¡||j<q|Sr-)r_rgr¿rkr.)r9rQrlrmroÚcolumnr*r*r+rh[s zSqliteMetadata.get_column_types)N)r'r(r)r‘rrr”r•r–r˜r\r“r’r—rur·r¸ÚformatZre_foreign_keyr¼r¿rhr*r*r*r+r©1s:îýÿr©Ú_DatabaseMetadata)roÚprimary_keysÚforeign_keysrRÚindexesc@seZdZdd„Zdd„ZdS)ÚDatabaseMetadatacsLg}ˆjˆD]8}t|jƒdkr‡‡fdd„|jDƒ}| ||jf¡q|S)Nrfcs*g|]"}|ˆjˆkrˆjˆ|j‘qSr*)ror.)r>rÀ©r9rQr*r+Ú rsþz9DatabaseMetadata.multi_column_indexes..)rÅrjroÚappendr5)r9rQÚaccumr4Úfield_namesr*rÇr+Úmulti_column_indexesnsÿz%DatabaseMetadata.multi_column_indexescCs6i}|j|D]"}t|jƒdkr|j||jd<q|S)Nrfr)rÅrjror5)r9rQrÊr4r*r*r+Úcolumn_indexesxs zDatabaseMetadata.column_indexesN)r'r(r)rÌrÍr*r*r*r+rÆms rÆc@sxeZdZeegZddd„Zdd„Zeddd„ƒZ dd „Z d d„Zdd „Zdd„Z dd„Zdd„Zddd„Zddd„ZdS)ÚIntrospectorNcCs||_||_dSr-)rerl)r9rerlr*r*r+r:ƒszIntrospector.__init__cCsd|jjS)Nz©rer_r@r*r*r+rD‡szIntrospector.__repr__cCs<t|tƒrt|ƒ}nt|tƒr(t|ƒ}nt|ƒ}|||dS©N©rl)Ú isinstanceÚPostgresqlDatabaserwÚ MySQLDatabaseršr©)Úclsr_rlrer*r*r+Ú from_databaseŠs zIntrospector.from_databasecCst|jjƒSr-)Útyperer_r@r*r*r+Úget_database_class”szIntrospector.get_database_classcCs |jjjSr-rÏr@r*r*r+Úget_database_name—szIntrospector.get_database_namecCs |jjjSr-)rer_Úconnect_kwargsr@r*r*r+Úget_database_kwargsšsz Introspector.get_database_kwargscCs|jjrd|jjSdS)NÚ r^)rer`rvr@r*r*r+Úget_additional_importssz#Introspector.get_additional_importscCs@t dd|¡}d dd„| d¡Dƒ¡}|d ¡sr½r*r*r+rA¤sz/Introspector.make_model_name..Ú_rÚT)rºr½rBÚsplitÚisalpha)r9rQÚmodelZ model_namer*r*r+Úmake_model_name¢s zIntrospector.make_model_namecCs@t dd| ¡ ¡¡p| ¡}t dd|¡}|tkr<|d7}|S)Nz_id$r^rÞrà)rºr½r¹ÚstripÚRESERVED_WORDS)r9rÀr*r*r+Úmake_column_name©s zIntrospector.make_column_namecs¸|jr|jjj|jd}n|jj ¡}ˆdk r@‡fdd„|Dƒ}i}i}i}i}i}|D]D}|j ||j¡} |j ||j¡} z|j ||j¡||<Wn4tk rÊ}zt|j Žg||<W5d}~XYnX| |¡||<tdd„| Dƒƒ}| ¡D]>\} }| | ¡}| ¡}| d¡r,||kr,| ¡}||_qô| D]@}t|jƒdkr8|jd}|| kr8|j| |_d | |_q8|j ||j¡||<| ||<| ||<qXi}d d„}|D]‚}tƒ}t|||dD]d}z|||j}Wntk rüYqÊYnX|j}||kr"d ||jf||<n | |¡qÊq°|D]n}||D]^}||j|j}z||j|j}Wntk r†d}YnX|j|||| |¡dqDq8t|||||ƒS)NrÑcsg|]}|ˆkr|‘qSr*r*)r>rQ©Útable_namesr*r+rÈ¸sz+Introspector.introspect..css|]}| ¡VqdSr-)r¹)r>Zcolumn_namer*r*r+rAÖsÿz*Introspector.introspect..Ú_idrfrTcSs|jSr-)rÀ)rOr*r*r+Úôóz)Introspector.introspect..)Úkeyz %s_%s_set)rOrRrSr7) rlrer_Ú get_tablesrtrgrsÚ ValueErrorÚerrÚargsrår‡rXrèr¹Úendswithr.rjror5r4rirWrÀÚKeyErrorrPÚaddrQÚdest_columnrTr‰rÆ)r9rêZtablesrorÃrÄrRrÅrQZ table_indexesZ table_columnsÚexcZlower_col_namesÚcol_namerÀÚnew_nameZ lower_namer4Z related_namesZsort_fnZmodels_referencedrOrPÚsrcrSr*rér+Ú introspect°s¢ ÿ ÿ ÿ þÿ üûzIntrospector.introspectFcsdˆj|d‰i}G‡fdd„dtƒ‰‡‡‡‡fdd„‰tˆj ¡ƒD]\}}||krDˆ||ƒqD|S)NrécseZdZG‡fdd„dƒZdS)z/Introspector.generate_models..BaseModelcseZdZ”jjZdS)z4Introspector.generate_models..BaseModel.MetaN)r'r(r)rer_r*r@r*r+ÚMeta!srüN)r'r(r)rür*r@r*r+Ú BaseModel srýcs"ˆj|D]$}|j}||kr ||kr ˆ||ƒq g‰ˆj|}| ¡D]\}}|jrFˆ |j¡qFˆ |¡‰ˆ |¡}G‡fdd„dƒ}d} t ˆƒdkr | ¡‰t ˆƒdkrÌt‡fdd„| ¡DƒŽ|_d} d|i} | ¡D]\}}|j}|t krøt}||jd œ}|jr*| r*|tkr t}d|d <n|jrD|tk rDd|d <| ¡r˜| ¡rbd|d<n|jj} || |d<|jrˆ|j|d <d||f|d<||krÒ| ¡sÒ||rÀd|d<n| ¡sÒd|d<|f|Ž| |j<qÜztt|ƒˆf| ƒ||<Wntk rˆs‚YnXdS)NcseZdZ”ZdS)zAIntrospector.generate_models.._create_model..MetaN)r'r(r)rÅr*)rÌr*r+rü4srüFrrfcsg|]\}}|ˆkr|j‘qSr*rd)r>ÚcolrY)rÃr*r+rÈ<sÿzGIntrospector.generate_models.._create_model..T)r3rEr2r9r6r8z %s_%s_relr7r5r4)rÄrProrXr2rÉr.rÌrÍrjÚkeysÚCompositeKeyr/r&Ú BareFieldr1rHr\rIrNrOr8rJr×Ústrrð)rQÚmodelsrOrSror3rÀrÍrüÚ composite_keyrCZ FieldClassrKrP)rýÚ _create_modelr_Úskip_invalid)rÌrÃr+r$sj ÿ þ z3Introspector.generate_models.._create_model)rûÚModelrWrRrX)r9rrêrrQrär*)rýrr_r9rr+Úgenerate_modelssFzIntrospector.generate_models)N)N)N)FN)r'r(r)rHr\Z pk_classesr:rDÚclassmethodrÖrØrÙrÛrÝrårèrûrr*r*r*r+rÎ€s lrÎcCstj||d}| ¡SrÐ)rÎrÖrû)r_rlZintrospectorr*r*r+rûqsrû)N)ÚcollectionsrÚImportErrorÚdictrrºÚpeeweeZMySQLdb.constantsrZpymysql.constantsÚ playhouserr‡rçÚobjectr&r,r]rwršr©rÂrÆrÎrûr*r*r*r+Úsz üc3L.4 r