U
SWUU @ s zd dl mZ W n ek
r( eZY nX d dl mZ d dlZd dlT zd dlmZ W n> ek
r zd dl mZ W n ek
r dZY nX Y nX zd dl
m Z W n ek
r dZ Y nX e ddd d
d d d
ddddddddddddddd d d d d d!d"d#d$d%g Z
G d&d' d'eZG d(d) d)eZG d*d+ d+eZG d,d- d-eZG d.d/ d/eZG d0d1 d1eZed2d3ZG d4d5 d5eZG d6d7 d7eZd:d8d9ZdS ); ) OrderedDict)
namedtupleN)*)
FIELD_TYPE) postgres_extandasassertbreakclasscontinuedefdelelifelseexceptexecfinallyforfromglobalifimportinislambdanotorpassprintraisereturntrywhilewithyieldc @ s e Zd ZdS ) UnknownFieldN)__name__
__module__ __qualname__ r* r* /reflection.pyr& s r& c @ s\ e Zd ZdZeefZdddZdd Zdd Z d
d Z
d d
Z dd Z dddZ
dd ZdS )Columnz1
Store metadata about a database column.
FNc C sF || _ || _|| _|| _|| _|| _|| _|| _d | _d | _ d | _
d S N) name field_classraw_column_typenullable primary_key db_columnindexunique rel_model related_nameto_field) selfr. r/ r0 r1 r2 r3 r4 r5 r* r* r+ __init__' s zColumn.__init__c s4 dddddg}d fdd|D }d j|f S )
Nr/ r0 r1 r2 r3 , c 3 s | ]}d |t |f V qdS z%s=%sN)getattr).0attrr9 r* r+ <genexpr>> s z"Column.__repr__.<locals>.<genexpr>zColumn(%s, %s))joinr. )r9 attrsZ keyword_argsr* r@ r+ __repr__7 s zColumn.__repr__c C s i }| j rd|d< | jtks(| j| jkr6d| j |d< | jrN| jtk rNd|d< | r| j|d< | j rtd| j |d< | j
rd| j
|d< | s| j rd |d
< n| j
r| sd |d < |S ) NTnullz'%s'r3 r2 r6 r8 r7 Truer5 r4 )r1 r/ ForeignKeyFieldr. r3 r2 PrimaryKeyFieldis_foreign_keyr6 r8 r7 is_primary_keyr5 r4 )r9 paramsr* r* r+ get_field_parametersC s&
zColumn.get_field_parametersc C s | j tkp| jS r- )r/ rH r2 r@ r* r* r+ rJ _ s zColumn.is_primary_keyc C s
| j tkS r- )r/ rG r@ r* r* r+ rI b s zColumn.is_foreign_keyc C s | j tko| jdkS Nz'self')r/ rG r6 r@ r* r* r+ is_self_referential_fke s
z Column.is_self_referential_fkc C sJ || _ t| _|j|jkr d| _n ||j | _|r6|jp8d | _|pBd | _d S rM ) foreign_keyrG r/
dest_tabletabler6 r. r8 r7 )r9 rO model_namesdestr7 r* r* r+ set_foreign_keyi s zColumn.set_foreign_keyc C sT | }ddd t| D }d| j| jj|f }| jtkrPd|| jf }|S )Nr; c s s | ]\}}d ||f V qdS r< r* )r> kvr* r* r+ rA w s z#Column.get_field.<locals>.<genexpr>z %s = %s(%s)z%s # %s) rL rB sorteditemsr. r/ r' r& r0 )r9 Z field_paramsZ param_strfieldr* r* r+ get_fieldt s
zColumn.get_field)FNFF)NN)r' r( r) __doc__ IntegerFieldrH Zprimary_key_typesr: rD rL rJ rI rN rT rZ r* r* r* r+ r, ! s
r, c @ sV e Zd Zi ZdZdd Zdd ZdddZdd d
Zdd d Z dd
dZ
dddZ dS )Metadata c C s || _ d| _d S )NF)databaserequires_extension)r9 r_ r* r* r+ r: s zMetadata.__init__c G s | j ||S r- )r_ Z execute_sql)r9 ZsqlrK r* r* r+ execute s zMetadata.executeNc
C s t dd | j||D }| ||}| ||}t|dkr\|d }|| tkr\t||< t }| D ]*\}} t ||| | j
| j | j |d||< qj|S )Nc s s | ]}|j |fV qd S r- r. )r> metadatar* r* r+ rA s z'Metadata.get_columns.<locals>.<genexpr> r )r/ r0 r1 r2 r3 )
r r_ get_columnsget_column_typesget_primary_keyslenr\ rH rX r, data_typerE r2 )
r9 rQ schemarc column_typesZpk_namesZpkcolumnsr. Z column_datar* r* r+ re s( zMetadata.get_columnsc C s t d S r- )NotImplementedErrorr9 rQ rj r* r* r+ rf s zMetadata.get_column_typesc C s | j ||S r- )r_ get_foreign_keysrn r* r* r+ ro s zMetadata.get_foreign_keysc C s | j ||S r- )r_ rg rn r* r* r+ rg s zMetadata.get_primary_keysc C s | j ||S r- )r_ get_indexesrn r* r* r+ rp s zMetadata.get_indexes)N)N)N)N)N) r' r( r)
column_mapextension_importr: ra re rf ro rg rp r* r* r* r+ r] s
r] c s e Zd Zeeeeeeeee e e
e e e e e
edZdZ fddZdd Zd fdd Zd fd
d Zd fd d
Zd fdd Z ZS )PostgresqlMetadata) i i i i i: iZ i i; i i i z$from playhouse.postgres_ext import *c s| t t| | td k rx| d}| }|D ]H\}}d|krLtj| j|< q.d|krbtj| j|< q.d|kr.tj | j|< q.d S )Nz!select oid, typname from pg_type;ZjsonZhstoreZtsvector)
superrs r: r ra Zfetchall JSONFieldrq HStoreField
TSVectorField)r9 r_ cursorresultsZoidZtypname __class__r* r+ r: s
zPostgresqlMetadata.__init__c C sz i }t d k r tt jt jt jfnt }d||f }| d| }|jD ].}| j|j t
||j < ||j |krFd| _ qF|S )Nz "%s"."%s"zSELECT * FROM %s LIMIT 1T)
r setr{ r} r| ra descriptionrq get type_coder& r. r` )r9 rQ rj rk Zextension_typesZ
identifierr~ column_descriptionr* r* r+ rf s&
z#PostgresqlMetadata.get_column_typesNc s |pd}t t| ||S NZpublic)rz rs re rn r r* r+ re s z PostgresqlMetadata.get_columnsc s |pd}t t| ||S r )rz rs ro rn r r* r+ ro s z#PostgresqlMetadata.get_foreign_keysc s |pd}t t| ||S r )rz rs rg rn r r* r+ rg s z#PostgresqlMetadata.get_primary_keysc s |pd}t t| ||S r )rz rs rp rn r r* r+ rp s z PostgresqlMetadata.get_indexes)N)N)N)N)r' r( r) BooleanField BlobFieldBigIntegerFieldr\ TextField
FloatField CharField DateField
DateTimeField TimeField DecimalFieldrq rr r: rf re ro rg rp
__classcell__r* r* r r+ rs s2 rs c ( s e Zd Zedkri Zn|ejeejeej e
ej e ej
eejeejeejeejeejeejeejeejeejeejeej e ej e ej eej eej!eiZ fddZ"dddZ# Z$S )
MySQLMetadataNc s. d|kr| d|d< tt| j|f| d S )NZpasswordZpasswd)poprz r r: )r9 r_ kwargsr r* r+ r: s zMySQLMetadata.__init__c C sD i }| d| }|jD ]&}|d d \}}| j|t||< q|S )NzSELECT * FROM `%s` LIMIT 1 )ra r rq r r& )r9 rQ rj rk r~ r r. r r* r* r+ rf # s
z MySQLMetadata.get_column_types)N)%r' r( r) r rq ZBLOBr ZCHARr ZDATEr ZDATETIMEr ZDECIMALr ZDOUBLEr ZFLOATZINT24r\ Z LONG_BLOBZLONGZLONGLONGr Z MEDIUM_BLOBZ
NEWDECIMALZSHORTSTRING TIMESTAMPZTIMEr Z TINY_BLOBZTINYZ
VAR_STRINGr: rf r r* r* r r+ r sZ r c @ sb e Zd Zeeeeeeee e
e
e
ee e
e
e
e e
dZdZdZdjeedZdZdd Zd d
d Zd S )
SqliteMetadata)ZbigintZblobboolZbooleanchardateZdatetimeZdecimalZintegerzinteger unsignedintZlongrealZ smallintegerZsmallintzsmallint unsignedtexttimez
(?:["\[\(]+)?z
(?:["\]\)]+)?zi(?:FOREIGN KEY\s*)?{begin}(.+?){end}\s+(?:.+\s+)?references\s+{begin}(.+?){end}\s*\(["|\[]?(.+?)["|\]]?\))beginendz&^\s*(?:var)?char\s*\(\s*(\d+)\s*\)\s*$c C sR | }|| jkr | j| }n0t| j|r2t}n tdd|}| j|t}|S )Nz\(.+\)r^ ) lowerrq research
re_varcharr subr r& )r9 Z column_typer0 r/ r* r* r+ _map_colP s
zSqliteMetadata._map_colNc C s0 i }| j |}|D ]}| |j||j< q|S r- )r_ re r ri r. )r9 rQ rj rk rl columnr* r* r+ rf [ s
z SqliteMetadata.get_column_types)N)r' r( r) r r r r r r r r\ r r r rq r r formatZre_foreign_keyr r rf r* r* r* r+ r 1 s: r _DatabaseMetadata)rl primary_keys foreign_keysrR indexesc @ s e Zd Zdd Zdd ZdS )DatabaseMetadatac sL g } j D ]8}t|jdkr fdd|jD }|||jf q|S )Nrd c s* g | ]"}| j kr j | jqS r* )rl r. )r> r r9 rQ r* r+
<listcomp>r s z9DatabaseMetadata.multi_column_indexes.<locals>.<listcomp>)r rh rl appendr5 )r9 rQ accumr4 field_namesr* r r+ multi_column_indexesn s z%DatabaseMetadata.multi_column_indexesc C s6 i }| j | D ]"}t|jdkr|j||jd < q|S )Nrd r )r rh rl r5 )r9 rQ r r4 r* r* r+ column_indexesx s
z DatabaseMetadata.column_indexesN)r' r( r) r r r* r* r* r+ r m s
r c @ sx e Zd ZeegZdddZdd ZedddZ dd Z
d
d Z d d
Z dd Z
dd Zdd ZdddZd ddZdS ) IntrospectorNc C s || _ || _d S r- )rc rj )r9 rc rj r* r* r+ r: s zIntrospector.__init__c C s d| j j S )Nz<Introspector: %s>rc r_ r@ r* r* r+ rD s zIntrospector.__repr__c C s< t |trt|}n t |tr(t|}nt|}| ||dS Nrj )
isinstanceZPostgresqlDatabasers Z
MySQLDatabaser r )clsr_ rj rc r* r* r+
from_database s
zIntrospector.from_databasec C s t | jjS r- )typerc r_ r@ r* r* r+ get_database_class s z Introspector.get_database_classc C s
| j jjS r- r r@ r* r* r+ get_database_name s z Introspector.get_database_namec C s
| j jjS r- )rc r_ Zconnect_kwargsr@ r* r* r+ get_database_kwargs s z Introspector.get_database_kwargsc C s | j jrd| j j S dS )N
r^ )rc r` rr r@ r* r* r+ get_additional_imports s z#Introspector.get_additional_importsc C s@ t dd|}ddd |dD }|d s<d| }|S )N[^\w]+r^ c s s | ]}| V qd S r- )title)r> r r* r* r+ rA s z/Introspector.make_model_name.<locals>.<genexpr>_r T)r r rB splitisalpha)r9 rQ modelZ
model_namer* r* r+ make_model_name s
z Introspector.make_model_namec C s@ t dd| p | }t dd|}|tkr<|d7 }|S )Nz_id$r^ r r )r r r stripRESERVED_WORDS)r9 r r* r* r+ make_column_name s
z Introspector.make_column_namec s | j r| jjj| j d}n | jj } d k r@ fdd|D }i }i }i }i }i }|D ]D}| j|| j } | j|| j }
z| j|| j ||< W n4 tk
r } zt| j g ||< W 5 d } ~ X Y nX |
|||< t dd |
D } |
D ]>\}
}|
|
}|
}|dr,|| kr,|
}||_q| D ]@}t|jdkr8|jd }||
kr8|j|
| _d |
| _q8| j|| j ||< |
||< | ||< qXi }d
d }|D ]}t }t|| |d D ]d}z|| |j }W n tk
r Y qY nX |j}||kr"d
||jf ||< n
|| qʐq|D ]n}|| D ]^}||j |j }z||j |j }W n tk
r d }Y nX |j |||| |d qDq8t |||||S )Nr c s g | ]}| kr|qS r* r* )r> rQ table_namesr* r+ r s z+Introspector.introspect.<locals>.<listcomp>c s s | ]}| V qd S r- )r )r> Z column_namer* r* r+ rA s z*Introspector.introspect.<locals>.<genexpr>Z_idrd r Tc S s | j S r- )r )rO r* r* r+ <lambda> z)Introspector.introspect.<locals>.<lambda>)keyz %s_%s_set)rO rR rS r7 ) rj rc r_ Z
get_tablesrp re ro
ValueErrorerrargsr r rX r r endswithr. rh rl r5 r4 rg rW r KeyErrorrP addrQ Z dest_columnrT r r )r9 r Ztablesrl r r rR r rQ Z
table_indexesZ
table_columnsexcZlower_col_namesZcol_namer new_nameZ
lower_namer4 Z
related_namesZsort_fnZmodels_referencedrO rP srcrS r* r r+
introspect s
zIntrospector.introspectFc sd j |di }G fdddt fddtj D ]\}}||krD|| qD|S )Nr c s e Zd ZG fdddZdS )z/Introspector.generate_models.<locals>.BaseModelc s e Zd Z jjZdS )z4Introspector.generate_models.<locals>.BaseModel.MetaN)r' r( r) rc r_ r* r@ r* r+ Meta! s r N)r' r( r) r r* r@ r* r+ BaseModel s r c s" j | D ]$}|j}||kr
|| kr
|| q
g j| }| D ]\}}|jrF|j qF| | }G fddd}d} t dkr|
t dkrt fdd| D |_d} d|i}
| D ]\}}|j } | t
krt} ||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|| rd| d< n| sd| d< | f | |
|j< qztt| f|
|| < W n tk
r s Y nX d S )Nc s e Zd Z ZdS )zAIntrospector.generate_models.<locals>._create_model.<locals>.MetaN)r' r( r) r r* )r r* r+ r 4 s r Fr rd c s g | ]\}}| kr|j qS r* rb )r> colrY )r r* r+ r < s zGIntrospector.generate_models.<locals>._create_model.<locals>.<listcomp>T)r3 rE r2 r9 r6 r8 z %s_%s_relr7 r5 r4 )r rP rl rX r2 r r. r r rh keysZ CompositeKeyr/ r& Z BareFieldr1 rH r\ rI rN rO r8 rJ r strr )rQ modelsrO rS rl r3 r r r Z
composite_keyrC Z
FieldClassrK rP )r
_create_modelr_ skip_invalid)r r r+ r $ sj
z3Introspector.generate_models.<locals>._create_model)r ZModelrW rR rX )r9 r r r rQ r r* )r r r_ r9 r r+ generate_models s F z Introspector.generate_models)N)N)N)FN)r' r( r) rH r\ Z
pk_classesr: rD classmethodr r r r r r r r r r* r* r* r+ r s
lr c C s t j| |d}| S r )r r r )r_ rj Z introspectorr* r* r+ r q s r )N) collectionsr ImportErrordictr r ZpeeweeZMySQLdb.constantsr Zpymysql.constantsZ playhouser r r objectr& r, r] rs r r r r r r r* r* r* r+ <module> sz
c3L.4
r