mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
130 lines
4.2 KiB
Python
Executable file
130 lines
4.2 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
'''Deploy docker-ci report container on Digital Ocean.
|
|
Usage:
|
|
export CONFIG_JSON='
|
|
{ "DROPLET_NAME": "Digital_Ocean_dropplet_name",
|
|
"DO_CLIENT_ID": "Digital_Ocean_client_id",
|
|
"DO_API_KEY": "Digital_Ocean_api_key",
|
|
"DOCKER_KEY_ID": "Digital_Ocean_ssh_key_id",
|
|
"DOCKER_CI_KEY_PATH": "docker-ci_private_key_path",
|
|
"DOCKER_CI_PUB": "$(cat docker-ci_ssh_public_key.pub)",
|
|
"DOCKER_CI_ADDRESS" "user@docker-ci_fqdn_server",
|
|
"SMTP_USER": "SMTP_server_user",
|
|
"SMTP_PWD": "SMTP_server_password",
|
|
"EMAIL_SENDER": "Buildbot_mailing_sender",
|
|
"EMAIL_RCP": "Buildbot_mailing_receipient" }'
|
|
python deployment.py
|
|
'''
|
|
|
|
import re, json, requests, base64
|
|
from fabric import api
|
|
from fabric.api import cd, run, put, sudo
|
|
from os import environ as env
|
|
from time import sleep
|
|
from datetime import datetime
|
|
|
|
# Populate environment variables
|
|
CONFIG = json.loads(env['CONFIG_JSON'])
|
|
for key in CONFIG:
|
|
env[key] = CONFIG[key]
|
|
|
|
# Load DOCKER_CI_KEY
|
|
env['DOCKER_CI_KEY'] = open(env['DOCKER_CI_KEY_PATH']).read()
|
|
|
|
DROPLET_NAME = env.get('DROPLET_NAME','report')
|
|
TIMEOUT = 120 # Seconds before timeout droplet creation
|
|
IMAGE_ID = 1004145 # Docker on Ubuntu 13.04
|
|
REGION_ID = 4 # New York 2
|
|
SIZE_ID = 66 # memory 512MB
|
|
DO_IMAGE_USER = 'root' # Image user on Digital Ocean
|
|
API_URL = 'https://api.digitalocean.com/'
|
|
|
|
|
|
class digital_ocean():
|
|
|
|
def __init__(self, key, client):
|
|
'''Set default API parameters'''
|
|
self.key = key
|
|
self.client = client
|
|
self.api_url = API_URL
|
|
|
|
def api(self, cmd_path, api_arg={}):
|
|
'''Make api call'''
|
|
api_arg.update({'api_key':self.key, 'client_id':self.client})
|
|
resp = requests.get(self.api_url + cmd_path, params=api_arg).text
|
|
resp = json.loads(resp)
|
|
if resp['status'] != 'OK':
|
|
raise Exception(resp['error_message'])
|
|
return resp
|
|
|
|
def droplet_data(self, name):
|
|
'''Get droplet data'''
|
|
data = self.api('droplets')
|
|
data = [droplet for droplet in data['droplets']
|
|
if droplet['name'] == name]
|
|
return data[0] if data else {}
|
|
|
|
def json_fmt(data):
|
|
'''Format json output'''
|
|
return json.dumps(data, sort_keys = True, indent = 2)
|
|
|
|
|
|
do = digital_ocean(env['DO_API_KEY'], env['DO_CLIENT_ID'])
|
|
|
|
# Get DROPLET_NAME data
|
|
data = do.droplet_data(DROPLET_NAME)
|
|
|
|
# Stop processing if DROPLET_NAME exists on Digital Ocean
|
|
if data:
|
|
print ('Droplet: {} already deployed. Not further processing.'
|
|
.format(DROPLET_NAME))
|
|
exit(1)
|
|
|
|
# Create droplet
|
|
do.api('droplets/new', {'name':DROPLET_NAME, 'region_id':REGION_ID,
|
|
'image_id':IMAGE_ID, 'size_id':SIZE_ID,
|
|
'ssh_key_ids':[env['DOCKER_KEY_ID']]})
|
|
|
|
# Wait for droplet to be created.
|
|
start_time = datetime.now()
|
|
while (data.get('status','') != 'active' and (
|
|
datetime.now()-start_time).seconds < TIMEOUT):
|
|
data = do.droplet_data(DROPLET_NAME)
|
|
print data['status']
|
|
sleep(3)
|
|
|
|
# Wait for the machine to boot
|
|
sleep(15)
|
|
|
|
# Get droplet IP
|
|
ip = str(data['ip_address'])
|
|
print 'droplet: {} ip: {}'.format(DROPLET_NAME, ip)
|
|
|
|
api.env.host_string = ip
|
|
api.env.user = DO_IMAGE_USER
|
|
api.env.key_filename = env['DOCKER_CI_KEY_PATH']
|
|
|
|
# Correct timezone
|
|
sudo('echo "America/Los_Angeles" >/etc/timezone')
|
|
sudo('dpkg-reconfigure --frontend noninteractive tzdata')
|
|
|
|
# Load JSON_CONFIG environment for Dockerfile
|
|
CONFIG_JSON= base64.b64encode(
|
|
'{{"DOCKER_CI_PUB": "{DOCKER_CI_PUB}",'
|
|
' "DOCKER_CI_KEY": "{DOCKER_CI_KEY}",'
|
|
' "DOCKER_CI_ADDRESS": "{DOCKER_CI_ADDRESS}",'
|
|
' "SMTP_USER": "{SMTP_USER}",'
|
|
' "SMTP_PWD": "{SMTP_PWD}",'
|
|
' "EMAIL_SENDER": "{EMAIL_SENDER}",'
|
|
' "EMAIL_RCP": "{EMAIL_RCP}"}}'.format(**env))
|
|
|
|
run('mkdir -p /data/report')
|
|
put('./', '/data/report')
|
|
with cd('/data/report'):
|
|
run('chmod 700 report.py')
|
|
run('echo "{}" > credentials.json'.format(CONFIG_JSON))
|
|
run('docker build -t report .')
|
|
run('rm credentials.json')
|
|
run("echo -e '30 09 * * * /usr/bin/docker run report\n' |"
|
|
" /usr/bin/crontab -")
|