forked from YellLabs/yellfabric
/
operations.py
175 lines (127 loc) · 4.74 KB
/
operations.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
162
163
164
165
166
167
168
169
170
171
172
173
174
175
import java
import python
import glassfish
import play
import static
import utils
import csv
import os.path
import pprint
import glob
from fabric.api import env, require, runs_once, local
from fabric.utils import abort
from fabric.contrib.project import rsync_project
from fabric.operations import prompt
@runs_once
def fab_setup_paths():
"""
Wrapper for setting up language dependent paths.
"""
require("lang")
if env.lang in ["django", "python"]:
python.setup_paths()
elif env.lang in ["tomcat", "java"]:
java.setup_paths()
elif env.lang == "glassfish":
glassfish.setup_paths()
elif env.lang == "play":
play.setup_paths()
elif env.lang == "static":
static.setup_paths()
else:
abort("Project language %r unknown" % env.lang)
def scm_echo_info():
require("scm_type")
pprint.pprint(utils.scm_get_info(env.scm_type))
def rsync_from_local():
"""
Push a local checkout of the code to a remote machine.
"""
require("tempdir", "project_path", "sudo_user")
rsync_exclude = ["*.pyc"]
rsync_opts = []
if env.get("rsync_exclude"):
rsync_exclude = rsync_exclude + env.rsync_exclude
if env.sudo_user:
rsync_opts = '--rsync-path="sudo -u %s rsync"' % env.sudo_user
rsync_project(
local_dir="%s/" % env.tempdir,
remote_dir="%s/" % env.project_path,
exclude=rsync_exclude,
delete=True,
extra_opts=rsync_opts)
@runs_once
def use_maven_build():
require("war_path", provided_by="setup_paths")
try:
env.war_file = glob.glob("target/*.war")[0]
env.app_config_archive = glob.glob("target/*-config.tar.gz")[0]
except IndexError:
sys.exit("Failed to find maven build products in target directory")
@runs_once
def fetch_from_repo():
# repo_base should be something like
# http://cis-dev.local:8080/artifactory/libs-release-local/com/yelllabs
require("project_name")
prompt("Base URL for this project:", "repo_base")
prompt("Version:", "proj_version")
fetch = {}
fetch[env.war_file] = \
"%(repo_base)s/%(project_name)s/%(proj_version)s/" \
"%(project_name)s-%(proj_version)s.war" % env
fetch[env.app_config_archive] = \
"%(repo_base)s/%(project_name)s/%(proj_version)s/" \
"%(project_name)s-%(proj_version)s-config.tar.gz" % env
if env.get('has_sql_archive'):
fetch[env.sql_archive] = \
"%(repo_base)s/%(project_name)s/%(proj_version)s/" \
"%(project_name)s-%(proj_version)s-sql.tar.gz" % env
for name, url in fetch.iteritems():
local("wget -O%s '%s'" % (name, url))
def fetch_render_copy(ref=None, debug=False, dirty=False, copy_remote=False, build_local_cmd=None):
"""
Fetch source code, render settings file, push remotely and delete checkout.
"""
require("scm_type", "scm_url", "config_source", "config_target", "settings_vars")
env.tempdir = utils.fetch_source(env.scm_type, env.scm_url, ref, dirty)
config_source = os.path.join(env.tempdir, env.config_source)
config_target = os.path.join(env.tempdir, env.config_target)
utils.render_settings_template(config_source, config_target, env.settings_vars, debug)
utils.render_custom_templates(env.tempdir, env.settings_vars, debug)
# Don't try to handle any errors here - the deploy should fail.
if build_local_cmd:
build_local_cmd(env.tempdir)
if copy_remote:
rsync_from_local()
utils.delete_source_conditional(env.tempdir, dirty)
def render_settings_template(debug=False):
"""
Render settings files - parameters will be set based upon parms in fabfile
"""
utils.render_settings_template(env.config_source, env.config_target, env.settings_vars, debug)
utils.render_custom_templates(".", env.settings_vars, debug)
def load_extdata(filename):
"""
Populate env.extdata as a dictionary of Puppet's extlookup/extdata
key+value entries for later use by extlookup().
Takes a single CSV file as a mandatory argument.
"""
if not os.path.exists(filename):
abort("File %r not found" % filename)
extdata = csv.reader(open(filename))
env.extdata = {}
for row in extdata:
if len(row) < 1:
continue
env.extdata[row[0]] = row[1:]
def extlookup(key):
"""
Lookup a password from by Puppet's extlookup/extdata store.
This requires that env.extdata has been populated by load_extdata()
Currently returns a single value that is only suitable for passwords.
"""
require("extdata", provided_by="load_extdata")
value = env.extdata.get(key, [])
if len(value) != 1:
abort("Could not extlookup single value for key %r" % key)
return value[0]