Hot-keys on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

# 

#    Licensed under the Apache License, Version 2.0 (the "License"); you may 

#    not use this file except in compliance with the License. You may obtain 

#    a copy of the License at 

# 

#         http://www.apache.org/licenses/LICENSE-2.0 

# 

#    Unless required by applicable law or agreed to in writing, software 

#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 

#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 

#    License for the specific language governing permissions and limitations 

#    under the License. 

 

"""Starter script for magnum-db-manage.""" 

 

import os 

 

from oslo_config import cfg 

from oslo_db import options 

from oslo_db.sqlalchemy.migration_cli import manager 

 

from magnum.openstack.common import log as logging 

 

LOG = logging.getLogger(__name__) 

CONF = cfg.CONF 

 

 

def do_version(mgr): 

    print('Current DB revision is %s' % mgr.version()) 

 

 

def do_upgrade(mgr): 

    mgr.upgrade(CONF.command.revision) 

 

 

def do_stamp(mgr): 

    mgr.stamp(CONF.command.revision) 

 

 

def do_revision(mgr): 

    mgr.revision(message=CONF.command.message, 

                 autogenerate=CONF.command.autogenerate) 

 

 

def add_command_parsers(subparsers): 

    parser = subparsers.add_parser('version') 

    parser.set_defaults(func=do_version) 

 

    parser = subparsers.add_parser('upgrade') 

    parser.add_argument('revision', nargs='?') 

    parser.set_defaults(func=do_upgrade) 

 

    parser = subparsers.add_parser('stamp') 

    parser.add_argument('revision') 

    parser.set_defaults(func=do_stamp) 

 

    parser = subparsers.add_parser('revision') 

    parser.add_argument('-m', '--message') 

    parser.add_argument('--autogenerate', action='store_true') 

    parser.set_defaults(func=do_revision) 

 

 

def get_manager(): 

    if cfg.CONF.database.connection is None: 

        raise ValueError( 

            'Database connection not set in /etc/magnum/magnum.conf') 

 

    alembic_path = os.path.abspath( 

        os.path.join(os.path.dirname(__file__), 

                     '..', 'db', 'sqlalchemy', 'alembic.ini')) 

    migrate_path = os.path.abspath( 

        os.path.join(os.path.dirname(__file__), 

                     '..', 'db', 'sqlalchemy', 'alembic')) 

    migration_config = {'alembic_ini_path': alembic_path, 

                        'alembic_repo_path': migrate_path, 

                        'db_url': CONF.database.connection} 

    return manager.MigrationManager(migration_config) 

 

 

def main(): 

    command_opt = cfg.SubCommandOpt('command', 

                                    title='Command', 

                                    help='Available commands', 

                                    handler=add_command_parsers) 

    CONF.register_cli_opt(command_opt) 

 

    # set_defaults() is called to register the db options. 

    options.set_defaults(CONF) 

 

    CONF(project='magnum') 

    CONF.command.func(get_manager())