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)
def get_all_agentids(username, customer_name, count=30, offset=0, uri=None, method=None, conn=None): try: count = (r.table(AgentsCollection).get_all( customer_name, index=AgentKey.CustomerName).count().run(conn)) data = list( r.table(AgentsCollection).get_all( customer_name, index=AgentKey.CustomerName).pluck(AgentKey.AgentId).order_by( AgentKey.ComputerName).skip(offset).limit(count).run(conn)) if data: for agent in data: agent[BASIC_RV_STATS] = (get_all_app_stats_by_agentid( agent[AgentKey.AgentId])) status = (GenericResults(username, uri, method).information_retrieved(data, count)) logger.info(status['message']) except Exception as e: status = (GenericResults(username, uri, method).something_broke( 'get_all_agents', 'agent', e)) return (status)
def get_all_agentids(username, customer_name, count=30, offset=0, uri=None, method=None, conn=None): try: count = ( r .table(AgentsCollection) .get_all(customer_name, index=AgentKey.CustomerName) .count() .run(conn) ) data = list( r .table(AgentsCollection) .get_all(customer_name, index=AgentKey.CustomerName) .pluck(AgentKey.AgentId) .order_by(AgentKey.ComputerName) .skip(offset) .limit(count) .run(conn) ) if data: for agent in data: agent[BASIC_RV_STATS] = ( get_all_app_stats_by_agentid(agent[AgentKey.AgentId]) ) status = ( GenericResults( username, uri, method ).information_retrieved(data, count) ) logger.info(status['message']) except Exception as e: status = ( GenericResults( username, uri, method ).something_broke('get_all_agents', 'agent', e) ) return(status)