(20241127) Google API OAuth2.0 support ready!

utils_v2/goog/gmail/gmail_client.py

@@ -34,6 +34,8 @@
 
 # To make sibling directories accessible for imports:
 import sys
+from http.client import responses
+
 sys.path.append(".")
 sys.path.append("..")
 
@@ -47,7 +49,6 @@ from utils_v2.date_time import date_time
 from utils_v2.mail import mail_parser
 
 # My Google utils:
-from utils_v2.oauth.services.goog import GoogleOAuth
 from utils_v2.goog.base import AsyncGoogleBase
 from utils_v2.goog.models.data.auth_tokens import GoogleAuthTokens
 from utils_v2.goog.models.data.api_call import GoogleApiResponse
@@ -88,7 +89,12 @@ import base64
 #   *****************************************************************************************************************
 
 
-# --- Nothing Yet
+# Google Scopes:
+SCOPES_GMAIL_MAIL_MANAGEMENT = [
+    r"https://www.googleapis.com/auth/gmail.modify",
+    r"https://www.googleapis.com/auth/gmail.labels"
+]
+SCOPES_GMAIL_FULL = [r"https://mail.google.com/"]
 
 
 #   *****************************************************************************************************************
@@ -618,7 +624,7 @@ class AsyncGMailClient(AsyncGoogleBase):
         if api_response.httpCode in [200]:
             api_response.success = True
             api_json = await api_response.get_json()
-            raw_message = base64.urlsafe_b64decode(api_json["raw"])
+            raw_message = base64.urlsafe_b64decode(api_json["raw"]).decode()
             if return_raw: api_response.data = raw_message
             else:
                 parsed_message = mail_parser.parse(raw_message)
@@ -897,19 +903,33 @@ if __name__ == "__main__":
         # Create an instance of the client:
         my_gmail = AsyncGMailClient(
             service_name = "gmail",
-            client_id = secrets_dict["web"]["client_id"],
-            client_secret = secrets_dict["web"]["client_secret"],
+            oauth_json = secrets_dict,
             http_client = test_client,
+            redirect_url = r"https://api.thecaoffice.com/converse/mail/callback/gmail",
+            scopes = [
+                r"https://www.googleapis.com/auth/gmail.modify",
+                r"https://www.googleapis.com/auth/gmail.labels"
+            ],
             debug = True,
             debug_prefix = "GMail (M) | ",
             debug_only_errors = False
         )
 
+        # Request Auth:
+        # print("AUTH URL:", await my_gmail.get_authorization_url(
+        #     state = "User123-Bhopli"
+        # ))
+
+        # Get tokens from callback:
+        # print("TOKENS:", await my_gmail.get_authorization_tokens(
+        #     redirect_url = r"https://api.thecaoffice.com/converse/mail/callback/gmail?state=User123-Bhopli..."
+        # ))
+
         # Create a sample mail:
         my_mail = GMailMessage(
             from_email = "pskhushal@gmail.com",
             to_email = "orangebhopli@gmail.com",
-            subject = "Re: Bhopli is the best! (Thread Test)",
+            subject = "Bhopli is the best! (Parts Sequence Test)",
             cc_emails = None,
             bcc_emails = None
         )
@@ -936,7 +956,7 @@ if __name__ == "__main__":
             </html>
             """
         )
-        my_mail.add_text("This is how you should pet her 👇")
+        my_mail.add_text("(3rd part - text) This is how you should pet her 👇")
         my_mail.add_inline_image(r"../../../data/images/cat_petting.png")
         my_mail.add_attachment(r"../../../data/pdf/sample_label.pdf")
         # print(my_mail.get_raw_message(as_base64 = False))
@@ -945,7 +965,7 @@ if __name__ == "__main__":
         response = await my_gmail.send_message(
             tokens = test_tokens,
             message = my_mail,
-            thread_id = "1936caffb67996d3"
+            thread_id = None
         )
         print("SUCCESS:", response.success)
         print("SUMMARY:", response.to_markdown())