Revision d50ed8d4 snf-pithos-backend/pithos/backends/migrate.py
| b/snf-pithos-backend/pithos/backends/migrate.py | ||
|---|---|---|
| 66 | 66 |
import sqlalchemy as sa |
| 67 | 67 |
|
| 68 | 68 |
DEFAULT_ALEMBIC_INI_PATH = os.path.join( |
| 69 |
os.path.abspath(os.path.dirname(sqlalchemy_backend.__file__)),
|
|
| 70 |
'alembic.ini')
|
|
| 69 |
os.path.abspath(os.path.dirname(sqlalchemy_backend.__file__)), |
|
| 70 |
'alembic.ini') |
|
| 71 | 71 |
|
| 72 | 72 |
|
| 73 | 73 |
def initialize_db(): |
| 74 | 74 |
alembic_cfg = Config(DEFAULT_ALEMBIC_INI_PATH) |
| 75 | 75 |
|
| 76 |
db = alembic_cfg.get_main_option("sqlalchemy.url", PITHOS_BACKEND_DB_CONNECTION)
|
|
| 76 |
db = alembic_cfg.get_main_option( |
|
| 77 |
"sqlalchemy.url", PITHOS_BACKEND_DB_CONNECTION) |
|
| 77 | 78 |
alembic_cfg.set_main_option("sqlalchemy.url", db)
|
| 78 | 79 |
|
| 79 | 80 |
engine = sa.engine_from_config( |
| 80 |
alembic_cfg.get_section(alembic_cfg.config_ini_section),
|
|
| 81 |
prefix='sqlalchemy.')
|
|
| 81 |
alembic_cfg.get_section(alembic_cfg.config_ini_section), |
|
| 82 |
prefix='sqlalchemy.') |
|
| 82 | 83 |
|
| 83 | 84 |
node.create_tables(engine) |
| 84 | 85 |
groups.create_tables(engine) |
| ... | ... | |
| 90 | 91 |
command.stamp(alembic_cfg, "head") |
| 91 | 92 |
|
| 92 | 93 |
|
| 93 |
|
|
| 94 | 94 |
def main(argv=None, **kwargs): |
| 95 | 95 |
if not argv: |
| 96 | 96 |
argv = sys.argv |
| ... | ... | |
| 104 | 104 |
print "DB initialized." |
| 105 | 105 |
exit(1) |
| 106 | 106 |
|
| 107 |
|
|
| 108 | 107 |
# default config arg, if not already set |
| 109 | 108 |
if not '-c' in argv: |
| 110 | 109 |
argv.insert(0, DEFAULT_ALEMBIC_INI_PATH) |
| 111 | 110 |
argv.insert(0, '-c') |
| 112 | 111 |
|
| 113 |
|
|
| 114 | 112 |
alembic_main(argv, **kwargs) |
| 115 | 113 |
if __name__ == '__main__': |
| 116 | 114 |
import sys |
| 117 | 115 |
main(sys.argv) |
| 118 |
|
|
Also available in: Unified diff