Add command line argument for maxResults; format with black

Author: gvanrossum-ms

gmail/gmail_dump.py

@@ -1,16 +1,20 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
-from pathlib import Path
 from base64 import urlsafe_b64decode as b64d
+from pathlib import Path
+import sys
+
+from google.oauth2.credentials import Credentials
 from google_auth_oauthlib.flow import InstalledAppFlow
 from googleapiclient.discovery import build
-from google.oauth2.credentials import Credentials
 
 SCOPES = ["https://www.googleapis.com/auth/gmail.readonly"]
 CREDS = "client_secret.json"
 TOKEN = Path("token.json")
-OUT = Path("mail_dump"); OUT.mkdir(exist_ok=True)
+OUT = Path("mail_dump")
+OUT.mkdir(exist_ok=True)
+
 
 def get_creds():
     if TOKEN.exists():
@@ -20,10 +24,20 @@ def get_creds():
     TOKEN.write_text(creds.to_json())
     return creds
 
+
 svc = build("gmail", "v1", credentials=get_creds())
 
-resp = svc.users().messages().list(userId="me", maxResults=50, q="").execute()
+maxResults = 50
+if sys.argv[1:]:
+    maxResults = int(sys.argv[1])
+
+resp = svc.users().messages().list(userId="me", maxResults=maxResults, q="").execute()
 for m in resp.get("messages", []):
-    raw = svc.users().messages().get(userId="me", id=m["id"], format="raw").execute()["raw"]
+    raw = (
+        svc.users()
+        .messages()
+        .get(userId="me", id=m["id"], format="raw")
+        .execute()["raw"]
+    )
     Path(OUT / f"{m['id']}.eml").write_bytes(b64d(raw.encode()))
 print("Done.")