148 lines
3.1 KiB
Python
148 lines
3.1 KiB
Python
|
import os
|
||
|
|
import os.path
|
||
|
|
|
||
|
|
import jinja2
|
||
|
|
|
||
|
|
QUAYPATH = os.getenv("QUAYPATH", ".")
|
||
|
|
QUAYDIR = os.getenv("QUAYDIR", "/")
|
||
|
|
QUAYCONF_DIR = os.getenv("QUAYCONF", os.path.join(QUAYDIR, QUAYPATH, "conf"))
|
||
|
|
|
||
|
|
QUAY_SERVICES = os.getenv("QUAY_SERVICES", [])
|
||
|
|
QUAY_OVERRIDE_SERVICES = os.getenv("QUAY_OVERRIDE_SERVICES", [])
|
||
|
|
|
||
|
|
|
||
|
|
def default_services():
|
||
|
|
return {
|
||
|
|
"blobuploadcleanupworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"buildlogsarchiver": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"builder": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"chunkcleanupworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"expiredappspecifictokenworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"exportactionlogsworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"gcworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"globalpromstats": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"labelbackfillworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"logrotateworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"namespacegcworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"notificationworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"queuecleanupworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"repositoryactioncounter": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"security_notification_worker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"securityworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"storagereplication": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"tagbackfillworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"teamsyncworker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"dnsmasq": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"gunicorn-registry": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"gunicorn-secscan": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"gunicorn-verbs": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"gunicorn-web": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"ip-resolver-update-worker": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"jwtproxy": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"memcache": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"nginx": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"prometheus-aggregator": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"servicekey": {
|
||
|
|
"autostart": "true"
|
||
|
|
},
|
||
|
|
"repomirrorworker": {
|
||
|
|
"autostart": "false"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def generate_supervisord_config(filename, config):
|
||
|
|
with open(filename + ".jnj") as f:
|
||
|
|
template = jinja2.Template(f.read())
|
||
|
|
rendered = template.render(config=config)
|
||
|
|
|
||
|
|
with open(filename, 'w') as f:
|
||
|
|
f.write(rendered)
|
||
|
|
|
||
|
|
|
||
|
|
def limit_services(config, enabled_services):
|
||
|
|
if enabled_services == []:
|
||
|
|
return
|
||
|
|
|
||
|
|
for service in config.keys():
|
||
|
|
if service in enabled_services:
|
||
|
|
config[service]["autostart"] = "true"
|
||
|
|
else:
|
||
|
|
config[service]["autostart"] = "false"
|
||
|
|
|
||
|
|
|
||
|
|
def override_services(config, override_services):
|
||
|
|
if override_services == []:
|
||
|
|
return
|
||
|
|
|
||
|
|
for service in config.keys():
|
||
|
|
if service + "=true" in override_services:
|
||
|
|
config[service]["autostart"] = "true"
|
||
|
|
elif service + "=false" in override_services:
|
||
|
|
config[service]["autostart"] = "false"
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
config = default_services()
|
||
|
|
limit_services(config, QUAY_SERVICES)
|
||
|
|
override_services(config, QUAY_OVERRIDE_SERVICES)
|
||
|
|
generate_supervisord_config(os.path.join(QUAYCONF_DIR, 'supervisord.conf'), config)
|