U
_3 @ s d dl Z d dlZd dlmZ d dlmZ d dlT d dlmZ d dlmZ d dl T dZ
e dd ge d
d ge
d d
giZedd eD Zdd Zd(ddZdd Zdd Zdd Zdd Zed krejZe Ze \Z Z e j rzd d l m Z W n( e!k
r ed e"d Y nX e#e d k r@ed! e$ e"d ee Z%e d" Z&dZ'e j'rrd#d$ e j'(d%D Z'ee j)e&fe%Z*e j+rd&,ed d Z-ee-e* ee*e'e j d' dS )) N)getpass) OptionParser)*)print_) __version__zfrom peewee import *%s
database = %s('%s', **%s)
class UnknownField(object):
def __init__(self, *_, **__): pass
class BaseModel(Model):
class Meta:
database = database
ZmysqlZmysqldbZpostgres
postgresqlZsqliteZsqlite3c c s$ | ] }t | D ]}||fV qqd S N)DATABASE_ALIASES).0keyvalue r
/pwiz.py <genexpr> s
r c K sV | t kr(tddt td |dd }t | }||f|}tj||dS )Nz)Unrecognized database, must be one of: %s, schema)r ) DATABASE_MAPerrjoinkeyssysexitpopZ IntrospectorZ
from_database)Z
database_typeZ
database_namekwargsr Z
DatabaseClassZdbr
r
r make_introspector# s
r Fc s j |dtt j t f d fdd t }t
j D ]"}||kr^|rv||kr^ || q^d S )N)Z table_namesc s |pg }j | }|D ]Z}|j}||kr<| |kr<td| ||kr||kr|| || kr|||| g qtdj| j| }st|}j| |D ]^\}}t | k|dkt
dk|j j kg} | rq|j
rt
dkrd|_
td| qtd td td | | }
|
r|td
t|
D ]*\} } td d d
d | D | f qHtd jrtdj t
dkrt fdd|D }
d dd |
D }td| td || d S )Nz # Possible reference cycle: %szclass %s(BaseModel):idr Fz %s z class Meta:z db_table = '%s'z indexes = (z ((%s), %s),r c s s | ]}d | V qdS z'%s'Nr
)r
fieldr
r
r r e s z5print_models.<locals>._print_table.<locals>.<genexpr>z )z schema = '%s'c s g | ]\}}| kr|j qS r
)name)r
colr primary_keysr
r
<listcomp>m s z6print_models.<locals>._print_table.<locals>.<listcomp>c s s | ]}d | V qdS r r
)r
Zpkr
r
r r p s z& primary_key = CompositeKey(%s)) foreign_keysZ
dest_tabler add model_namescolumnsitemssortedr# alllenZ field_classZ
pk_classesZ primary_keyZ get_fieldmulti_column_indexesr r )tableseenZaccumr% Z foreign_keydestr( r columnskipr- ZfieldsuniqueZpk_field_namesZpk_list _print_tabledatabase introspectorpreserve_orderr" r r5 7 sb
z"print_models.<locals>._print_table)N)
Z
introspectr TEMPLATEZget_additional_importsZget_database_class__name__get_database_namereprZget_database_kwargssetr* r' r )r7 tablesr8 r/ r. r
r4 r print_models. s
? r? c C sX t j }td td| td|d td| tdt td d S )Nz# Code generated by:z# python -m pwiz %sz
# Date: %sz%B %d, %Y %I:%M%pz# Database: %sz# Peewee version: %sr )datetimeZnowr strftimer; peewee_version)cmd_liner7 Z timestampr
r
r print_header| s
rD c C s t jd| t j d S )Nz [91m%s[0m
)r stderrwriteflush)msgr
r
r r s r c C s t dd} | j}|dddd |ddd d
d |d d
dd |ddddd tt}|dddd|dd |ddd d |d d d d d! |d"d#d$dd%d& |d'd(dd)d*d+ | S ),Nz$usage: %prog [options] database_name)Zusagez-Hz--hosthost)r0 z-pz--portportint)r0 typez-uz--useruserz-Pz
--passwordpassword
store_true)r0 actionz-ez--engineenginer zIDatabase type, e.g. sqlite, mysql or postgresql. Default is "postgresql".)r0 defaultchoiceshelpz-sz--schemar z-tz--tablesr> zWOnly generate the specified tables. Multiple table names should be separated by commas.)r0 rT z-iz--infoinfozIAdd database information and other metadata to top of the generated file.)r0 rP rT z-oz--preserve-orderr8 z6Model definition column ordering matches source table.)rP r0 rT )r Z
add_optionr* r )parserZaoZenginesr
r
r get_option_parser s*
rW c s. d}t fdd|D } jr*t |d< |S )N)rI rJ rM r c 3 s& | ] }t |r|t |fV qd S r )getattr)r
ooptionsr
r r s
z%get_connect_kwargs.<locals>.<genexpr>rN )dictrN r )r[ Zopsr r
rZ r get_connect_kwargs s
r] __main__) OrderedDictz&Preserve order requires Python >= 2.7.r z%Missing required parameter "database"c C s g | ]}| r| qS r
)strip)r
r. r
r
r r$ s r$ , )r8 )NF).r@ r r Zoptparser Zpeeweer r rB Zplayhouse.reflectionr9 Z
MySQLDatabaseZPostgresqlDatabaseZSqliteDatabaser r\ r r r? rD r rW r] r: argvZraw_argvrV
parse_argsr[ argsr8 collectionsr_ ImportErrorr r, Z
print_helpZconnectr6 r> splitrQ r7 rU r rC r
r
r
r <module> s`
N