Skip to content
Snippets Groups Projects
Commit 1275a63a authored by Stéphane Diemer's avatar Stéphane Diemer
Browse files

Continued script to set domain (refs #21163).

parent 069a3048
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 65 additions and 24 deletions
set_app_domain.py
+ 65
− 24
View file @ 1275a63a
......@@ -64,13 +64,14 @@ class SetAppDomain():
sys.exit(1)
if args:
app = args.pop()
# Change domain
# Get Nginx conf path and instance user
instance = None
if app == 'cm':
nginx_conf = '/etc/nginx/sites-available/skyreach.conf'
self.change_cm_domain(new_domain)
key = 'CM_SERVER_NAME'
elif app == 'mon':
nginx_conf = '/etc/nginx/sites-available/msmonitor.conf'
key = 'MONITOR_SERVER_NAME'
elif app == 'ms' or app.startswith('ms-'):
if app.startswith('ms-'):
instance = app[3:].strip('. -\t\n')
......@@ -80,13 +81,25 @@ class SetAppDomain():
else:
instance = 'msuser'
nginx_conf = '/etc/nginx/sites-available/mediaserver-%s.conf' % instance
self.change_ms_domain(new_domain, instance)
key = 'MS_SERVER_NAME'
else:
log('Invalid app name requested.')
log(self.USAGE)
sys.exit(1)
# Change domain
self.change_nginx_domain(nginx_conf, new_domain)
self.change_hosts_file(new_domain)
self.change_envsetup_conf(key, new_domain)
if app == 'cm':
warning = self.change_cm_domain(new_domain)
elif app == 'mon':
warning = None
else:
warning = self.change_ms_domain(new_domain, instance)
log('\033[92mDone\033[0m')
if warning:
log('\033[93mWarning:\033[0m')
log(warning)
sys.exit(0)
def change_nginx_domain(self, path, new_domain):
......@@ -100,10 +113,9 @@ class SetAppDomain():
changed_lines = 0
for line in vhost.split('\n'):
if re.match(r'\s*server_name\s+([\w\-\_\.\ ]+);', line):
new_vhost += re.sub(r'server_name\s+([\w\-\_\.\ ]+);', line, 'server_name %s;' % new_domain) + '\n'
line = re.sub(r'server_name\s+([\w\-\_\.\ ]+);', 'server_name %s;' % new_domain, line)
changed_lines += 1
else:
new_vhost += line + '\n'
new_vhost += line + '\n'
if changed_lines != 2:
log('Warning the number of server_name occurence changed in Nginx configuration is not the expected number (2) but is %s.' % changed_lines)
if not self.force:
......@@ -128,6 +140,7 @@ class SetAppDomain():
def change_hosts_file(self, new_domain):
with open('/etc/hosts', 'r') as fo:
hosts = fo.read()
ori_hosts = hosts
if ' ' + new_domain in hosts:
hosts = hosts.replace(' ' + new_domain, '')
elif '\t' + new_domain in hosts:
......@@ -138,17 +151,39 @@ class SetAppDomain():
new_hosts += '\n' + line
if line.startswith('127.0.0.1'):
new_hosts += ' ' + new_domain
if new_hosts != hosts:
if new_hosts != ori_hosts:
if self.debug:
log('New hosts:')
log(new_hosts)
else:
with open('/etc/hosts', 'w') as fo:
fo.write(new_hosts)
log('The hosts file has been update.')
log('The "/etc/hosts" file has been update.')
utils.run_commands(['service nscd restart'])
else:
log('The hosts file is already up to date.')
log('The "/etc/hosts" file is already up to date.')
def change_envsetup_conf(self, key, new_domain):
confs = (
'conf.sh',
'auto-generated-conf.sh',
)
replaced = False
for path in confs:
if os.path.exists(path):
with open(path, 'r') as fo:
content = fo.read()
if key in content:
content = re.sub(r'%s=.*' % key, '%s=\'%s\'' % (key, new_domain), content)
with open(path, 'w') as fo:
fo.write(content)
replaced = True
log('Envsetup configration file "%s" updated.' % path)
break
if not replaced:
with open(confs[0], 'a') as fo:
fo.write('\n%s=\'%s\'' % (key, new_domain))
log('Envsetup configration file "%s" updated.' % confs[0])
def change_ms_domain(self, new_domain, instance):
try:
......@@ -158,24 +193,29 @@ class SetAppDomain():
log('Assuming that the new url is using HTTPS: "%s"' % new_url)
cmds = [
# set site url in site settings
'python3', os.path.join(ms_path, 'scripts', 'mssiteconfig.py'), instance, 'site_url="%s"' % new_url,
'python3 %s %s site_url="%s"' % (os.path.join(ms_path, 'scripts', 'mssiteconfig.py'), instance, new_url),
# reset all local resources managers
'python3', os.path.join(ms_path, 'scripts', 'reset_service_resources.py'), instance, 'local',
# change configuration of worker on MS side
'sed -i "s@\'url\': \'\\(http[s]*\\)://.*\'@\'url\': \'\\1://%s\'@" /etc/celerity/config.py' % new_domain,
'python3 %s %s local' % (os.path.join(ms_path, 'scripts', 'reset_service_resources.py'), instance),
# change configuration of celerity on MS side
'sed -i "s@\'url\': \'\\(http[s]*\\)://.*\',@\'url\': \'\\1://%s\',@" /etc/celerity/config.py' % new_domain,
'sed -i "s@SERVER_URL.*@SERVER_URL = \'%s:6200\'@" /etc/celerity/config.py' % new_url,
# restart ms
'mscontroller.py restart -u %s' % instance,
]
worker_ips = utils.get_conf('CELERITY_WORKER_IP') or '127.0.1.1'
# change configuration of worker on worker side
# change configuration of celerity on worker side
for worker_ip in worker_ips.split(','):
cmds.append('ssh root@%s sed -i "s@\'url\': \'\\(http[s]*\\)://.*\'@\'url\': \'\\1://%s\'@" /etc/celerity/config.py' % (worker_ip, new_domain))
cmds.append('rsync -avz /etc/celerity/config.py root@%s:/etc/celerity/config.py' % worker_ip)
cmds.append('ssh root@%s service celerity-workers restart' % worker_ip)
utils.run_commands(cmds)
except Exception as e:
log('Unable to set MS domain:\n%s' % e)
log('Unable to set domain in MS database and Celerity config:\n%s' % e)
sys.exit(1)
else:
log('MS domain changed')
log('Some steps to change the domain should be done manually:')
log(' - Change the domain of MS used in CM stations configuration.')
log('Domain changed in MS database and Celerity config.')
msg = 'Some steps to change the domain should be done manually:'
msg += '\n - Change the domain of MS used in CM stations configuration.'
return msg
def change_cm_domain(self, new_domain):
try:
......@@ -183,16 +223,17 @@ class SetAppDomain():
log('Assuming that the new url is using HTTPS: "%s"' % new_url)
cmds = [
# set site url in site settings
'echo \'from skyreach_site.base.models import SiteSettings; ss = SiteSettings.get_singleton(); ss.url = "%s"; ss.save(); print("site settings saved")\' | su skyreach -c "python3 /home/skyreach/htdocs/skyreach_site/manage.py shell --plain --no-startup"' % new_url,
'echo \'from skyreach_site.base.models import SiteSettings; ss = SiteSettings.get_singleton(); ss.url = "%s"; ss.save(); print("Site settings saved.")\' | su skyreach -c "python3 /home/skyreach/htdocs/skyreach_site/manage.py shell --plain --no-startup"' % new_url,
]
utils.run_commands(cmds)
except Exception as e:
log('Unable to set CM domain:\n%s' % e)
log('Unable to set domain in CM database:\n%s' % e)
sys.exit(1)
else:
log('CM domain changed')
log('Some steps to change the domain should be done manually:')
log(' - Change the url of CM in the related MS.')
log('Domain changed in CM database.')
msg = 'Some steps to change the domain should be done manually:'
msg += '\n - Change the url of CM in the related MS.'
return msg
if __name__ == '__main__':
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment