o g))@s0ddlZddlZddlmZddlZddlZddlmZddlm Z ddl m Z ddl m Z dd l mZddl m Z dd lmZdd lmZd d Zd7ddZddZddZddZd8ddZd9ddZd:dd Zd;d"d#Zd$d%Zd&d'Z drrr_sqlite_file_db]s  rDc Cs\ttd}d|d}|pdtjjjv}td|d|d|r!dnd d |r(d nd d S)Nrz sqlite:///z/foo.dbr? [alembic] script_location =  sqlalchemy.url =  sourceless = truefalse zsqlalchemy.future = truea [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 rrrdb __class__ __module___write_config_file)r%r?r5r>sqlalchemy_futurerrr_sqlite_testing_configes    rRrKc CsZttd}dtjjjv}d|}td|d|d|rdndd |r$dndd |d S) Nrr?zsqlite:///%s/foo.dbrErFz sqlalchemy.future = rHrIrGzH 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 rL)r%extra_version_locationr5rQr>rrr_multi_dir_testing_configs   rT postgresqlcCs(ttd}td|d|d|dS)zLuse a postgresql url with no host so that connections guaranteed to failrz [alembic] script_location =rFz:// 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 )rrrP)dialect directivesr5rrr_no_sql_testing_configs rXcCsBt}t|jd}||Wd|S1swY|S)Nr1)rr2config_file_namer3)textr&r6rrrrPs  rPcCs8ddlm}tttjstt|ttdS)NrConfigztest_alembic.ini)alembic.configr\raccessrF_OKmkdirrr[rrrrs  rasciic Cs|j|}|j}t|}|r||}t|d }||Wdn1s*wYt |}|r;t |t ||} |j| j}|j| jkrRtd|jj| dd|rit||dkrddnddSdS)Nwbz2Can't change down_revision on a refresh operation.T)_replacerrr) revision_map get_revisionrtextwrapdedentencoder2r3rr8rr9r _from_pathrevision down_revision Exception add_revisionr") scriptdirrev_idcontentencodingr%oldrfpr:r rrr write_scripts*         rtcCsddl}|||dkr-t|}tjjd}tj |\}}||}t |||}n |dvs3Jt|}t |tj sAJt|dS)Nrr)rr) py_compilecompilerr8 importlib machineryBYTECODE_SUFFIXESrrsplitextr.mover^r_r9)rstylerur:suffixfilepathextsimple_pyc_pathrrrr"s      r"cCst}t}t}t|}|j|ddddt||d|d|j|dd|dt||d|d |d d d |j|d d|dt||d|d |d|||fS)Nz revision aTbase)refreshheadz"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") zutf-8)rqz revision cz"Rev C" revision = 'z}' from alembic import op def upgrade(): op.execute("CREATE STEP 3") def downgrade(): op.execute("DROP STEP 3") rror r#generate_revisionrt)r&abcr rrrthree_rev_fixture#sD  rcCst}t}t}t|}|j|d|dddt||d|d|d|j|d|dddt||d|d|d |j|d |dddt||d |d|d |||fS) z:Create a multiple head fixture from the three-revs fixturezrevision d from bT)rsplicerz"Rev D" revision = 'rz}' from alembic import op def upgrade(): op.execute("CREATE STEP 4") def downgrade(): op.execute("DROP STEP 4") zrevision 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") zrevision 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&rrrder6r rrrmulti_heads_fixturensR     rc Csjttd}dtjjjv}d|}ddd|D}t d|d|r)d nd d |d|d S) z?alembic.ini fixture to work exactly with the 'multidb' templaterr?z, z css&|]\}}d|d|jVqdS)[z] sqlalchemy.url = N)r>).0keyvaluerrr s  z*_multidb_testing_config..rEz( sourceless = false sqlalchemy.future = rHrIz 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 ) rrrrMrNrOjoinkeysitemsrP)r*r5rQ databasesrrr_multidb_testing_configs"   rrmorecGstt|j|S)N)strrjoinpathas_posix)rrrrrrsr)TrF)r<FN)FF)FrK)rUrK)raF)#importlib.machineryrwrpathlibrr.rfr+rrrKrrArr r r rr(r0r7r;rDrRrTrXrPrrtr"rrrrrrrrrs>          "   . +' KV1