[IMP] nsca_client - Auto-create the configuration file used by the 'send_nsca' binary, password and encryption method are set in 'nsca.server' records

nsca_client/models/nsca_server.py

@@ -2,7 +2,10 @@
 # © 2015 ABF OSIELL <http://osiell.com>
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
 
-from openerp import fields, models
+import os
+
+from openerp import api, fields, models
+from openerp.tools import config
 
 
 class NscaServer(models.Model):
@@ -11,11 +14,80 @@ class NscaServer(models.Model):
 
     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 node declared in the "
              u"monitoring server.")
-    config_file_path = fields.Char(
-        u"Configuration file", default="/etc/send_nsca.cfg", required=True)
     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

nsca_client/views/nsca_server.xml

@@ -14,6 +14,8 @@
                 <group string="NSCA Server">
                     <field name="name" style="width: 30%;"/>
                     <field name="port"/>
+                    <field name="password" password="1"/>
+                    <field name="encryption_method"/>
                     <field name="config_file_path"/>
                 </group>
                 <group string="Node identity">