Gmail message count insights with Python

import os.path
from google.auth.transport.requests import Request
from google.oauth2.credentials import Credentials
from google_auth_oauthlib.flow import InstalledAppFlow
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError

# Define the Google Drive API scopes for accessing files and folders
# If modifying these scopes, delete the file token.json
SCOPES = ['https://www.googleapis.com/auth/gmail.readonly']
TOKEN_PATH = 'token.json'
CREDENTIALS_PATH = 'credentials.json'

def get_credentials():
    # Gets user credentials or prompts for authorization if needed
    # Try to load existing credentials from 'token.json'
    creds = Credentials.from_authorized_user_file(TOKEN_PATH, SCOPES) if os.path.exists(TOKEN_PATH) else None

    if not creds or not creds.valid:
        if creds and creds.expired and creds.refresh_token:
            # Refresh the credentials if they are expired and can be refreshed
            creds.refresh(Request())
        else:
            # If no valid credentials exist, initiate the OAuth flow to obtain them
            flow = InstalledAppFlow.from_client_secrets_file(CREDENTIALS_PATH, SCOPES)
            creds = flow.run_local_server(port=0)

        # Save the obtained credentials for future runs
        with open(TOKEN_PATH, 'w') as token_file:
            token_file.write(creds.to_json())

    return creds

def get_label_info(service, label_id):
    # Fetch label information for a given label ID
    label_info = service.users().labels().get(userId='me', id=label_id).execute()
    return label_info

def print_label_tree(labels, label_type, service):
    # Print label information in a hierarchical tree structure
    print(f"{label_type}")
    sorted_labels = sorted(labels, key=lambda x: x['name'].lower())
    for label in sorted_labels:
        label_info = get_label_info(service, label['id'])
        total_emails = label_info.get('messagesTotal', 0)
        print(f"- {label['name']} - [{total_emails}]")

def count_emails_by_label_type():
    # Count and display emails grouped by label type
    creds = get_credentials()
    service = build('gmail', 'v1', credentials=creds)
    profile = service.users().getProfile(userId='me').execute()

    if 'emailAddress' in profile:
        email_address = profile['emailAddress']
        print(f"Email Address: {email_address}")

        total_emails = profile['messagesTotal']
        total_conversations = profile['threadsTotal']

        print(f"Total Emails: {total_emails}")
        print(f"Total Conversations: {total_conversations}")
        print("\n")

        labels_response = service.users().labels().list(userId='me').execute()
        all_labels = labels_response.get('labels', [])

        # Group labels into system and user-defined categories
        system_labels = [label for label in all_labels if label.get('type') == 'system']
        user_labels = [label for label in all_labels if label.get('type') == 'user']

        # Print system and user labels with their respective email counts
        print_label_tree(system_labels, "System Labels\n", service)
        print("\n")
        print_label_tree(user_labels, "User Labels\n", service)
    else:
        print("Email address not found in the profile.")
        return

if __name__ == '__main__':
    count_emails_by_label_type()