chunk for receiver

Author: webjay

helpers.py

@@ -12,6 +12,10 @@ def status_code_success(status_code):
 
 
 def chunks(data, step):
-    data_iterable = iter(data)
-    for _ in range(0, len(data), step):
-        yield {k: data[k] for k in islice(data_iterable, step)}
+    if isinstance(data, dict):
+        data_iterable = iter(data)
+        for _ in range(0, len(data), step):
+            yield {k: data[k] for k in islice(data_iterable, step)}
+    elif isinstance(data, list):
+        for i in range(0, len(data), step):
+            yield data[i : i + step]

main.py

@@ -69,7 +69,10 @@ def get_args():
         for report in report_list:
             if "privateKeyNumeric" in report:
                 del report["privateKeyNumeric"]
-        send_reports(command_args.endpoint, report_list, headers=command_args.headers)
-        if command_args.verbose:
-            print(report_list)
+        for report_list_chunk in chunks(report_list, 100):
+            send_reports(
+                command_args.endpoint, report_list_chunk, headers=command_args.headers
+            )
+            if command_args.verbose:
+                print(report_list)
         sleep(0.1)