You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.5 KiB
49 lines
1.5 KiB
import splunk.admin as admin
|
|
import cp_aws_bin.utils.app_util as util
|
|
import splunklib.client as client
|
|
|
|
from cp_aws_bin.utils.local_manager import LocalServiceManager
|
|
from data_model.data_model import update_description
|
|
|
|
logger = util.get_logger()
|
|
|
|
ARG_TAGS = 'tags'
|
|
DEFAULT_APP_NAME = 'DA-ITSI-CP-aws-dashboards'
|
|
DEFAULT_OWNER = 'nobody'
|
|
DATAMODEL_REST = 'datamodel/model'
|
|
|
|
|
|
"""
|
|
@api {post} /saas-aws/da_itsi_cp_aws_data_model_cur update datamodel schema
|
|
@apiGroup data model
|
|
@apiName update-datamodel
|
|
@apiParam {string} tags list of tags
|
|
@apiSuccess (201) {Atom.Entry} entry nothing
|
|
"""
|
|
|
|
class DataModelHandler(admin.MConfigHandler):
|
|
def __init__(self, scriptMode, ctxInfo):
|
|
admin.MConfigHandler.__init__(self, scriptMode, ctxInfo)
|
|
self._service = LocalServiceManager(DEFAULT_APP_NAME, DEFAULT_OWNER, self.getSessionKey()).get_local_service()
|
|
self._collection = client.Collection(self._service, DATAMODEL_REST)
|
|
self.shouldAutoList = False
|
|
|
|
|
|
def setup(self):
|
|
for arg in [ARG_TAGS]:
|
|
self.supportedArgs.addReqArg(arg)
|
|
|
|
|
|
def handleCreate(self, confInfo):
|
|
tags = self.callerArgs[ARG_TAGS][0]
|
|
tags = tags.split('|') if tags else []
|
|
|
|
models = self._collection.list(search='name=Detailed_Billing_CUR OR name=Instance_Hour_CUR')
|
|
|
|
for model in models:
|
|
description = update_description(model.content.description, tags)
|
|
model.update(**{'description': description})
|
|
|
|
return
|
|
|
|
admin.init(DataModelHandler, admin.CONTEXT_APP_ONLY) |