Idoit-Tests 01

2024-12-15 14:38:48 +01:00
parent 16d9b8bd2c
commit e06c448a46
51 changed files with 1181 additions and 0 deletions
--- a/idoit_scaleup/object.py
+++ b/idoit_scaleup/object.py
@@ -0,0 +1,98 @@
+from typing import List
+from .base import IDoitApiBase
+
+
+class IDoitObject(IDoitApiBase):
+    def __init__(self, cfg, obj_type: str):
+        super().__init__(cfg)
+        self.obj_type = obj_type
+
+    def get_by_title(self, title: str, categories: List = []):
+        params = {
+            'filter': {
+                'type': self.obj_type,
+                'title': title,
+            },
+        }
+        if len(categories) > 0:
+            params['categories'] = categories
+        rtn = self.xml_rpc_call('cmdb.objects', params)
+        if len(rtn['result']) == 0:
+            return None
+        else:
+            return rtn['result'][0]
+
+    def get_by_id(self, obj_id: str, categories: List = []):
+        params = {
+            'id': obj_id
+        }
+        if len(categories) > 0:
+            params['categories'] = categories
+        rtn = self.xml_rpc_call('cmdb.object', params)
+        return rtn['result']
+
+    def get_all(self, categories: List = [], ids = None):
+        params = {
+            'filter': {
+                'type': self.obj_type
+            },
+        }
+        if ids:
+            params['filter']['ids'] = ids
+        if len(categories) > 0:
+            params['categories'] = categories
+        rtn = self.xml_rpc_call('cmdb.objects', params)
+        return rtn['result']
+
+    def create_object_with_title(self, title: str):
+        params = {
+            'title': title,
+            'type': self.obj_type
+        }
+        return self.xml_rpc_call('cmdb.object.create', params)
+
+    def create_object_if_not_there(self, title):
+        obj = self.get_by_title(title)
+        if obj is None:
+            r = self.create_object_with_title(title)
+            print('-------------------')
+            print("%s  (%s) " % (title, self.obj_type))
+            print('-------------------')
+            objId = r['result']['id']
+        else:
+            objId = obj['id']
+        return objId
+
+    def update_object(self, obj_id: str, title: str):
+        params = {
+            'id': obj_id,
+            'title': title,
+        }
+        return self.xml_rpc_call('cmdb.object.update', params)
+
+    def archive_object(self, obj_id: str):
+        params = {
+            'id': obj_id,
+            'status': 'C__RECORD_STATUS__ARCHIVED',
+        }
+        return self.xml_rpc_call('cmdb.object.delete', params)
+
+    def delete_object(self, obj_id: str):
+        params = {
+            'id': obj_id,
+            'status': 'C__RECORD_STATUS__DELETED',
+        }
+        return self.xml_rpc_call('cmdb.object.delete', params)
+
+    def purge_object(self, obj_id: str):
+        params = {
+            'id': obj_id,
+            'status': 'C__RECORD_STATUS__PURGE',
+        }
+        return self.xml_rpc_call('cmdb.object.delete', params)
+
+    def recycle_object(self, obj_id: str):
+        params = {
+            'id': obj_id,
+        }
+        return self.xml_rpc_call('cmdb.object.recycle', params)