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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
#!/usr/bin/python
# -*- coding: utf-8 -*-
DOCUMENTATION = '''
---
module: heat_manage
short_description: Initialize Orchestration (heat) database
description: Create the tables for the database backend used by heat
options:
action:
description:
- action to perform. Currently only dbysnc is supported
required: true
conf:
description:
- path to keystone config file.
required: false
default: /etc/heat/heat.conf
requirements: [ python-heatclient ]
author: Gauvain Pocentek
'''
EXAMPLES = '''
heat_manage: action=dbsync
'''
import subprocess
try:
import heat
from heat.db.sqlalchemy import migration
from migrate.versioning import api as versioning_api
from heat.db import api
from oslo.config import cfg
except ImportError:
heat_found = False
else:
heat_found = True
def will_db_change(conf):
""" Check if the database version will change after the sync.
conf is the path to the heat config file
"""
# Load the config file options
cfg.CONF(project='heat', default_config_files=[conf])
try:
current_version = migration.db_version()
except TypeError: # juno
current_version = api.db_version(api.get_engine())
repo_path = os.path.join(os.path.dirname(heat.__file__),
'db', 'sqlalchemy', 'migrate_repo')
repo_version = versioning_api.repository.Repository(repo_path).latest
return current_version != repo_version
def do_dbsync():
"""Do the dbsync. Returns (returncode, stdout, stderr)"""
# We call heat-manage db_sync on the shell rather than trying to
# do this in Python since we have no guarantees about changes to the
# internals.
args = ['heat-manage', 'db_sync']
call = subprocess.Popen(args, shell=False,
stdout=subprocess.PIPE, stderr=subprocess.PIPE)
out, err = call.communicate()
return (call.returncode, out, err)
def main():
module = AnsibleModule(
argument_spec=dict(
action=dict(required=True),
conf=dict(required=False, default="/etc/heat/heat.conf")
),
supports_check_mode=True
)
if not heat_found:
module.fail_json(msg="python-heatclient could not be found")
action = module.params['action']
conf = module.params['conf']
if action not in ['dbsync', 'db_sync']:
module.fail_json(msg="Only supported action is 'dbsync'")
changed = will_db_change(conf)
if module.check_mode:
module.exit_json(changed=changed)
(res, stdout, stderr) = do_dbsync()
if res == 0:
module.exit_json(changed=changed, stdout=stdout, stderr=stderr)
else:
module.fail_json(msg="heat-manage returned non-zero value: %d" % res,
stdout=stdout, stderr=stderr)
from ansible.module_utils.basic import *
main()
|
