Current File : //opt/alt/python38/lib64/python3.8/site-packages/__pycache__/pwiz.cpython-38.pyc
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(dd�Zdd� Zdd� Zdd� Zdd � Zed k�r�ejZe� 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+�r�d&�,ed d� �Z-ee-e*� ee*e'e j d'� dS ))� 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 ]}||fV qqd S �N)�DATABASE_ALIASES)�.0�key�value� r
�/pwiz.py� <genexpr> s
�r c K sV | t kr(tdd�t �� � � t�d� |�dd �}t | }||f|�}tj||d�S )Nz)Unrecognized database, must be one of: %s�, � �schema)r ) � 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��tt��� ��� 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 � qtd�j| � �j| �� }�s�t|�}�j| � |D ]^\}}t |� k|dkt
� �dk|j �j kg�} | r�q�|j
r�t
� �dkr�d|_
td|�� � q�td� td� td | � ��| �}
|
�r|td
� t|
�D ]*\} } td d �d
d� | D ��| f � �qHtd� �j�r�td�j � t
� �dk�r�t� 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_names�columns�items�sortedr# �all�lenZ field_classZ
pk_classesZ primary_keyZ 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_orderr" r r5 7 sb
�
� � z"print_models.<locals>._print_table)N)
Z
introspectr �TEMPLATEZget_additional_importsZget_database_class�__name__�get_database_name�reprZget_database_kwargs�setr* 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 j�d| � t j�� d S )Nz [91m%s[0m
)r �stderr�write�flush)�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--host�host)r0 z-pz--port�port�int)r0 �typez-uz--user�userz-Pz
--password�password�
store_true)r0 �actionz-ez--engine�enginer zIDatabase type, e.g. sqlite, mysql or postgresql. Default is "postgresql".)r0 �default�choices�helpz-sz--schemar z-tz--tablesr> zWOnly generate the specified tables. Multiple table names should be separated by commas.)r0 rT z-iz--info�infozIAdd 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
�o��optionsr
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
Mini Shell