def get_tags(self, uri=None, method=None, conn=None):
try:
tags = get_tags_by_agent_id(agent_id=self.agent_id)
if tags:
for i in range(len(tags)):
name = tags[i].pop('tag_name')
tags[i]['name'] = name
tag_id = tags[i].pop('tag_id')
tags[i]['id'] = tag_id
status = (
GenericResults(
self.username, uri, method
).information_retrieved(tags, 1)
)
logger.info(status['message'])
except Exception as e:
status = (
GenericResults(
self.username, uri, method
).something_broke(self.agent_id, 'agent', e)
)
logger.error(status['message'])
return(status)
def get_data(self):
try:
agent_data = get_agent_info(agent_id)
agent_data[AgentKey.LastAgentUpdate] = (
int(agent_data[AgentKey.LastAgentUpdate].strftime('%s'))
)
if agent_data:
agent_data['tags'] = get_tags_by_agent_id(agent_id=self.agent_id)
agent_data[CommonAppKeys.BASIC_RV_STATS] = (
get_all_app_stats_by_agentid(self.agent_id)
)
status = (
GenericResults(
self.username, uri, method
).information_retrieved(agent_data, 1)
)
logger.info(status['message'])
else:
status = (
GenericResults(
self.username, uri, method
).invalid_id(self.agent_id, 'agent_id')
)
logger.info(status['message'])
except Exception as e:
agent_data = None
status = (
GenericResults(
self.username, uri, method
).something_broke(self.agent_id, 'agents', e)
)
logger.error(status['message'])
return(status)
