B

�M!VU�@s�dd
lZdd
l
Z
dd
lZddlmZ
ddlmZmZ
ddlm	Z	
ddlm
Z

ddlmZ
d	d
�Zd.dd�
Z
dd�Zdd�Zdd�Zd/dd�
Zd0dd�
Zd1dd�
Zd2dd �
Zd!d"�Zd#d$�Zd3d&d'�
Zd(d)�Zd*d+�Zd,d-�Zd
S)4�N�)
�
u)�Script�ScriptDirectory)
�util�
)
�engines)
�	provisioncCstj
rd
tj
SdSdS)Nz
scratch_%sZscratch)r	ZFOLLOWER_IDENT�r
r
�D/opt/alt/python37/lib/python3.7/site-packages/alembic/testing/env.py�_get_staging_directorys


rT�genericFcCs�d
dlm
}m}
t�}|r�tj�t�d�}tj�|�
r@t	�
|�

|j|||
d�
|r�yt�
|d�
Wntk
rx


YnXttj�|d��

|j�|�
}|S)Nr
)�command�script�scripts)
�templatezenv.py)�alembicrr�_testing_config�os�path�joinr�exists�shutil�rmtree�initrZload_python_file�AttributeError�make_sourcelessr�from_config)�creater�
sourcelessrr�cfgrZscr
r
r�staging_envs












r!cCst�
t�d
�
dS)NT)rrrr
r
r
r�clear_staging_env.s
r"c
	CsBtj
�t�d
�}
tj
�|
d�}t|d��}|�|�

WdQRXdS)Nrzscript.py.mako�
w)rrrr�open�write)�txt�dir_r�
fr
r
r�script_file_fixture2s



r)c
	Csltj
�t�d
�}
d|}tj
�|
d�}t�|�
}t�|tj�rHt�|�

t	|d��}|�
|�

WdQRXdS)Nrz6
from alembic import context

config = context.config
zenv.pyr#)rrrrr�pyc_file_from_path�access�F_OK�unlinkr$r%)r&r'r�pyc_pathr(r
r
r�env_file_fixture9s






r/�foo.dbc
Cs(tj
�t�d
�}
d|
|f}tj|d�
S)Nrzsqlite:///%s/%s)
�url)rrrrrZtesting_engine)�tempnamer'r1r
r
r�_sqlite_file_dbJs


r3c
Cs2tj
�t�d
�}
d|
}td|
||r(dndf�
S)Nrzsqlite:///%s/foo.dba�

[alembic]
script_location = %s
sqlalchemy.url = %s
sourceless = %s

[loggers]
keys = root

[handlers]
keys = console

[logger_root]
level = WARN
handlers = console
qualname =

[handler_console]
class = StreamHandler
args = (sys.stderr,)
level = NOTSET
formatter = generic

[formatters]
keys = generic

[formatter_generic]
format = %%(levelname)-5.5s [%%(name)s] %%(message)s
datefmt = %%H:%%M:%%S
    �true�false)rrrr�_write_config_file)rr'r1r
r
r�_sqlite_testing_configPs

r7c
Cs2tj
�t�d
�}
d|
}td|
||r(dndf�
S)Nrzsqlite:///%s/foo.dba�

[alembic]
script_location = %s
sqlalchemy.url = %s
sourceless = %s
version_locations = %%(here)s/model1/ %%(here)s/model2/ %%(here)s/model3/

[loggers]
keys = root

[handlers]
keys = console

[logger_root]
level = WARN
handlers = console
qualname =

[handler_console]
class = StreamHandler
args = (sys.stderr,)
level = NOTSET
formatter = generic

[formatters]
keys = generic

[formatter_generic]
format = %%(levelname)-5.5s [%%(name)s] %%(message)s
datefmt = %%H:%%M:%%S
    r4r5)rrrrr6)rr'r1r
r
r�_multi_dir_testing_configvs

r8�
postgresql�cCs"tj
�t�d
�}td|||
f�
S)zLuse a postgresql url with no host so that
    connections guaranteed to failra�

[alembic]
script_location = %s
sqlalchemy.url = %s://
%s

[loggers]
keys = root

[handlers]
keys = console

