Statistics
| Branch: | Tag: | Revision:

root / snf-pithos-backend / pithos / backends / migrate.py @ 8ee4ec50

History | View | Annotate | Download (3.6 kB)

1
# Copyright 2011-2012 GRNET S.A. All rights reserved.
2
#
3
# Redistribution and use in source and binary forms, with or
4
# without modification, are permitted provided that the following
5
# conditions are met:
6
#
7
#   1. Redistributions of source code must retain the above
8
#      copyright notice, this list of conditions and the following
9
#      disclaimer.
10
#
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.
15
#
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.
28
#
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.
33

    
34
"""
35
Alembic migration wrapper for pithos backend database.
36

37
- Locate alembic.ini in backends/lib/sqlalchemy package and pass it
38
  as parameter to alembic
39

40
e.g::
41

42
    $ pithos-migrate upgrade head
43

44
"""
45

    
46
import sys
47
import os
48

    
49
from alembic.config import main as alembic_main, Config
50
from alembic import context, command
51

    
52
from pithos.backends.lib import sqlalchemy as sqlalchemy_backend
53
from pithos.backends.lib.sqlalchemy import node, groups, public, xfeatures
54

    
55
try:
56
    # pithos-app case
57
    from synnefo.settings import PITHOS_BACKEND_DB_CONNECTION
58
except ImportError:
59
    try:
60
        # plankton case
61
        from synnefo.settings import BACKEND_DB_CONNECTION as \
62
            PITHOS_BACKEND_DB_CONNECTION
63
    except ImportError:
64
        PITHOS_BACKEND_DB_CONNECTION = None
65

    
66
import sqlalchemy as sa
67

    
68
DEFAULT_ALEMBIC_INI_PATH = os.path.join(
69
        os.path.abspath(os.path.dirname(sqlalchemy_backend.__file__)),
70
        'alembic.ini')
71

    
72

    
73
def initialize_db():
74
    alembic_cfg = Config(DEFAULT_ALEMBIC_INI_PATH)
75

    
76
    db = alembic_cfg.get_main_option("sqlalchemy.url", PITHOS_BACKEND_DB_CONNECTION)
77
    alembic_cfg.set_main_option("sqlalchemy.url", db)
78

    
79
    engine = sa.engine_from_config(
80
                alembic_cfg.get_section(alembic_cfg.config_ini_section),
81
                prefix='sqlalchemy.')
82

    
83
    node.create_tables(engine)
84
    groups.create_tables(engine)
85
    public.create_tables(engine)
86
    xfeatures.create_tables(engine)
87

    
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")
91

    
92

    
93

    
94
def main(argv=None, **kwargs):
95
    if not argv:
96
        argv = sys.argv
97

    
98
    # clean up args
99
    argv.pop(0)
100

    
101
    if len(argv) >= 1 and argv[0] == 'initdb':
102
        print "Initializing db."
103
        initialize_db()
104
        print "DB initialized."
105
        exit(1)
106

    
107

    
108
    # default config arg, if not already set
109
    if not '-c' in argv:
110
        argv.insert(0, DEFAULT_ALEMBIC_INI_PATH)
111
        argv.insert(0, '-c')
112

    
113

    
114
    alembic_main(argv, **kwargs)
115
if __name__ == '__main__':
116
    import sys
117
    main(sys.argv)
118