���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 r�e d! � e j$ � e j" d � e e � Z% e d' Z& d Z' e j' r�d" 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__z�from 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)�.0�key�value� 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_MAP�err�join�keys�sys�exit�popZ IntrospectorZ
from_database)Z
database_typeZ
database_name�kwargsr 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 � q�Wt d � � j r(t d � j � t
� � d k r�t � 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_names�columns�items�sortedr �all�lenZ field_classZ
pk_classesZ primary_key� get_field�multi_column_indexesr r )�table�seenZaccumr! Z foreign_key�destr$ r �column�skipr* Zfields�uniqueZpk_field_namesZpk_list)� _print_table�database� introspector�preserve_order)r r
r1 7 s\
z"print_models.<locals>._print_table)
Z
introspectr �TEMPLATEZget_additional_importsZget_database_class�__name__�get_database_name�reprZget_database_kwargs�setr& 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 �stderr�write�flush)�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)+N�usagez$usage: %prog [options] database_namez-Hz--hostr- �hostz-pz--port�port�type�intz-uz--user�userz-Pz
--password�password�action�
store_truez-ez--engine�engine�defaultr �choices�helpzIDatabase 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--info�infozIAdd 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_help�connectr2 r: �splitrN r3 rR r r? r r r r
�<module> sV
N