/
aegir_non_package_upgrade.py
executable file
·161 lines (134 loc) · 6.32 KB
/
aegir_non_package_upgrade.py
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
#! /usr/bin/env python
"""Aegir upgrade testing script
Valid options:
--aegir_version - The branch of aegir to test (required)
--drush_version - The version of Drush to use in the testing (required)
--upgrade_version - The version of Aegir to upgrade to (required)
--test_type - This will use the jenkins tests by default, specify 'provision' to use the provision tests.
--help - Print this help
"""
from libcloud.types import Provider
from libcloud.providers import get_driver
from libcloud.deployment import MultiStepDeployment, ScriptDeployment, SSHKeyDeployment
from libcloud.ssh import SSHClient, ParamikoSSHClient
from aegir_common import Usage, dependency_check, fab_prepare_firewall, fab_prepare_apache, fab_prepare_user, fab_fetch_drush, fab_fetch_provision, fab_install_platform, fab_install_site, run_platform_tests, run_site_tests, run_provision_tests, fab_run_dispatch, fab_hostmaster_install
import os, sys, string, ConfigParser, socket, getopt
import fabric.api as fabric
import time
# Fetch some values from the config file
config = ConfigParser.RawConfigParser()
config.read(os.path.expanduser("~/frigg.ini"))
# Try to abstract the provider here, as we may end up supporting others
# Theoretically since we are using libcloud, it should support any
# provider that supports the deploy_node function (Amazon EC2 doesn't)
provider = config.get('Aegir', 'provider')
provider_driver = config.get(provider, 'driver')
# API credentials
user = config.get(provider, 'user')
key = config.get(provider, 'key')
# Preferred image and size
config_distro = config.get(provider, 'distro')
config_size = config.get(provider, 'size')
# These are used as options to Aegir during install
email = config.get('Aegir', 'email')
# Fabric command to fetch the upgrade.sh aegir script
def fab_fetch_upgrade_script(upgrade_version = 'HEAD'):
print "===> Fetching the Aegir upgrade script"
fabric.run("wget -O /tmp/upgrade.sh 'http://drupalcode.org/project/provision.git/blob_plain/%s:/upgrade.sh.txt'" % (upgrade_version), pty=True)
# Fabric command to run the upgrade.sh aegir script
def fab_run_upgrade_script(domain):
print "===> Running the Aegir upgrade script"
fabric.run("su - -s /bin/sh aegir -c 'yes | sh /tmp/upgrade.sh %s'" % domain, pty=True)
fab_run_dispatch()
def main(argv=None):
if argv is None:
argv = sys.argv
try:
try:
opts, args = getopt.getopt(argv[1:], "h", ["help", "aegir_version=", "drush_version=", "upgrade_version=", "test_type="])
except getopt.error, msg:
raise Usage(msg)
# process options
aegir_version = None
drush_version = None
upgrade_version = None
test_type = 'jenkins'
for o, a in opts:
if o in ("-h", "--help"):
print __doc__
return 0
if o in ("--aegir_version"):
aegir_version = a
if o in ("--drush_version"):
drush_version = a
if o in ("--upgrade_version"):
upgrade_version = a
if o in ("--test_type"):
test_type = a
# Check the command line options
if aegir_version is None:
raise Usage, "the --aegir_version option must be specified"
if drush_version is None:
raise Usage, "the --drush_version option must be specified"
if upgrade_version is None:
raise Usage, "the --upgrade_version option must be specified"
# Run some tests
dependency_check()
# Make a new connection
Driver = get_driver( getattr(Provider, provider_driver) )
conn = Driver(user, key)
# Get a list of the available images and sizes
images = conn.list_images()
sizes = conn.list_sizes()
# We'll use the distro and size from the config ini
preferred_image = [image for image in images if config_distro in image.name]
assert len(preferred_image) == 1, "We found more than one image for %s, will be assuming the first one" % config_distro
preferred_size = [size for size in sizes if config_size in size.name]
# The MySQL root password is hardcoded here for now, as it's in our Squeeze LAMP image.
mysqlpass = "8su43x"
# Commands to run immediately after installation
dispatch = [
SSHKeyDeployment(open(os.path.expanduser("~/.ssh/id_rsa.pub")).read()),
]
msd = MultiStepDeployment(dispatch)
# Create and deploy a new server now, and run the deployment steps defined above
print "Provisioning server and running deployment processes"
try:
node = conn.deploy_node(name='aegir' + os.environ['BUILD_ID'], image=preferred_image[0], size=preferred_size[0], deploy=msd)
except:
e = sys.exc_info()[1]
raise SystemError(e)
print "Provisioning complete, you can ssh as root to %s" % node.public_ip[0]
if node.extra.get('password'):
print "The root user's password is %s" % node.extra.get('password')
# Setting some parameters for fabric
domain = socket.getfqdn(node.public_ip[0])
fabric.env.host_string = domain
fabric.env.user = 'root'
try:
fab_prepare_firewall()
fab_prepare_apache()
fab_prepare_user()
fab_fetch_drush(drush_version)
fab_fetch_provision('package', aegir_version)
fab_hostmaster_install(domain, email, mysqlpass)
fab_fetch_upgrade_script(upgrade_version)
fab_run_upgrade_script(domain)
if test_type == 'jenkins':
run_platform_tests()
run_site_tests()
else:
run_provision_tests()
except:
print "===> Test failure"
raise
finally:
print "===> Destroying this node"
conn.destroy_node(node)
return 0
except Usage, err:
print >>sys.stderr, err.msg
print >>sys.stderr, "for help use --help"
return 2
if __name__ == "__main__":
sys.exit(main())