375 lines
13 KiB
Python
375 lines
13 KiB
Python
"""
|
|
|
|
AUTHOR:
|
|
|
|
Khushal P Soonderji
|
|
|
|
DATE:
|
|
|
|
Tuesday, 26th Nov., 2024
|
|
|
|
OBJECTIVE:
|
|
|
|
To provide a base class for common behaviour of Google's APIs.
|
|
|
|
REFERENCES:
|
|
|
|
1. Quickstart: https://developers.google.com/gmail/api/quickstart/python
|
|
|
|
DOWNLOADS:
|
|
|
|
N/A
|
|
|
|
"""
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** IMPORT ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
# To make sibling directories accessible for imports:
|
|
import sys
|
|
sys.path.append(".")
|
|
sys.path.append("..")
|
|
|
|
# System-level activities:
|
|
import io
|
|
|
|
# My utils:
|
|
from utils_v2.string import json
|
|
from utils_v2.string import regex
|
|
from utils_v2.date_time import date_time
|
|
from utils_v2.oauth.services.goog import GoogleOAuth
|
|
from utils_v2.goog.models.data.api_call import GoogleApiResponse
|
|
from utils_v2.mail import mail_parser
|
|
|
|
# Related to Google:
|
|
from google.auth.transport.requests import Request
|
|
from google.oauth2.credentials import Credentials
|
|
from googleapiclient.discovery import build
|
|
|
|
# To make API calls:
|
|
import httpx
|
|
|
|
# For asynchronous activities:
|
|
import asyncio
|
|
|
|
# To work with date and time:
|
|
import datetime
|
|
|
|
# For working with datatypes:
|
|
from typing import Dict, Literal, List, Any
|
|
|
|
# For debugging:
|
|
from icecream import IceCreamDebugger
|
|
import inspect
|
|
|
|
# For computational help:
|
|
import math
|
|
|
|
# For base64 encoding:
|
|
import base64
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** MACROS / ONE-TIME INIT ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
# --- Nothing Yet
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** VARIABLES ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
# --- Nothing Yet
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** FUNCTIONS ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
# --- Nothing Yet
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** CLASSES ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
class AsyncGoogleBase:
|
|
|
|
def __init__(
|
|
self,
|
|
service_name: str,
|
|
client_id: str,
|
|
client_secret: str,
|
|
http_client: httpx.AsyncClient,
|
|
debug = True,
|
|
debug_prefix = "GMail | ",
|
|
debug_only_errors = True
|
|
):
|
|
|
|
"""
|
|
To initialize any Google API from one base class. The client's id and secret are available in the file
|
|
downloaded form https://console.cloud.google.com/apis/credentials (do not forget to select your app).
|
|
:param service_name: A string to identify this service.
|
|
:param client_id: From the OAuth JSON downloaded from
|
|
:param http_client: An asynchronous HTTP client to make API calls.
|
|
:param debug: Whether, or not, you would like to show debugging messages on the terminal.
|
|
:param debug_prefix: The prefix string to identify the debugging messages.
|
|
:param debug_only_errors: Whether you would like to show all debugging messages or just error messages.
|
|
"""
|
|
|
|
# Prepare the debugging utility:
|
|
self._debug_prefix = debug_prefix
|
|
self._printer = IceCreamDebugger(prefix = debug_prefix, includeContext = True)
|
|
if not debug: self._printer.disable()
|
|
self._debug_only_errors = debug_only_errors
|
|
|
|
# Accept the input configuration:
|
|
self._service_name = service_name
|
|
self._client_id = client_id
|
|
self._client_secret = client_secret
|
|
self._http_client = http_client
|
|
|
|
def enable_debug(self):
|
|
self._printer.enable()
|
|
|
|
def disable_debug(self):
|
|
self._printer.disable()
|
|
|
|
def debug_only_errors(self):
|
|
self._debug_only_errors = True
|
|
|
|
def debug_everything(self):
|
|
self._debug_only_errors = False
|
|
|
|
# ┏┓ ┳┓ ┓•
|
|
# ┣ ┏┓┏┓┏┓┏┓ ┃┃┏┓┏┏┓┏┫┓┏┓┏┓
|
|
# ┗┛┛ ┛ ┗┛┛ ┻┛┗ ┗┗┛┗┻┗┛┗┗┫
|
|
# ┛
|
|
|
|
@staticmethod
|
|
async def get_error_message(api_response: GoogleApiResponse):
|
|
|
|
try: return (await api_response.get_json())["error"]["message"]
|
|
except: return api_response.response.reason_phrase
|
|
|
|
# ┏┓┏┓┳ ┏┓ ┓┓•
|
|
# ┣┫┃┃┃ ┃ ┏┓┃┃┓┏┓┏┓
|
|
# ┛┗┣┛┻ ┗┛┗┻┗┗┗┛┗┗┫
|
|
# ┛
|
|
|
|
async def get(
|
|
self,
|
|
url: str,
|
|
headers: dict = None,
|
|
params: dict = None
|
|
) -> GoogleApiResponse:
|
|
|
|
"""
|
|
To call an API using the GET method.
|
|
:param url: The URL to call.
|
|
:param headers: The headers to pass.
|
|
:param params: The params to send in the query string itself.
|
|
:return: A structured response that includes the raw response, the exception (if any), and so on.
|
|
"""
|
|
|
|
# Prepare the structure of the response:
|
|
api_response = GoogleApiResponse(
|
|
serviceName = self._service_name,
|
|
action = inspect.stack()[1].function,
|
|
url = url,
|
|
method = "GET"
|
|
)
|
|
|
|
try:
|
|
|
|
# Make the API call:
|
|
response = await self._http_client.get(
|
|
url = url,
|
|
headers = headers,
|
|
params = params
|
|
)
|
|
|
|
# Note down the results:
|
|
api_response.response = response
|
|
api_response.httpCode = response.status_code
|
|
api_response.message = await self.get_error_message(api_response)
|
|
|
|
# If something goes wrong:
|
|
except Exception as exception:
|
|
api_response.exception = exception
|
|
api_response.message = str(exception)
|
|
self._printer(exception, api_response.url, api_response.method, headers, params)
|
|
|
|
# Done here:
|
|
return api_response
|
|
|
|
async def post(
|
|
self,
|
|
url: str,
|
|
headers: dict = None,
|
|
json: dict = None,
|
|
data: dict = None
|
|
) -> GoogleApiResponse:
|
|
|
|
"""
|
|
To call an API using the POST method.
|
|
:param url: The URL to call.
|
|
:param headers: The headers to pass.
|
|
:param json: The params to send in the JSON body.
|
|
:param data: The params to send in the form-data in the body.
|
|
:return: A structured response that includes the raw response, the exception (if any), and so on.
|
|
"""
|
|
|
|
# Prepare the structure of the response:
|
|
api_response = GoogleApiResponse(
|
|
serviceName = self._service_name,
|
|
action = inspect.stack()[1].function,
|
|
url = url,
|
|
method = "POST"
|
|
)
|
|
|
|
try:
|
|
|
|
# Make the API call:
|
|
response = await self._http_client.post(
|
|
url = url,
|
|
headers = headers,
|
|
json = json,
|
|
data = data
|
|
)
|
|
|
|
# Note down the results:
|
|
api_response.response = response
|
|
api_response.httpCode = response.status_code
|
|
api_response.message = await self.get_error_message(api_response)
|
|
|
|
# If something goes wrong:
|
|
except Exception as exception:
|
|
api_response.exception = exception
|
|
api_response.message = str(exception)
|
|
self._printer(exception, api_response.url, api_response.method, headers, json, data)
|
|
|
|
# Done here:
|
|
return api_response
|
|
|
|
async def put(
|
|
self,
|
|
url: str,
|
|
headers: dict = None,
|
|
json: dict = None,
|
|
data: dict = None
|
|
) -> GoogleApiResponse:
|
|
|
|
"""
|
|
To call an API using the PUT method.
|
|
:param url: The URL to call.
|
|
:param headers: The headers to pass.
|
|
:param json: The params to send in the JSON body.
|
|
:param data: The params to send in the form-data in the body.
|
|
:return: A structured response that includes the raw response, the exception (if any), and so on.
|
|
"""
|
|
|
|
# Prepare the structure of the response:
|
|
api_response = GoogleApiResponse(
|
|
serviceName = self._service_name,
|
|
action = inspect.stack()[1].function,
|
|
url = url,
|
|
method = "PUT"
|
|
)
|
|
|
|
try:
|
|
|
|
# Make the API call:
|
|
response = await self._http_client.put(
|
|
url = url,
|
|
headers = headers,
|
|
json = json,
|
|
data = data
|
|
)
|
|
|
|
# Note down the results:
|
|
api_response.response = response
|
|
api_response.httpCode = response.status_code
|
|
api_response.message = await self.get_error_message(api_response)
|
|
|
|
# If something goes wrong:
|
|
except Exception as exception:
|
|
api_response.exception = exception
|
|
api_response.message = str(exception)
|
|
self._printer(exception, api_response.url, api_response.method, headers, json, data)
|
|
|
|
# Done here:
|
|
return api_response
|
|
|
|
async def delete(
|
|
self,
|
|
url: str,
|
|
headers: dict = None
|
|
) -> GoogleApiResponse:
|
|
|
|
"""
|
|
To call an API using the DELETE method.
|
|
:param url: The URL to call.
|
|
:param headers: The headers to pass.
|
|
:return: A structured response that includes the raw response, the exception (if any), and so on.
|
|
"""
|
|
|
|
# Prepare the structure of the response:
|
|
api_response = GoogleApiResponse(
|
|
serviceName = self._service_name,
|
|
action = inspect.stack()[1].function,
|
|
url = url,
|
|
method = "DELETE"
|
|
)
|
|
|
|
try:
|
|
|
|
# Make the API call:
|
|
response = await self._http_client.delete(
|
|
url = url,
|
|
headers = headers
|
|
)
|
|
|
|
# Note down the results:
|
|
api_response.response = response
|
|
api_response.httpCode = response.status_code
|
|
api_response.message = await self.get_error_message(api_response)
|
|
|
|
# If something goes wrong:
|
|
except Exception as exception:
|
|
api_response.exception = exception
|
|
api_response.message = str(exception)
|
|
self._printer(exception, api_response.url, api_response.method, headers)
|
|
|
|
# Done here:
|
|
return api_response
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** MAIN PROGRAM ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
pass |