
    Eh-                     ,   S SK rS SKrS SKJr  S SKrS SKrS SKJr  S SKJ	r	  SSK
Jr  SSK
Jr  SS	K
Jr  SSK
Jr  SS
KJr  SSKJr  S rS!S jrS rS rS rS"S jrS#S jrS$S jrS%S jrS%S jrS rS rS r S&S jrS rS r S r!S r"S\#S\#4S  jr$g)'    N)Path)config)	provision   )util   )command)script)Script)ScriptDirectoryc                  T    [         R                  (       a  S[         R                   3$ g)Nscratch_scratch)r   FOLLOWER_IDENT     V/home/kali/devsecops-assessor/venv/lib/python3.13/site-packages/alembic/testing/env.py_get_staging_directoryr      s"    )22344r   c                    [        5       nU (       a  [        [        5       S5      n[        R                  R                  U5      (       a
   SU-  5       e[        R                  " X4US9  U(       aC   [        R                  " US5        US;   d   U5       e[        [        US5      SU;   a  SOS5        [        R                  R                  U5      nU$ ! [         a     NXf = f)Nscriptsz2staging directory %s already exists; poor cleanup?)templateenv.py)pep3147_envonlysimplepep3147_everythingpep3147r   )_testing_config
_join_pathr   ospathexistsr	   initr   load_python_fileAttributeErrormake_sourcelessr
   r   from_config)creater   
sourcelesscfgr    scs         r   staging_envr+      s    

