From e2a641e829dd304c1b56b534e314f23092e335ab Mon Sep 17 00:00:00 2001
From: Mark Huang <mlhuang@cs.princeton.edu>
Date: Fri, 20 Oct 2006 18:26:05 +0000
Subject: [PATCH] conf_file functions

---
 PLC/Methods/AddConfFile.py    | 38 ++++++++++++++++++++++++++++++++
 PLC/Methods/DeleteConfFile.py | 31 ++++++++++++++++++++++++++
 PLC/Methods/GetConfFiles.py   | 26 ++++++++++++++++++++++
 PLC/Methods/UpdateConfFile.py | 41 +++++++++++++++++++++++++++++++++++
 4 files changed, 136 insertions(+)
 create mode 100644 PLC/Methods/AddConfFile.py
 create mode 100644 PLC/Methods/DeleteConfFile.py
 create mode 100644 PLC/Methods/GetConfFiles.py
 create mode 100644 PLC/Methods/UpdateConfFile.py

diff --git a/PLC/Methods/AddConfFile.py b/PLC/Methods/AddConfFile.py
new file mode 100644
index 00000000..eed0ae0d
--- /dev/null
+++ b/PLC/Methods/AddConfFile.py
@@ -0,0 +1,38 @@
+from PLC.Faults import *
+from PLC.Method import Method
+from PLC.Parameter import Parameter, Mixed
+from PLC.ConfFiles import ConfFile, ConfFiles
+from PLC.Auth import PasswordAuth
+
+can_update = lambda (field, value): field not in \
+             ['conf_file_id', 'source', 'dest', 'node_ids', 'nodegroup_ids']
+
+class AddConfFile(Method):
+    """
+    Adds a new node configuration file. Any fields specified in
+    conf_file_fields are used, otherwise defaults are used.
+
+    Returns 1 if successful, faults otherwise.
+    """
+
+    roles = ['admin']
+
+    update_fields = dict(filter(can_update, ConfFile.fields.items()))
+
+    accepts = [
+        PasswordAuth(),
+        ConfFile.fields['source'],
+        ConfFile.fields['dest'],
+        update_fields
+        ]
+
+    returns = Parameter(int, '1 if successful')
+
+    def call(self, auth, source, dest, conf_file_fields = {}):
+        conf_file_fields = dict(filter(can_update, conf_file_fields.items()))
+        conf_file = ConfFile(self.api, conf_file_fields)
+        conf_file['source'] = source
+        conf_file['dest'] = dest
+        conf_file.sync()
+
+        return 1
diff --git a/PLC/Methods/DeleteConfFile.py b/PLC/Methods/DeleteConfFile.py
new file mode 100644
index 00000000..226e9ddb
--- /dev/null
+++ b/PLC/Methods/DeleteConfFile.py
@@ -0,0 +1,31 @@
+from PLC.Faults import *
+from PLC.Method import Method
+from PLC.Parameter import Parameter, Mixed
+from PLC.ConfFiles import ConfFile, ConfFiles
+from PLC.Auth import PasswordAuth
+
+class DeleteConfFile(Method):
+    """
+    Returns an array of structs containing details about node
+    configuration files. If conf_file_ids is specified, only the
+    specified configuration files will be queried.
+    """
+
+    roles = ['admin']
+
+    accepts = [
+        PasswordAuth(),
+        ConfFile.fields['conf_file_id']
+        ]
+
+    returns = Parameter(int, '1 if successful')
+
+    def call(self, auth, conf_file_id):
+        conf_files = ConfFiles(self.api, [conf_file_id]).values()
+        if not conf_files:
+            raise PLCInvalidArgument, "No such configuration file"
+
+        conf_file = conf_files[0]
+        conf_file.delete()
+
+        return 1
diff --git a/PLC/Methods/GetConfFiles.py b/PLC/Methods/GetConfFiles.py
new file mode 100644
index 00000000..15dee3b8
--- /dev/null
+++ b/PLC/Methods/GetConfFiles.py
@@ -0,0 +1,26 @@
+from PLC.Faults import *
+from PLC.Method import Method
+from PLC.Parameter import Parameter, Mixed
+from PLC.ConfFiles import ConfFile, ConfFiles
+from PLC.Auth import PasswordAuth
+
+class GetConfFiles(Method):
+    """
+    Returns an array of structs containing details about node
+    configuration files. If conf_file_ids is specified, only the
+    specified configuration files will be queried.
+    """
+
+    roles = ['admin']
+
+    accepts = [
+        PasswordAuth(),
+        [ConfFile.fields['conf_file_id']]
+        ]
+
+    returns = [ConfFile.fields]
+
+    def call(self, auth, conf_file_ids = None):
+        conf_files = ConfFiles(self.api, conf_file_ids).values()
+        conf_files = [dict(conf_file) for conf_file in conf_files]
+        return conf_files
diff --git a/PLC/Methods/UpdateConfFile.py b/PLC/Methods/UpdateConfFile.py
new file mode 100644
index 00000000..b3dfc18c
--- /dev/null
+++ b/PLC/Methods/UpdateConfFile.py
@@ -0,0 +1,41 @@
+from PLC.Faults import *
+from PLC.Method import Method
+from PLC.Parameter import Parameter, Mixed
+from PLC.ConfFiles import ConfFile, ConfFiles
+from PLC.Auth import PasswordAuth
+
+can_update = lambda (field, value): field not in \
+             ['conf_file_id', 'node_ids', 'nodegroup_ids']
+
+class UpdateConfFile(Method):
+    """
+    Updates a node configuration file. Only the fields specified in
+    conf_file_fields are updated, all other fields are left untouched.
+
+    Returns 1 if successful, faults otherwise.
+    """
+
+    roles = ['admin']
+
+    update_fields = dict(filter(can_update, ConfFile.fields.items()))
+
+    accepts = [
+        PasswordAuth(),
+        ConfFile.fields['conf_file_id'],
+        update_fields
+        ]
+
+    returns = Parameter(int, '1 if successful')
+
+    def call(self, auth, conf_file_id, conf_file_fields):
+        conf_file_fields = dict(filter(can_update, conf_file_fields.items()))
+
+        conf_files = ConfFiles(self.api, [conf_file_id]).values()
+        if not conf_files:
+            raise PLCInvalidArgument, "No such configuration file"
+
+        conf_file = conf_files[0]
+        conf_file.update(conf_file_fields)
+        conf_file.sync()
+
+        return 1
-- 
2.47.0