server-tools/nsca_client/models/nsca_server.py

# -*- coding: utf-8 -*-
# © 2015 ABF OSIELL <http://osiell.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

import os

from openerp import api, fields, models
from openerp.tools import config


class NscaServer(models.Model):
    _name = "nsca.server"
    _description = u"NSCA Server"

    name = fields.Char(u"Hostname", required=True)
    port = fields.Integer(u"Port", default=5667, required=True)
    password = fields.Char(u"Password")
    encryption_method = fields.Selection(
        selection='_selection_encryption_method',
        string=u"Encryption method", default='1', required=True)
    config_file_path = fields.Char(
        u"Configuration file",
        compute='_compute_config_file_path')
    node_hostname = fields.Char(
        u"Hostname of this node", required=True,
        help=u"This is the hostname of the current Odoo node declared in the "
             u"monitoring server.")
    check_ids = fields.One2many(
        'nsca.check', 'server_id', string=u"Checks")

    def _selection_encryption_method(self):
        return [
            ('0', u"0 - None (Do NOT use this option)"),
            ('1', u"1 - Simple XOR"),
            ('2', u"2 - DES"),
            ('3', u"3 - 3DES (Triple DES)"),
            ('4', u"4 - CAST-128"),
            ('5', u"5 - CAST-256"),
            ('6', u"6 - xTEA"),
            ('7', u"7 - 3WAY"),
            ('8', u"8 - BLOWFISH"),
            ('9', u"9 - TWOFISH"),
            ('10', u"10 - LOKI97"),
            ('11', u"11 - RC2"),
            ('12', u"12 - ARCFOUR"),
            ('14', u"14 - RIJNDAEL-128"),
            ('15', u"15 - RIJNDAEL-192"),
            ('16', u"16 - RIJNDAEL-256"),
            ('19', u"19 - WAKE"),
            ('20', u"20 - SERPENT"),
            ('22', u"22 - ENIGMA (Unix crypt)"),
            ('23', u"23 - GOST"),
            ('24', u"24 - SAFER64"),
            ('25', u"25 - SAFER128"),
            ('26', u"26 - SAFER+"),
        ]

    @api.multi
    def _compute_config_file_path(self):
        for server in self:
            data_dir_path = config.get('data_dir')
            file_name = 'send_nsca_%s.cfg' % server.id
            full_path = os.path.join(data_dir_path, 'nsca_client', file_name)
            server.config_file_path = full_path

    @api.multi
    def write_config_file(self):
        for server in self:
            data_dir_path = config.get('data_dir')
            config_dir_path = os.path.join(data_dir_path, 'nsca_client')
            try:
                os.makedirs(config_dir_path)
            except OSError as exception:
                if exception.errno != os.errno.EEXIST:
                    raise
            with open(server.config_file_path, 'w') as config_file:
                if server.password:
                    config_file.write('password=%s\n' % server.password)
                config_file.write(
                    'encryption_method=%s\n' % server.encryption_method)
        return True

    @api.multi
    def write(self, vals):
        res = super(NscaServer, self).write(vals)
        self.write_config_file()
        return res

    @api.model
    def create(self, vals):
        res = super(NscaServer, self).create(vals)
        res.write_config_file()
        return res