1 # Copyright 2011-2012 GRNET S.A. All rights reserved.
3 # Redistribution and use in source and binary forms, with or
4 # without modification, are permitted provided that the following
7 # 1. Redistributions of source code must retain the above
8 # copyright notice, this list of conditions and the following
11 # 2. Redistributions in binary form must reproduce the above
12 # copyright notice, this list of conditions and the following
13 # disclaimer in the documentation and/or other materials
14 # provided with the distribution.
16 # THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
17 # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19 # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
20 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
23 # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
24 # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
26 # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27 # POSSIBILITY OF SUCH DAMAGE.
29 # The views and conclusions contained in the software and
30 # documentation are those of the authors and should not be
31 # interpreted as representing official policies, either expressed
32 # or implied, of GRNET S.A.
35 Alembic migration wrapper for pithos backend database.
37 - Locate alembic.ini in backends/lib/sqlalchemy package and pass it
38 as parameter to alembic
42 $ pithos-migrate upgrade head
49 from alembic.config import main as alembic_main, Config
50 from alembic import context, command
52 from pithos.backends.lib import sqlalchemy as sqlalchemy_backend
53 from pithos.backends.lib.sqlalchemy import node, groups, public, xfeatures
57 from synnefo.settings import PITHOS_BACKEND_DB_CONNECTION
61 from synnefo.settings import BACKEND_DB_CONNECTION as \
62 PITHOS_BACKEND_DB_CONNECTION
64 PITHOS_BACKEND_DB_CONNECTION = None
66 import sqlalchemy as sa
68 DEFAULT_ALEMBIC_INI_PATH = os.path.join(
69 os.path.abspath(os.path.dirname(sqlalchemy_backend.__file__)),
74 alembic_cfg = Config(DEFAULT_ALEMBIC_INI_PATH)
76 db = alembic_cfg.get_main_option("sqlalchemy.url", PITHOS_BACKEND_DB_CONNECTION)
77 alembic_cfg.set_main_option("sqlalchemy.url", db)
79 engine = sa.engine_from_config(
80 alembic_cfg.get_section(alembic_cfg.config_ini_section),
83 node.create_tables(engine)
84 groups.create_tables(engine)
85 public.create_tables(engine)
86 xfeatures.create_tables(engine)
88 # then, load the Alembic configuration and generate the
89 # version table, "stamping" it with the most recent rev:
90 command.stamp(alembic_cfg, "head")
94 def main(argv=None, **kwargs):
101 if len(argv) >= 1 and argv[0] == 'initdb':
102 print "Initializing db."
104 print "DB initialized."
108 # default config arg, if not already set
110 argv.insert(0, DEFAULT_ALEMBIC_INI_PATH)
114 alembic_main(argv, **kwargs)
115 if __name__ == '__main__':