diff options
Diffstat (limited to 'lavalab-gen.py')
| -rwxr-xr-x | lavalab-gen.py | 86 |
1 files changed, 81 insertions, 5 deletions
diff --git a/lavalab-gen.py b/lavalab-gen.py index c2c5347..539d80c 100755 --- a/lavalab-gen.py +++ b/lavalab-gen.py @@ -64,7 +64,15 @@ template_settings_conf = string.Template(""" "ALLOWED_HOSTS": [ $allowed_hosts ], "CSRF_TRUSTED_ORIGINS": ["$lava_http_fqdn"], "CSRF_COOKIE_SECURE": $cookie_secure, - "SESSION_COOKIE_SECURE": $session_cookie_secure + "SESSION_COOKIE_SECURE": $session_cookie_secure, + "SERVER_EMAIL": "$server_email", + "EMAIL_HOST": "$email_host", + "EMAIL_HOST_USER": "$email_host_user", + "EMAIL_HOST_PASSWORD": "$email_host_password", + "EMAIL_PORT": $email_port, + "EMAIL_USE_TLS": $email_use_tls, + "EMAIL_USE_SSL": $email_use_ssl, + "EMAIL_BACKEND": "$email_backend" } """) @@ -105,7 +113,7 @@ def main(): else: masters = workers["masters"] for master in masters: - keywords_master = [ "name", "type", "host", "users", "groups", "tokens", "webadmin_https", "persistent_db", "zmq_auth", "zmq_auth_key", "zmq_auth_key_secret", "http_fqdn", "slave_keys", "slaveenv", "loglevel", "allowed_hosts", "lava-coordinator", "healthcheck_url" ] + keywords_master = [ "name", "type", "host", "users", "groups", "tokens", "webadmin_https", "persistent_db", "zmq_auth", "zmq_auth_key", "zmq_auth_key_secret", "http_fqdn", "slave_keys", "slaveenv", "loglevel", "allowed_hosts", "lava-coordinator", "healthcheck_url", "smtp" ] for keyword in master: if not keyword in keywords_master: print("WARNING: unknown keyword %s" % keyword) @@ -185,8 +193,60 @@ def main(): f_fqdn = open("%s/lava_http_fqdn" % workerdir, 'w') f_fqdn.write(lava_http_fqdn) f_fqdn.close() + # DJANGO defaults + email_host = "localhost" + email_host_user = "" + email_host_password = "" + email_port = 25 + email_use_tls = 'false' + email_use_ssl = 'false' + email_backend = 'django.core.mail.backends.smtp.EmailBackend' + server_email = "root@localhost" + if "smtp" in worker: + if "server_email" in worker["smtp"]: + server_email = worker["smtp"]["server_email"] + if "email_host" in worker["smtp"]: + email_host = worker["smtp"]["email_host"] + if "email_host_user" in worker["smtp"]: + email_host_user = worker["smtp"]["email_host_user"] + if "email_host_password" in worker["smtp"]: + email_host_password = worker["smtp"]["email_host_password"] + if "email_port" in worker["smtp"]: + email_port = worker["smtp"]["email_port"] + # django does not like True or False but want true/false (no upper case) + if "email_use_tls" in worker["smtp"]: + email_use_tls = worker["smtp"]["email_use_tls"] + if isinstance(email_use_tls, bool): + if email_use_tls: + email_use_tls = 'true' + else: + email_use_tls = 'false' + if "email_use_ssl" in worker["smtp"]: + email_use_ssl = worker["smtp"]["email_use_ssl"] + if isinstance(email_use_ssl, bool): + if email_use_ssl: + email_use_ssl = 'true' + else: + email_use_ssl = 'false' + if "email_backend" in worker["smtp"]: + email_backend = worker["smtp"]["email_backend"] fsettings = open("%s/settings.conf" % workerdir, 'w') - fsettings.write(template_settings_conf.substitute(cookie_secure=cookie_secure, session_cookie_secure=session_cookie_secure, lava_http_fqdn=lava_http_fqdn, allowed_hosts=allowed_hosts)) + fsettings.write( + template_settings_conf.substitute( + cookie_secure=cookie_secure, + session_cookie_secure=session_cookie_secure, + lava_http_fqdn=lava_http_fqdn, + allowed_hosts=allowed_hosts, + email_host = email_host, + email_host_user = email_host_user, + email_host_password = email_host_password, + email_port = email_port, + email_use_tls = email_use_tls, + email_use_ssl = email_use_ssl, + email_backend = email_backend, + server_email = server_email + ) + ) fsettings.close() master_use_zmq_auth = False if "zmq_auth" in worker: @@ -304,7 +364,7 @@ def main(): else: slaves = workers["slaves"] for slave in slaves: - keywords_slaves = [ "name", "host", "dispatcher_ip", "remote_user", "remote_master", "remote_address", "remote_rpc_port", "remote_proto", "extra_actions", "zmq_auth_key", "zmq_auth_key_secret", "default_slave", "export_ser2net", "expose_ser2net", "remote_user_token", "zmq_auth_master_key", "expose_ports", "env", "bind_dev", "loglevel", "use_nfs", "arch", "devices", "lava-coordinator", "use_tap", "host_healthcheck" ] + keywords_slaves = [ "name", "host", "dispatcher_ip", "remote_user", "remote_master", "remote_address", "remote_rpc_port", "remote_proto", "extra_actions", "zmq_auth_key", "zmq_auth_key_secret", "default_slave", "export_ser2net", "expose_ser2net", "remote_user_token", "zmq_auth_master_key", "expose_ports", "env", "bind_dev", "loglevel", "use_nfs", "arch", "devices", "lava-coordinator", "use_tap", "host_healthcheck", "use_tftp", "use_nbd", "use_overlay_server" ] for keyword in slave: if not keyword in keywords_slaves: print("WARNING: unknown keyword %s" % keyword) @@ -334,7 +394,7 @@ def main(): dockcomp["services"][name] = {} dockcomp["services"][name]["hostname"] = name dockcomp["services"][name]["dns_search"] = "" - dockcomp["services"][name]["ports"] = [ "69:69/udp", "80:80", "61950-62000:61950-62000" ] + dockcomp["services"][name]["ports"] = [] dockcomp["services"][name]["volumes"] = [ "/boot:/boot", "/lib/modules:/lib/modules" ] dockcomp["services"][name]["environment"] = {} dockcomp["services"][name]["build"] = {} @@ -474,6 +534,22 @@ def main(): fudev.close() if not "bind_dev" in slave or not slave["bind_dev"]: dockcomp_add_device(dockcomp, worker_name, "/dev/%s:/dev/%s" % (udev_dev["name"], udev_dev["name"])) + use_tftp = True + if "use_tftp" in worker: + use_tftp = worker["use_tftp"] + if use_tftp: + dockcomp["services"][name]["ports"].append("69:69/udp") + # TODO permit to change the range of NBD ports + use_nbd = True + if "use_nbd" in worker: + use_nbd = worker["use_nbd"] + if use_nbd: + dockcomp["services"][name]["ports"].append("61950-62000:61950-62000") + use_overlay_server = True + if "use_overlay_server" in worker: + use_overlay_server = worker["use_overlay_server"] + if use_overlay_server: + dockcomp["services"][name]["ports"].append("80:80") use_nfs = False if "use_nfs" in worker: use_nfs = worker["use_nfs"] |