108 lines
3.7 KiB
Python
108 lines
3.7 KiB
Python
"""uWSGI related tools."""
|
|
|
|
import os
|
|
|
|
from .. import package
|
|
from .. import system
|
|
from .. import utils
|
|
|
|
from . import base
|
|
|
|
|
|
class Uwsgi(base.Installer):
|
|
"""uWSGI installer."""
|
|
|
|
appname = "uwsgi"
|
|
packages = {
|
|
"deb": ["uwsgi", "uwsgi-plugin-python"],
|
|
"rpm": ["uwsgi", "uwsgi-plugin-python"],
|
|
}
|
|
|
|
def get_socket_path(self, app):
|
|
"""Return socket path."""
|
|
if package.backend.FORMAT == "deb":
|
|
return "/run/uwsgi/app/{}_instance/socket".format(app)
|
|
return "/run/uwsgi/{}_instance.sock".format(app)
|
|
|
|
def get_template_context(self, app):
|
|
"""Additionnal variables."""
|
|
context = super(Uwsgi, self).get_template_context()
|
|
context.update({
|
|
"app_user": self.config.get(app, "user"),
|
|
"app_venv_path": self.config.get(app, "venv_path"),
|
|
"app_instance_path": (
|
|
self.config.get(app, "instance_path")),
|
|
"uwsgi_socket_path": self.get_socket_path(app),
|
|
})
|
|
return context
|
|
|
|
def get_config_dir(self):
|
|
"""Return appropriate configuration directory."""
|
|
if package.backend.FORMAT == "deb":
|
|
return os.path.join(self.config_dir, "apps-available")
|
|
return "{}.d".format(self.config_dir)
|
|
|
|
def _enable_config_debian(self, dst):
|
|
"""Enable config file."""
|
|
link = os.path.join(
|
|
self.config_dir, "apps-enabled", os.path.basename(dst))
|
|
if os.path.exists(link):
|
|
return
|
|
os.symlink(dst, link)
|
|
|
|
def _setup_config(self, app):
|
|
"""Common setup code."""
|
|
context = self.get_template_context(app)
|
|
src = self.get_file_path("{}.ini.tpl".format(app))
|
|
dst = os.path.join(
|
|
self.get_config_dir(), "{}_instance.ini".format(app))
|
|
utils.copy_from_template(src, dst, context)
|
|
return dst
|
|
|
|
def _setup_modoboa_config(self):
|
|
"""Custom modoboa configuration."""
|
|
dst = self._setup_config("modoboa")
|
|
if package.backend.FORMAT == "deb":
|
|
self._enable_config_debian(dst)
|
|
else:
|
|
system.add_user_to_group(
|
|
"uwsgi", self.config.get("modoboa", "user"))
|
|
utils.exec_cmd("chmod -R g+w {}/media".format(
|
|
self.config.get("modoboa", "instance_path")))
|
|
utils.exec_cmd("chmod -R g+w {}/pdfcredentials".format(
|
|
self.config.get("modoboa", "home_dir")))
|
|
pattern = (
|
|
"s/emperor-tyrant = true/emperor-tyrant = false/")
|
|
utils.exec_cmd(
|
|
"perl -pi -e '{}' /etc/uwsgi.ini".format(pattern))
|
|
|
|
def _setup_automx_config(self):
|
|
"""Custom automx configuration."""
|
|
dst = self._setup_config("automx")
|
|
if package.backend.FORMAT == "deb":
|
|
self._enable_config_debian(dst)
|
|
else:
|
|
system.add_user_to_group(
|
|
"uwsgi", self.config.get("automx", "user"))
|
|
pattern = (
|
|
"s/emperor-tyrant = true/emperor-tyrant = false/")
|
|
utils.exec_cmd(
|
|
"perl -pi -e '{}' /etc/uwsgi.ini".format(pattern))
|
|
|
|
def post_run(self):
|
|
"""Additionnal tasks."""
|
|
self._setup_modoboa_config()
|
|
if self.config.getboolean("automx", "enabled"):
|
|
self._setup_automx_config()
|
|
|
|
def restart_daemon(self):
|
|
"""Restart daemon process."""
|
|
instances = ["modoboa_instance"]
|
|
if self.config.getboolean("automx", "enabled"):
|
|
instances.append("automx_instance")
|
|
for instance in instances:
|
|
code, output = utils.exec_cmd("service uwsgi status {}".format(
|
|
instance))
|
|
action = "start" if code else "restart"
|
|
utils.exec_cmd("service uwsgi {}".format(action))
|