f3
@ sQ d d l Z d d l Z d d l m Z d d l m Z d d l Td d l m Z d d l m Z d d l Td Z
e d d g e d
d g e
d d
g i Z e d d e D Z d d Z d d d d Z d d Z d d Z d d Z d d Z e d k rMe j Z e Z e j \ Z Z e j r|y d d l m Z Wn) e! k
r{e d e j" d Yn Xe# e d k re d! e j$ e j" d e e Z% e d' Z& d Z' e j' rd" d# e j' j( d$ D Z' e e j) e& e% Z* e j+ r7d% j, e d d Z- e e- e* e e* e' d& e j d S)( 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] } | | f Vq q d S)N)DATABASE_ALIASES).0keyvalue r /pwiz.py <genexpr> s r c K st | t k r6 t d d j t j t j d | j d d } t | } | | | } t j | d | S)Nz)Unrecognized database, must be one of: %sz, schema) DATABASE_MAPerrjoinkeyssysexitpopZ IntrospectorZ
from_database)Z
database_typeZ
database_namekwargsr Z
DatabaseClassZdbr r r
make_introspector# s
r Fc s j d | t t j j j j t j f d f d d t } xI t
j j D]2 } | | k r | s | | k r | | q Wd S)NZ table_namesc s | p g } j | } x | D]} } | j } | | k rU | | k rU t d | | | k r | | k r | j | | | k r | | | | g q Wt d j | j | j } s t | } j | x | D] \ } } t | k | d k t
d k | j j k g } | r@q | j
rdt
d k rdd | _
t d | j q Wt d t d t d | j | }
|
rt d
xD t |
D]6 \ } } t d d j d
d | D | f qWt d j r(t d j t
d k rt f d d | D }
d j d d |
D } t d | t d | j | d S)Nz # Possible reference cycle: %szclass %s(BaseModel):idr Fz %s z class Meta:z db_table = '%s'z indexes = (z ((%s), %s),z, c s s | ] } d | Vq d S)z'%s'Nr )r fieldr r r
r e s z5print_models.<locals>._print_table.<locals>.<genexpr>z )z schema = '%s'c s+ g | ]! \ } } | k r | j q Sr )name)r colr ) primary_keysr r
<listcomp>n s z6print_models.<locals>._print_table.<locals>.<listcomp>c s s | ] } d | Vq d S)z'%s'Nr )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_key get_fieldmulti_column_indexesr r )tableseenZaccumr! Z foreign_keydestr$ r columnskipr* ZfieldsuniqueZpk_field_namesZpk_list) _print_tabledatabase introspectorpreserve_order)r r
r1 7 s\
z"print_models.<locals>._print_table)
Z
introspectr TEMPLATEZget_additional_importsZget_database_class__name__get_database_namereprZget_database_kwargssetr& r# r )r3 tablesr4 r, r+ r )r1 r2 r3 r4 r
print_models. s ? r; c C sn t j j } t d t d | t d | j d t d | j t d t t d 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 strftimer7 peewee_version)cmd_liner3 Z timestampr r r
print_header| s
r@ c C s% t j j d | t j j d S)Nz [91m%s[0m
)r stderrwriteflush)msgr r r
r s r c C s t d d } | j } | d d d d | d d d d d
d | d d
d d | d d d d d d t t } | d d d d d d d | d d | d d d d | d d d d! d d" | d# d$ d d% d d d d& | d' d( d d d d) d d* | S)+Nusagez$usage: %prog [options] database_namez-Hz--hostr- hostz-pz--portporttypeintz-uz--useruserz-Pz
--passwordpasswordaction
store_truez-ez--engineenginedefaultr choiceshelpzIDatabase type, e.g. sqlite, mysql or postgresql. Default is "postgresql".z-sz--schemar z-tz--tablesr: zWOnly generate the specified tables. Multiple table names should be separated by commas.z-iz--infoinfozIAdd database information and other metadata to top of the generated file.z-oz--preserve-orderr4 z6Model definition column ordering matches source table.)r Z
add_optionr& r )parserZaoZenginesr r r
get_option_parser s" rT c s? d } t f d d | D } j r; t | d <| S) NrF rG rJ r c 3 s3 | ]) } t | r | t | f Vq d S)N)getattr)r o)optionsr r
r s z%get_connect_kwargs.<locals>.<genexpr>rK )rF rG rJ r )dictrK r )rW Zopsr r )rW r
get_connect_kwargs s
rY __main__) OrderedDictz&Preserve order requires Python >= 2.7.r z%Missing required parameter "database"c C s( g | ] } | j r | j q Sr )strip)r r+ r r r
r s r , r4 ).r< r r Zoptparser Zpeeweer r r> Zplayhouse.reflectionr5 Z
MySQLDatabaseZPostgresqlDatabaseZSqliteDatabaser rX r r r; r@ r rT rY r6 argvZraw_argvrS
parse_argsrW argsr4 collectionsr[ ImportErrorr r( Z
print_helpconnectr2 r: splitrN r3 rR r r? r r r r
<module> sV
N