Fix/init sdk before bulk processing (#109)

* remove `write-to` and `send-to` and rename `print` to `search` on alerts/security-data cmds

* remove unused loggers

* update tests

* fix case where exception is printed on KeyboardInterrupt and closed pipe.

* update readme and changelog

* add TCP examples

* remove tests for removed func

* remove tests for removed loggers

* fix bug from change in latest py42

* add note about external tools for more complex requirements

* initalize sdk before passing to bulk func so it doesn't get initialized in each separate thread

* updated docs

Author: timabrmsn

src/code42cli/cmds/alert_rules.py

@@ -112,7 +112,8 @@ def bulk(state):
 @read_csv_arg(headers=ALERT_RULES_CSV_HEADERS)
 @sdk_options
 def add(state, csv_rows):
-    row_handler = lambda rule_id, username: _add_user(state.sdk, rule_id, username)
+    sdk = state.sdk
+    row_handler = lambda rule_id, username: _add_user(sdk, rule_id, username)
     run_bulk_process(row_handler, csv_rows, progress_label="Adding users to alert-rules:")
 
 
@@ -124,7 +125,8 @@ def add(state, csv_rows):
 @read_csv_arg(headers=ALERT_RULES_CSV_HEADERS)
 @sdk_options
 def remove(state, csv_rows):
-    row_handler = lambda rule_id, username: _remove_user(state.sdk, rule_id, username)
+    sdk = state.sdk
+    row_handler = lambda rule_id, username: _remove_user(sdk, rule_id, username)
     run_bulk_process(row_handler, csv_rows, progress_label="Removing users from alert-rules:")
 
 

src/code42cli/cmds/departing_employee.py

@@ -62,8 +62,9 @@ def bulk(state):
 @read_csv_arg(headers=DEPARTING_EMPLOYEE_CSV_HEADERS)
 @sdk_options
 def add(state, csv_rows):
+    sdk = state.sdk
     row_handler = lambda username, cloud_alias, departure_date, notes: _add_departing_employee(
-        state.sdk, username, cloud_alias, departure_date, notes
+        sdk, username, cloud_alias, departure_date, notes
     )
     run_bulk_process(
         row_handler, csv_rows, progress_label="Adding users to departing employee detection list:"
@@ -77,7 +78,8 @@ def add(state, csv_rows):
 @read_flat_file_arg
 @sdk_options
 def remove(state, file_rows):
-    row_handler = lambda username: _remove_departing_employee(state.sdk, username)
+    sdk = state.sdk
+    row_handler = lambda username: _remove_departing_employee(sdk, username)
     run_bulk_process(
         row_handler,
         file_rows,

src/code42cli/cmds/high_risk_employee.py

@@ -101,8 +101,9 @@ def bulk(state):
 @read_csv_arg(headers=HIGH_RISK_EMPLOYEE_CSV_HEADERS)
 @sdk_options
 def add(state, csv_rows):
+    sdk = state.sdk
     row_handler = lambda username, cloud_alias, risk_tag, notes: _add_high_risk_employee(
-        state.sdk, username, cloud_alias, risk_tag, notes
+        sdk, username, cloud_alias, risk_tag, notes
     )
     run_bulk_process(
         row_handler, csv_rows, progress_label="Adding users to high risk employee detection list:"
@@ -116,7 +117,8 @@ def add(state, csv_rows):
 @read_flat_file_arg
 @sdk_options
 def remove(state, file_rows):
-    row_handler = lambda username: _remove_high_risk_employee(state.sdk, username)
+    sdk = state.sdk
+    row_handler = lambda username: _remove_high_risk_employee(sdk, username)
     run_bulk_process(
         row_handler,
         file_rows,
@@ -132,7 +134,8 @@ def remove(state, file_rows):
 @read_csv_arg(headers=RISK_TAG_CSV_HEADERS)
 @sdk_options
 def add_risk_tags(state, csv_rows):
-    row_handler = lambda username, tag: _add_risk_tags(state.sdk, username, tag)
+    sdk = state.sdk
+    row_handler = lambda username, tag: _add_risk_tags(sdk, username, tag)
     run_bulk_process(
         row_handler, csv_rows, progress_label="Adding risk tags to users:",
     )
@@ -146,7 +149,8 @@ def add_risk_tags(state, csv_rows):
 @read_csv_arg(headers=RISK_TAG_CSV_HEADERS)
 @sdk_options
 def remove_risk_tags(state, csv_rows):
-    row_handler = lambda username, tag: _remove_risk_tags(state.sdk, username, tag)
+    sdk = state.sdk
+    row_handler = lambda username, tag: _remove_risk_tags(sdk, username, tag)
     run_bulk_process(
         row_handler, csv_rows, progress_label="Removing risk tags from users:",
     )

src/code42cli/cmds/legal_hold.py

@@ -139,9 +139,8 @@ def bulk(state):
 @read_csv_arg(headers=LEGAL_HOLD_CSV_HEADERS)
 @sdk_options
 def add(state, csv_rows):
-    row_handler = lambda matter_id, username: _add_user_to_legal_hold(
-        state.sdk, matter_id, username
-    )
+    sdk = state.sdk
+    row_handler = lambda matter_id, username: _add_user_to_legal_hold(sdk, matter_id, username)
     run_bulk_process(row_handler, csv_rows, progress_label="Adding users to legal hold:")
 
 
@@ -153,9 +152,8 @@ def add(state, csv_rows):
 @read_csv_arg(headers=LEGAL_HOLD_CSV_HEADERS)
 @sdk_options
 def remove(state, csv_rows):
-    row_handler = lambda matter_id, username: _remove_user_from_legal_hold(
-        state.sdk, matter_id, username
-    )
+    sdk = state.sdk
+    row_handler = lambda matter_id, username: _remove_user_from_legal_hold(sdk, matter_id, username)
     run_bulk_process(row_handler, csv_rows, progress_label="Removing users from legal hold:")