HEX

File: //opt/alt/python38/lib64/python3.8/site-packages/__pycache__/pwiz.cpython-38.pyc
U

��_3�@s�ddlZddlZddlmZddlmZddlTddlmZddlmZddl	TdZ
edd	ged
dge
dd
giZedd�eD��Zdd�Zd(dd�Zdd�Zdd�Zdd�Zdd�Zedk�r�ejZe�Ze��\ZZej�rzddlm Z Wn(e!k
�red�e�"d �YnXe#e�d k�r@ed!�e�$�e�"d �ee�Z%ed"Z&dZ'ej'�rrd#d$�ej'�(d%�D�Z'eej)e&fe%�Z*ej+�r�d&�,ed d��Z-ee-e*�ee*e'ejd'�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
�mysqlZmysqldbZpostgres�
postgresqlZsqlite�sqlite3ccs$|]}t|D]}||fVqqdS�N)�DATABASE_ALIASES)�.0�key�value�r�7/opt/alt/python38/lib64/python3.8/site-packages/pwiz.py�	<genexpr>s
�rcKsV|tkr(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�popZIntrospectorZ
from_database)Z
database_typeZ
database_name�kwargsrZ
DatabaseClass�dbrrr�make_introspector#s�
rFcs��j|d��tt������j���t����f�d����fdd�	�t	�}t
�j���D]"}||kr^|rv||kr^�||�q^dS)N)Ztable_namescs�|pg}�j|}|D]Z}|j}||kr<||kr<td|�||kr||kr|�|�||kr�||||g�qtd�j|��j|��}�s�t|�}�j|�|D]^\}}t	|�k|dkt
��dk|j�jkg�}	|	r�q�|j
r�t
��dkr�d|_
td|���q�td�td�td	|���|�}
|
�r|td
�t|
�D]*\}}tdd�d
d�|D��|f��qHtd��j�r�td�j�t
��dk�r�t�fdd�|D��}
d�dd�|
D��}td|�td�|�|�dS)Nz# Possible reference cycle: %szclass %s(BaseModel):�idrFz    %s�z    class Meta:z        db_table = '%s'z        indexes = (z            ((%s), %s),rcss|]}d|VqdS�z'%s'Nr)r�fieldrrrresz5print_models.<locals>._print_table.<locals>.<genexpr>z	        )z        schema = '%s'csg|]\}}|�kr|j�qSr)�name)r�colr"��primary_keysrr�
<listcomp>ms�z6print_models.<locals>._print_table.<locals>.<listcomp>css|]}d|VqdSr!r)r�pkrrrrpsz&        primary_key = CompositeKey(%s))�foreign_keys�
dest_tabler�add�model_names�columns�items�sortedr&�all�lenZfield_classZ
pk_classes�primary_key�	get_field�multi_column_indexesrr)�table�seen�accumr)�foreign_key�destr-r#�column�skipr4�fields�uniqueZpk_field_namesZpk_list��_print_table�database�introspector�preserve_orderr%rr?7sb




�
��z"print_models.<locals>._print_table)N)
Z
introspectr�TEMPLATEZget_additional_importsZget_database_class�__name__�get_database_name�reprZget_database_kwargs�setr/r,r)rA�tablesrBr6r5rr>r�print_models.s
�?rIcCsXtj��}td�td|�td|�d��td|���tdt�td�dS)Nz# Code generated by:z# python -m pwiz %sz
# Date: %sz%B %d, %Y %I:%M%pz# Database: %sz# Peewee version: %sr )�datetime�nowr�strftimerE�peewee_version)�cmd_linerA�	timestamprrr�print_header|s
rPcCstj�d|�tj��dS)Nz[91m%s[0m
)r�stderr�write�flush)�msgrrrr�srcCs�tdd�}|j}|dddd�|ddd	d
d�|dd
dd�|ddddd�tt�}|dddd|dd�|dddd�|dddd d!�|d"d#d$dd%d&�|d'd(dd)d*d+�|S),Nz$usage: %prog [options] database_name)�usagez-Hz--host�host)r9z-pz--port�port�int)r9�typez-uz--user�userz-Pz
--password�password�
store_true)r9�actionz-ez--engine�enginerzIDatabase type, e.g. sqlite, mysql or postgresql. Default is "postgresql".)r9�default�choices�helpz-sz--schemarz-tz--tablesrHzWOnly generate the specified tables. Multiple table names should be separated by commas.)r9raz-iz--info�infozIAdd database information and other metadata to top of the generated file.)r9r]raz-oz--preserve-orderrBz6Model definition column ordering matches source table.)r]r9ra)r�
add_optionr/r)�parserZaoZenginesrrr�get_option_parser�s*
��
�
�recs.d}t�fdd�|D��}�jr*t�|d<|S)N)rVrWrZrc3s&|]}t�|�r|t�|�fVqdSr
)�getattr)r�o��optionsrrr�s
z%get_connect_kwargs.<locals>.<genexpr>r[)�dictr[r)ri�opsrrrhr�get_connect_kwargs�s

rl�__main__)�OrderedDictz&Preserve order requires Python >= 2.7.rz%Missing required parameter "database"���cCsg|]}|��r|���qSr)�strip)rr5rrrr'�s�r'�,� )rB)NF).rJrrZoptparser�peeweerrrMZplayhouse.reflectionrC�
MySQLDatabase�PostgresqlDatabase�SqliteDatabaserrjrrrIrPrrerlrD�argvZraw_argvrd�
parse_argsri�argsrB�collectionsrn�ImportErrorrr1�
print_help�connectr@rH�splitr^rArbrrNrrrr�<module>s`
��
N