def __init__(self,
customer_name=None,
count=DefaultQueryValues.COUNT,
offset=DefaultQueryValues.OFFSET,
sort=SortValues.ASC,
sort_key=AgentKey.ComputerName,
user_name=None,
uri=None,
method=None):
self.user_name = user_name
self.customer_name = customer_name
self.uri = uri
self.method = method
self.count = count
self.offset = offset
self.sort = sort
self.list_of_valid_keys = [
AgentKey.ComputerName, AgentKey.HostName, AgentKey.DisplayName,
AgentKey.OsCode, AgentKey.OsString, AgentKey.AgentId,
AgentKey.AgentStatus, AgentKey.NeedsReboot, AgentKey.BasicStats,
AgentKey.ProductionLevel, AgentKey.LastAgentUpdate
]
self.valid_keys_to_filter_by = ([
AgentKey.OsCode, AgentKey.OsString, AgentKey.AgentStatus,
AgentKey.ProductionLevel
])
valid_keys_to_sort_by = ([
AgentKey.ComputerName,
AgentKey.HostName,
AgentKey.DisplayName,
AgentKey.OsCode,
AgentKey.OsString,
AgentKey.AgentStatus,
AgentKey.ProductionLevel,
AgentCommonKeys.AVAIL_VULN,
AgentCommonKeys.AVAIL_UPDATES,
AgentKey.LastAgentUpdate,
])
if sort_key in valid_keys_to_sort_by:
self.sort_key = sort_key
else:
self.sort_key = AgentKey.ComputerName
self.fetch_agents = (FetchAgents(customer_name, self.count,
self.offset, self.sort,
self.sort_key))
