76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
# Copyright (c) 2015 Akanda, Inc. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import os
|
|
|
|
import jinja2
|
|
|
|
from astara_router.drivers import base
|
|
from astara_router.utils import execute
|
|
|
|
|
|
class NginxTemplateNotFound(Exception):
|
|
# TODO(adam_g): These should return 50x errors and not logged
|
|
# exceptions.
|
|
pass
|
|
|
|
|
|
class NginxLB(base.Manager):
|
|
NAME = 'nginx'
|
|
CONFIG_PATH = '/etc/nginx/sites-enabled/'
|
|
CONFIG_FILE_TEMPLATE = os.path.join(
|
|
os.path.dirname(__file__), 'nginx.conf.template')
|
|
INIT = 'nginx'
|
|
|
|
def __init__(self, root_helper='sudo astara-rootwrap /etc/rootwrap.conf'):
|
|
"""
|
|
Initializes DHCPManager class.
|
|
|
|
:type root_helper: str
|
|
:param root_helper: System utility used to gain escalate privileges.
|
|
"""
|
|
super(NginxLB, self).__init__(root_helper)
|
|
self._load_template()
|
|
|
|
def _load_template(self):
|
|
if not os.path.exists(self.CONFIG_FILE_TEMPLATE):
|
|
raise NginxTemplateNotFound(
|
|
'NGINX Config template not found @ %s' %
|
|
self.CONFIG_FILE_TEMPLATE
|
|
)
|
|
self.config_tmpl = jinja2.Template(
|
|
open(self.CONFIG_FILE_TEMPLATE).read())
|
|
|
|
def _render_config_template(self, path, config):
|
|
self._load_template()
|
|
with open(path, 'w') as out:
|
|
out.write(
|
|
self.config_tmpl.render(loadbalancer=config)
|
|
)
|
|
|
|
def restart(self):
|
|
execute(['service', self.INIT, 'restart'], self.root_helper)
|
|
|
|
def update_config(self, config):
|
|
path = os.path.join(
|
|
self.CONFIG_PATH, 'ak-loadbalancer-%s.conf' % config.id)
|
|
self._render_config_template(path=path, config=config)
|
|
self.restart()
|
|
|
|
|
|
class NginxPlusLB(NginxLB):
|
|
NAME = 'nginxplus'
|
|
CONFIG_FILE = '/tmp/nginx_plus.conf'
|
|
INIT = 'nginxplus'
|