diff --git a/Packs/Code42/Integrations/Code42/Code42.py b/Packs/Code42/Integrations/Code42/Code42.py index cceef47e6e8..5f3142c2ba8 100644 --- a/Packs/Code42/Integrations/Code42/Code42.py +++ b/Packs/Code42/Integrations/Code42/Code42.py @@ -189,12 +189,13 @@ def remove_user_from_departing_employee(self, username): def get_all_departing_employees(self, results): res = [] + results = int(results) if results else None pages = self._get_sdk().detectionlists.departing_employee.get_all() for page in pages: employees = page["items"] for employee in employees: res.append(employee) - if len(res) == results: + if results and len(res) == results: return res return res @@ -224,13 +225,14 @@ def remove_user_risk_tags(self, username, risk_tags): def get_all_high_risk_employees(self, risk_tags, results): risk_tags = _try_convert_str_list_to_list(risk_tags) + results = int(results) if results else None res = [] pages = self._get_sdk().detectionlists.high_risk_employee.get_all() for page in pages: employees = _get_all_high_risk_employees_from_page(page, risk_tags) for employee in employees: res.append(employee) - if len(res) == results: + if results and len(res) == results: return res return res @@ -637,7 +639,7 @@ def highriskemployee_remove_command(client, args): @logger def highriskemployee_get_all_command(client, args): tags = args.get("risktags") - results = args.get("results") + results = args.get("results") or 50 try: employees = client.get_all_high_risk_employees(tags, results) employees_context = [ diff --git a/Packs/Code42/Integrations/Code42/integration-Code42.yml b/Packs/Code42/Integrations/Code42/integration-Code42.yml index f7165b0fdde..f36c37bfd63 100644 --- a/Packs/Code42/Integrations/Code42/integration-Code42.yml +++ b/Packs/Code42/Integrations/Code42/integration-Code42.yml @@ -260,12 +260,13 @@ script: def get_all_departing_employees(self, results): res = [] + results = int(results) if results else None pages = self._get_sdk().detectionlists.departing_employee.get_all() for page in pages: employees = page["items"] for employee in employees: res.append(employee) - if len(res) == results: + if results and len(res) == results: return res return res @@ -295,13 +296,14 @@ script: def get_all_high_risk_employees(self, risk_tags, results): risk_tags = _try_convert_str_list_to_list(risk_tags) + results = int(results) if results else None res = [] pages = self._get_sdk().detectionlists.high_risk_employee.get_all() for page in pages: employees = _get_all_high_risk_employees_from_page(page, risk_tags) for employee in employees: res.append(employee) - if len(res) == results: + if results and len(res) == results: return res return res @@ -722,7 +724,7 @@ script: def highriskemployee_get_all_command(client, args): tags = args.get("risktags") - results = args.get("results") + results = args.get("results") or 50 try: employees = client.get_all_high_risk_employees(tags, results) employees_context = [