def add_supported_app_to_agents(username,
customer_name,
uri,
method,
agent_id=None):
if agent_id:
agent_info = get_agent_info(agent_id)
apps_info = (get_apps_data(customer_name,
table=SupportedAppsCollection,
os_code=agent_info[AgentKey.OsCode]))
if len(apps_info) > 0:
for app_info in apps_info:
app_id = app_info.get(SupportedAppsKey.AppId)
file_data = get_file_data(app_id)
update_file_data(app_id, agent_id, file_data)
agent_info_to_insert = ({
SupportedAppsPerAgentKey.AgentId:
agent_id,
SupportedAppsPerAgentKey.AppId:
app_id,
SupportedAppsPerAgentKey.Status:
AVAILABLE,
SupportedAppsPerAgentKey.CustomerName:
customer_name,
SupportedAppsPerAgentKey.InstallDate:
r.epoch_time(0.0)
})
apps_to_insert_per_agent(username,
uri,
method,
agent_info_to_insert,
table=SupportedAppsPerAgentCollection)
def add_supported_app_to_agents(username, customer_name, uri, method, agent_id=None):
if agent_id:
agent_info = get_agent_info(agent_id)
apps_info = (
get_apps_data(
customer_name,
table=SupportedAppsCollection,
os_code=agent_info[AgentKey.OsCode]
)
)
if len(apps_info) > 0:
for app_info in apps_info:
app_id = app_info.get(SupportedAppsKey.AppId)
file_data = get_file_data(app_id)
update_file_data(
app_id,
agent_id, file_data
)
agent_info_to_insert = (
{
SupportedAppsPerAgentKey.AgentId: agent_id,
SupportedAppsPerAgentKey.AppId: app_id,
SupportedAppsPerAgentKey.Status: AVAILABLE,
SupportedAppsPerAgentKey.CustomerName: customer_name,
SupportedAppsPerAgentKey.InstallDate: r.epoch_time(0.0)
}
)
apps_to_insert_per_agent(
username, uri, method, agent_info_to_insert,
table=SupportedAppsPerAgentCollection
)
def add_custom_app_to_agents(username,
customer_name,
uri,
method,
file_data,
agent_id=None,
app_id=None):
if app_id and not agent_id:
app_info = (get_app_data(app_id, table=CustomAppsCollection))
agent_ids = get_all_agent_ids(customer_name,
agent_os=app_info[AgentKey.OsCode])
if len(agent_ids) > 0:
for agentid in agent_ids:
update_file_data(app_id, agentid, file_data)
agent_info_to_insert = ({
CustomAppsPerAgentKey.AgentId:
agentid,
CustomAppsPerAgentKey.AppId:
app_id,
CustomAppsPerAgentKey.Status:
AVAILABLE,
CustomAppsPerAgentKey.CustomerName:
customer_name,
CustomAppsPerAgentKey.InstallDate:
r.epoch_time(0.0)
})
apps_to_insert_per_agent(username, uri, method,
agent_info_to_insert)
if agent_id and not app_id:
agent_info = get_agent_info(agent_id)
apps_info = get_apps_data(customer_name,
os_code=agent_info[AgentKey.OsCode])
if len(apps_info) > 0:
for app_info in apps_info:
app_id = app_info.get(CustomAppsKey.AppId)
file_data = get_file_data(app_id)
update_file_data(app_id, agent_id, file_data)
agent_info_to_insert = ({
CustomAppsPerAgentKey.AgentId:
agent_id,
CustomAppsPerAgentKey.AppId:
app_id,
CustomAppsPerAgentKey.Status:
AVAILABLE,
CustomAppsPerAgentKey.CustomerName:
customer_name,
CustomAppsPerAgentKey.InstallDate:
r.epoch_time(0.0)
})
apps_to_insert_per_agent(username, uri, method,
agent_info_to_insert)
def get_by_app_id(self, stats=False, conn=None):
"""
"""
try:
pkg = list(
r
.table(self.CurrentAppsCollection, use_outdated=True)
.get_all(self.app_id, index=self.CurrentAppsIndexes.AppId)
.map(self.map_hash)
.run(conn)
)
if pkg:
pkg[0][self.CurrentAppsKey.FileData] = get_file_data(self.app_id)
if stats:
pkg[0]['agent_stats'] = (
get_all_stats_by_appid(
self.username, self.customer_name,
self.uri, self.method, self.app_id,
table=self.CurrentAppsPerAgentCollection
)['data']
)
status = (
GenericResults(
self.username, self.uri, self.method
).information_retrieved(pkg[0], 1)
)
else:
status = (
GenericResults(
self.username, self.uri, self.method
).invalid_id(self.app_id, 'package')
)
except Exception as e:
status = (
GenericResults(
self.username, self.uri, self.method
).something_broke(self.app_id, 'package', e)
)
logger.exception(e)
return(status)