[logger_root]
level = WARN
handlers = console
qualname =

[handler_console]
class = StreamHandler
args = (sys.stderr,)
level = NOTSET
formatter = generic

[formatters]
keys = generic

[formatter_generic]
format = %%(levelname)-5.5s [%%(name)s] %%(message)s
datefmt = %%H:%%M:%%S

)rrrrr6)ZdialectZ
directivesr'r
r
r�_no_sql_testing_config�s
r;c
	Cs,t�}
t
|
jd
��}|�|�

WdQRX|
S)Nr#)rr$Zconfig_file_namer%)�textr r(r
r
rr6�s



r6c
Cs<d
dlm
}
t�t�tj�s(t�t��

|tj�t�d��
S)Nr
)
�Configztest_alembic.ini)	Zalembic.configr=rr+rr,�mkdirrr)
r=r
r
rr�s



r�asciic
	Cs�|j�
|
�
}|j}t�|�
}|r*|�|�
}t|d
��}|�|�

WdQRXt�	|�
}t
�|t
j�rlt
�
|�

t�||�}	|j�
|	j�
}|j|	jkr�td�
�
|jj|	dd�
|r�t|�

dS)N�wbz2Can't change down_revision on a refresh operation.T)
�_replace)Zrevision_mapZget_revisionr�textwrap�dedent�encoder$r%rr*rr+r,r-rZ
_from_path�revisionZ
down_revision�	ExceptionZadd_revisionr)
Z	scriptdir�rev_id�content�encodingr�oldr�fpr.rr
r
r�write_script�s"
















rLc
CsNt�
|�
}
t�|
tj�st�
t�|�
}t�|tj�s@t�|
|�
t�	|�

dS)
N)
rr*rr+r,�AssertionErrorZsimple_pyc_file_from_pathr�copyfiler-)rr.Zsimple_pyc_pathr
r
rr�s




rc
Cs�t�
�}
t�
�}t�
�}t�|�
}|j|
d
dd�
t||
d|
�
|j|ddd�
t||td�
||
fdd�
|j|d	dd�
t||d
||f�
|
||fS)Nz
revision aT)
Zrefreshz�"Rev A"
revision = '%s'
down_revision = None

from alembic import op

def upgrade():
    op.execute("CREATE STEP 1")

def downgrade():
    op.execute("DROP STEP 1")

z
revision bu�# coding: utf-8
"Rev B, méil"
revision = '%s'
down_revision = '%s'

from alembic import op

def upgrade():
    op.execute("CREATE STEP 2")

def downgrade():
    op.execute("DROP STEP 2")

zutf-8)
rIz
revision cz�"Rev C"
revision = '%s'
down_revision = '%s'

from alembic import op

def upgrade():
    op.execute("CREATE STEP 3")

def downgrade():
    op.execute("DROP STEP 3")

)rrGrrZgenerate_revisionrLr)r �
a�
b�
crr
r
r�three_rev_fixture�s











rRc
CsHtj
�t�d
�}
d�|���
}d�dd�|��D�
�
}td|
||f�
S)z?alembic.ini fixture to work exactly with the 'multidb' templaterz, z

c
ss |]\}
}d|
|jfV
qd
S)z[%s]
sqlalchemy.url = %sN)
r1)�.0�key�valuer
r
r�	<genexpr>6
sz*_multidb_testing_config.<locals>.<genexpr>a�

[alembic]
script_location = %s
sourceless = false

databases = %s

%s
[loggers]
keys = root

[handlers]
keys = console

[logger_root]
level = WARN
handlers = console
qualname =

[handler_console]
class = StreamHandler
args = (sys.stderr,)
level = NOTSET
formatter = generic

[formatters]
keys = generic

[formatter_generic]
format = %%(levelname)-5.5s [%%(name)s] %%(message)s
datefmt = %%H:%%M:%%S
    )rrrr�keys�itemsr6)rr'Z	databasesr
r
r�_multidb_testing_config-
s


rY)Tr
F)
r0)
F)
F)r9r:)r?F)rrrBZutil.compatrrrrr:rrr	rr!r"r)r/r3r7r8r;r6rrLrrRrYr
r
r
r�<module>s,








&
%
%
8