def POST(self):
try:
# 'startsecs=0' is to avoid restarting consul after `consul leave`.
os.system(
'sed -i \'/autorestart=true/cautorestart=false\' /etc/supervisor/conf.d/consul.conf'
)
os.system('echo startsecs=0 >> /etc/supervisor/conf.d/consul.conf')
os.system('supervisorctl update consul')
# As 'supervisorctl update' will restart Consul, we have to wait for it to be running.
deadline = time.time() + 15
while time.time() < deadline:
try:
get_current_datacenter()
break
except:
time.sleep(1)
deregister_services(gethostname())
os.system('consul leave')
finally:
post_data = json.loads(web.data() or '{}')
runtime_settings_path = '/opt/armada/runtime_settings.json'
if not post_data.get('keep-joined') and os.path.isfile(
runtime_settings_path):
with open(runtime_settings_path) as f:
runtime_settings = json.load(f)
runtime_settings['ships'] = []
runtime_settings['is_commander'] = True
with open(runtime_settings_path, 'w') as f:
json.dump(runtime_settings, f, sort_keys=True, indent=4)
return self.status_ok({'message': 'Shutdown complete.'})
def _version_check():
current_version = armada_api.get('version')
if not is_valid_response(current_version):
# skip version check since we cannot determinate current version
return
with SyncOpen(VERSION_CACHE_FILE_PATH, 'r+') as f:
data = json.load(f)
displayed_timestamp = data['displayed']
cache_version = data['latest_version']
current_is_newer = StrictVersion(current_version) >= StrictVersion(
cache_version)
if current_is_newer or time.time(
) - DISPLAY_INTERVAL < displayed_timestamp:
return
message = 'You are using armada version {}, however version {} is available. ' \
'You should consider upgrading armada via "bash <(curl -sL http://armada.sh/install)"' \
.format(armada_api.get('version'), data['latest_version'])
print('\n' + message, file=sys.stderr)
data['displayed'] = time.time()
f.seek(0)
f.truncate()
json.dump(data, f)
def _save_containers_parameters_list_in_file(containers_parameters_list,
saved_containers_path):
temp_file_path = saved_containers_path + '.tmp'
with open(temp_file_path, 'w') as temp_file:
json.dump(containers_parameters_list,
temp_file,
indent=4,
sort_keys=True)
shutil.copyfile(temp_file_path, saved_containers_path)
os.remove(temp_file_path)
def main():
version = armada_api.get('version')
if not is_valid_response(version):
# skip sync if we cannot determine current version
return
r = requests.get('http://version.armada.sh/version_check', data=dict(version=version), timeout=3)
r.raise_for_status()
data = r.json()
data.update({
'displayed': 0,
'synced': time.time(),
})
with SyncOpen(VERSION_CACHE_FILE_PATH, 'w') as f:
json.dump(data, f)
def main():
version = armada_api.get('version')
if not is_valid_response(version):
# skip sync if we cannot determine current version
return
r = requests.get('http://version.armada.sh/version_check',
data=dict(version=version),
timeout=3)
r.raise_for_status()
data = r.json()
data.update({
'displayed': 0,
'synced': time.time(),
})
with SyncOpen(VERSION_CACHE_FILE_PATH, 'w') as f:
json.dump(data, f)
def on_post(self, req, resp):
try:
# 'startsecs=0' is to avoid restarting consul after `consul leave`.
check_call(
'sed -i \'/autorestart=true/cautorestart=false\' /etc/supervisor/conf.d/consul.conf',
shell=True)
check_call(
'echo startsecs=0 >> /etc/supervisor/conf.d/consul.conf',
shell=True)
check_call(['supervisorctl', 'update', 'consul'])
# As 'supervisorctl update' will restart Consul, we have to wait for it to be running.
deadline = time.time() + 15
ok = False
while time.time() < deadline:
try:
get_current_datacenter()
ok = True
break
except Exception:
time.sleep(1)
if not ok:
get_logger().warn('Restarting consul timed out.')
deregister_services(gethostname())
check_call(['consul', 'leave'])
finally:
post_data = req.json
runtime_settings_path = '/opt/armada/runtime_settings.json'
if not post_data.get('keep-joined') and os.path.isfile(
runtime_settings_path):
with open(runtime_settings_path) as f:
runtime_settings = json.load(f)
runtime_settings['ships'] = []
runtime_settings['is_commander'] = True
with open(runtime_settings_path, 'w') as f:
json.dump(runtime_settings, f, sort_keys=True, indent=4)
return self.status_ok(resp, {'message': 'Shutdown complete.'})
def _version_check():
current_version = armada_api.get('version')
if not is_valid_response(current_version):
# skip version check since we cannot determinate current version
return
with SyncOpen(VERSION_CACHE_FILE_PATH, 'r+') as f:
data = json.load(f)
displayed_timestamp = data['displayed']
cache_version = data['latest_version']
current_is_newer = StrictVersion(current_version) >= StrictVersion(cache_version)
if current_is_newer or time.time() - DISPLAY_INTERVAL < displayed_timestamp:
return
message = 'You are using armada version {}, however version {} is available. ' \
'You should consider upgrading armada via "bash <(curl -sL http://armada.sh/install)"' \
.format(armada_api.get('version'), data['latest_version'])
print('\n' + message, file=sys.stderr)
data['displayed'] = time.time()
f.seek(0)
f.truncate()
json.dump(data, f)