apple chunks

Author: webjay

apple_fetch.py

@@ -2,11 +2,13 @@
 Fetch from Apple's acsnservice
 """
 
-from requests import post
+from requests import Session
 from haystack import get_headers
 from helpers import status_code_success
 from date import unix_epoch, date_milliseconds
 
+requestSession = Session()
+
 
 def acsnservice_fetch(decryption_key, ids, startdate, enddate):
     """Fetch from Apple's acsnservice"""
@@ -19,7 +21,7 @@ def acsnservice_fetch(decryption_key, ids, startdate, enddate):
             }
         ]
     }
-    return post(
+    return requestSession.post(
         "https://gateway.icloud.com/acsnservice/fetch",
         headers=get_headers(decryption_key),
         json=data,

main.py

@@ -7,9 +7,11 @@
 
 from argparse import ArgumentParser
 import json
+from time import sleep
 from api import fetch_devices, send_reports
 from cryptic import b64_ascii, get_hashed_public_key, bytes_to_int
 from apple_fetch import apple_fetch
+from helpers import chunks
 from report import create_reports
 
 
@@ -59,15 +61,15 @@ def get_args():
         publicHashBase64 = b64_ascii(get_hashed_public_key(privateKeyBytes))
         device_mapping[publicHashBase64] = device
 
-    apple_result = apple_fetch(command_args.key, list(device_mapping.keys()))
-    if command_args.verbose:
-        print("Results:", len(apple_result["results"]))
-
-    report_list = create_reports(apple_result, device_mapping)
-    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 devices_chunk in chunks(device_mapping, 10):
+        apple_result = apple_fetch(command_args.key, list(devices_chunk.keys()))
+        if command_args.verbose:
+            print("Results:", len(apple_result["results"]))
+        report_list = create_reports(apple_result, device_mapping)
+        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)
+        sleep(0.1)