C02I>77>>$'' 	
@4G	
' 	S2	%%dH5  "   	 
 4*&*4	(
 
			+	+C	0BI% "  s   1C 
C! C!c                      SSK Jn   U R                  R                  5         [        R
                  " [        5       S5        g )Nr   enginesT)sqlalchemy.testingr.   testing_reaper	close_allshutilrmtreer   r-   s    r   clear_staging_envr4   ;   s)    *$$&
MM(*D1r   c                     [        [        5       S5      n[        US5      n[        US5       nUR                  U 5        S S S 5        g ! , (       d  f       g = f)Nr   zscript.py.makow)r   r   openwrite)txtdir_r    fs       r   script_file_fixturer<   B   s?    ,.	:Dd,-D	dCA	 
s   A
Ac                 "   [        [        5       S5      nSU -   n [        US5      n[        R                  " U5      nU(       a  [        R
                  " U5        [        US5       nUR                  U 5        S S S 5        g ! , (       d  f       g = f)Nr   z6
from alembic import context

config = context.config
r   r6   )r   r   r   pyc_file_from_pathr   unlinkr7   r8   )r9   r:   r    pyc_pathr;   s        r   env_file_fixturerA   I   ss    ,.	:D	
 	  dH%D&&t,H
		(	dCA	 
s   %B  
Bc                 ~    [        [        5       S5      nSU< SU < 3nU(       a  X#S'   [        R                  " XQUS9$ )Nr   
sqlite:////scope)urlfutureoptions)r   r   testing_utiltesting_engine)tempnamerG   rE   rH   r:   rF   s         r   _sqlite_file_dbrL   ]   s:    ,.	:D#X
.C &&3wOOr   c                     [        [        5       S5      nSU S3nU=(       d'    S[        R                  R                  R
                  ;   n[        SU SU SU (       a  SOS	 S
U(       a  SOS S3	5      $ )Nr   rC   z/foo.dbrG   
[alembic]
script_location = 
sqlalchemy.url = 
sourceless = truefalse
zsqlalchemy.future = true a  

[loggers]
keys = root,sqlalchemy

[handlers]
keys = console

[logger_root]
level = WARNING
handlers = console
qualname =

[logger_sqlalchemy]
level = DEBUG
handlers =
qualname = sqlalchemy.engine

[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
    r   r   r   db	__class__
__module___write_config_file)r(   rG   r:   rF   sqlalchemy_futures        r   _sqlite_testing_configr[   e   s    ,.	:DtfG
$CN8vyy/B/B/M/M#M&  "f0 10b 9 :#	% %r   c                     [        [        5       S5      nS[        R                  R                  R
                  ;   nSU-  n[        SU SU SU(       a  SOS S	U (       a  SOS S
U S35      $ )Nr   rG   zsqlite:///%s/foo.dbrN   rO   z
sqlalchemy.future = rQ   rR   rP   z_
path_separator = space
version_locations = %(here)s/model1/ %(here)s/model2/ %(here)s/model3/ aR  

[loggers]
keys = root

[handlers]
keys = console

[logger_root]
level = WARNING
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
    rU   )r(   extra_version_locationr:   rZ   rF   s        r   _multi_dir_testing_configr^      s    ,.	:D FII$7$7$B$BB
$
&C&  0Vg> ?"f0 1   !	# #r   c                     [        [        5       S5      n[        SU S[        R                  " U5       S3SU  S35      $ )Huse a postgresql url with no host so that
connections guaranteed to failr   z"
[tool.alembic]
script_location ="z"
z


        z
[alembic]
sqlalchemy.url = aR  ://

[loggers]
keys = root

[handlers]
keys = console

[logger_root]
level = WARNING
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

)r   r   _write_toml_configtextwrapdedentdialect
directivesr:   s      r   _no_sql_pyproject_configrg      s]     ,.	:D& 	  		 	$ $r   c           	      T    [        [        5       S5      n[        SU SU  SU S35      $ )r`   r   z
[alembic]
script_location =rO   z://
aO  

[loggers]
keys = root

[handlers]
keys = console

[logger_root]
level = WARNING
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

)r   r   rY   rd   s      r   _no_sql_testing_configri      sI     ,.	:D   		   r   c                     [        U5      n[        UR                  S5       nUR                  U 5        S S S 5        U$ ! , (       d  f       U$ = fNr6   )rY   r7   toml_file_namer8   )tomltextinitextr)   r;   s       r   ra   ra     sB    
W
%C	c  #	&!	 
'J 
'	&Js	   >
Ac                     [        5       n[        UR                  S5       nUR                  U 5        S S S 5        U$ ! , (       d  f       U$ = frk   )r   r7   config_file_namer8   )textr)   r;   s      r   rY   rY     s?    

C	c""C	(A	 
)J 
)	(Js	   =
Ac                     SSK Jn   [        R                  " [	        5       [        R
                  5      (       d  [        R                  " [	        5       5        U " [        [	        5       S5      [        [	        5       S5      5      $ )Nr   Configztest_alembic.inizpyproject.toml)alembic.configrt   r   accessr   F_OKmkdirr   rs   s    r   r   r      sY    %99+-rww77
')*)+-?@)+-=> r   c                    U R                   R                  U5      nUR                  n[        R                  " U5      nU(       a  UR                  U5      n[        US5       nUR                  U5        S S S 5        [        R                  " U5      nU(       a  [        R                  " U5        [        R                  " X5      n	U R                   R                  U	R                  5      nUR                  U	R                  :w  a  [!        S5      eU R                   R#                  U	SS9  U(       a  [%        XdS:X  a  SOS5        g g ! , (       d  f       N= f)Nwbz2Can't change down_revision on a refresh operation.T)_replacer   r   r   )revision_mapget_revisionr    rb   rc   encoder7   r8   r   r>   r   r?   r   
_from_pathrevisiondown_revision	Exceptionadd_revisionr%   )
	scriptdirrev_idcontentencodingr(   oldr    fpr@   r
   s
             r   write_scriptr   +  s    
 
 
-
-f
5C88Doog&G..*	dD	R
 
&&t,H
		(y/F

 
 
-
-foo
>C
F000LMM'''>-AA)x	
  
	s   "E
Ec                    SS K nUR                  U 5        US:X  aq  [        R                  " U 5      n[        R
                  R                  S   n[        R                  R                  U 5      u  pVXT-   n[        R                  " X75        UnOUS;   d   e[        R                  " U 5      n[        R                  " U[        R                  5      (       d   e[        R                  " U 5        g )Nr   r   )r   r   )
py_compilecompiler   r>   	importlib	machineryBYTECODE_SUFFIXESr   r    splitextr2   moverv   rw   r?   )r    styler   r@   suffixfilepathextsimple_pyc_paths           r   r%   r%   E  s    t**40$$66q9((."+H."----**4099Xrww''''IIdOr   c           	         [         R                  " 5       n[         R                  " 5       n[         R                  " 5       n[        R                  " U 5      nUR                  " USSSS9  [        UUSU S35        UR                  " USSUS9  [        UUSU S	U S
3SS9  UR                  " USSUS9  [        UUSU S	U S35        XU4$ )Nz
revision aTbase)refreshheadz"Rev A"
revision = 'z'
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, %3"
revision = ''
down_revision = 'z}'

from alembic import op


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


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

zutf-8)r   z
revision cz"Rev C"
revision = 'z}'

from alembic import op


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


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

r   r   r   r&   generate_revisionr   )r)   abcr
   s        r   three_rev_fixturer   Z  s    AAA((-F
QdH	C 	* QdC	C  	  ', QdC	C  	( 7Nr   c           	         [         R                  " 5       n[         R                  " 5       n[         R                  " 5       n[        R                  " U 5      nUR                  " USUSSS9  [        UUSU SU S35        UR                  " USUSSS9  [        UUSU SU S	35        UR                  " US
USSS9  [        UUSU SU S35        XEU4$ )z:Create a multiple head fixture from the three-revs fixturezrevision d from bT)r   splicer   z"Rev D"
revision = 'r   z}'

from alembic import op


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


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

zrevision e from dz"Rev E"
revision = 'z}'

from alembic import op


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


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

zrevision f from bz"Rev F"
revision = 'z}'

from alembic import op


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


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

r   )r)   r   r   r   der;   r
   s           r   multi_heads_fixturer     s    	AAA((-F
	QtT 	C  	* 	QtT 	C  	* 	QtT 	C  	* 7Nr   c                 D   [        [        5       S5      nS[        R                  R                  R
                  ;   nSR                  U R                  5       5      nSR                  S U R                  5        5       5      n [        SU SU(       a  SOS	 S
U SU  S3	5      $ )z?alembic.ini fixture to work exactly with the 'multidb' templater   rG   z, z

c              3   J   #    U  H  u  pS U SUR                    3v   M     g7f)[z]
sqlalchemy.url = N)rF   ).0keyvalues      r   	<genexpr>*_multidb_testing_config.<locals>.<genexpr>  s+      )JC C5$UYYK0)s   !#rN   z(
sourceless = false
sqlalchemy.future = rQ   rR   z
databases = aQ  
[loggers]
keys = root

[handlers]
keys = console

[logger_root]
level = WARNING
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
    )
r   r   r   rV   rW   rX   joinkeysitemsrY   )r.   r:   rZ   	databasess       r   _multidb_testing_configr     s     ,.	:D FII$7$7$B$BB		',,.)Ikk !--/ G
 & 0Vg> ?K 	 
	! !r   r   morec                 `    [        [        U 5      R                  " U6 R                  5       5      $ )N)strr   joinpathas_posix)r   r   s     r   r   r   ,  s%    tDz""D)22455r   )TgenericF)zfoo.dbFN)FF)FrT   )
postgresqlrT   )asciiF)%importlib.machineryr   r   pathlibr   r2   rb   r/   r   r   rT   r   rI   r	   r
   r   r   r   r+   r4   r<   rA   rL   r[   r^   rg   ri   ra   rY   r   r   r%   r   r   r   r   r   r   r   r   <module>r      s     	    % ( "     $D2(P+\)X)X$N >C
4*HVSl.b6S 6 6r   