Statistics
| Branch: | Tag: | Revision:

root / snf-pithos-backend / pithos / backends / migrate.py @ c4399e62

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
os.environ['DJANGO_SETTINGS_MODULE'] = 'synnefo.settings'
56

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

    
68
import sqlalchemy as sa
69

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

    
74

    
75
def initialize_db():
76
    alembic_cfg = Config(DEFAULT_ALEMBIC_INI_PATH)
77

    
78
    db = alembic_cfg.get_main_option(
79
        "sqlalchemy.url", PITHOS_BACKEND_DB_CONNECTION)
80
    alembic_cfg.set_main_option("sqlalchemy.url", db)
81

    
82
    engine = sa.engine_from_config(
83
        alembic_cfg.get_section(alembic_cfg.config_ini_section),
84
        prefix='sqlalchemy.')
85

    
86
    node.create_tables(engine)
87
    groups.create_tables(engine)
88
    public.create_tables(engine)
89
    xfeatures.create_tables(engine)
90

    
91
    # then, load the Alembic configuration and generate the
92
    # version table, "stamping" it with the most recent rev:
93
    command.stamp(alembic_cfg, "head")
94

    
95

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

    
100
    # clean up args
101
    argv.pop(0)
102

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

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

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