Statistics
| Branch: | Tag: | Revision:

root / snf-pithos-backend / pithos / backends / migrate.py @ 4a7b190f

History | View | Annotate | Download (3.7 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
                                            quotaholder_serials)
55

    
56
os.environ['DJANGO_SETTINGS_MODULE'] = 'synnefo.settings'
57

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

    
69
import sqlalchemy as sa
70

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

    
75

    
76
def initialize_db(dbconnection):
77
    alembic_cfg = Config(DEFAULT_ALEMBIC_INI_PATH)
78

    
79
    db = alembic_cfg.get_main_option("sqlalchemy.url", dbconnection)
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
    quotaholder_serials.create_tables(engine)
91

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

    
96

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

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

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

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

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