99 lines
2.9 KiB
Python
Executable File
99 lines
2.9 KiB
Python
Executable File
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)
|