Google Email

"""Download attachments from a message
Usage: python download_attachments.py MESSAGE_ID --outdir ./attachments
"""
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
import base64, os, sys
from pathlib import Path

TOKEN = Path(r'C:/Users/mz038/.nanobot/workspace/token.json')
SCOPES = ['https://www.googleapis.com/auth/gmail.readonly']
if not TOKEN.exists():
    print('token.json not found at', TOKEN)
    sys.exit(1)
creds = Credentials.from_authorized_user_file(str(TOKEN), SCOPES)
service = build('gmail', 'v1', credentials=creds)

if len(sys.argv) < 2:
    print('Usage: python download_attachments.py MESSAGE_ID [OUTDIR]')
    sys.exit(1)
msg_id = sys.argv[1]
outdir = Path(sys.argv[2]) if len(sys.argv) > 2 else Path('./attachments')
outdir.mkdir(parents=True, exist_ok=True)

msg = service.users().messages().get(userId='me', id=msg_id, format='full').execute()
parts = msg.get('payload', {}).get('parts', [])
for part in parts:
    if part.get('filename') and part.get('body', {}).get('attachmentId'):
        att_id = part['body']['attachmentId']
        att = service.users().messages().attachments().get(userId='me', messageId=msg_id, id=att_id).execute()
        data = base64.urlsafe_b64decode(att['data'].encode('ASCII'))
        fn = part['filename']
        p = outdir / fn
        p.write_bytes(data)
        print('Wrote', p)

"""List messages matching a query using Gmail API
Usage: python list_messages.py "subject:APCS" --max 50
"""
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
import sys
from pathlib import Path

TOKEN = Path(r'C:/Users/mz038/.nanobot/workspace/token.json')
SCOPES = ['https://www.googleapis.com/auth/gmail.readonly']

if not TOKEN.exists():
    print('token.json not found at', TOKEN)
    sys.exit(1)
creds = Credentials.from_authorized_user_file(str(TOKEN), SCOPES)
service = build('gmail', 'v1', credentials=creds)

import argparse
p=argparse.ArgumentParser()
p.add_argument('q', help='Gmail search query (quoted)')
p.add_argument('--max', type=int, default=50)
args=p.parse_args()

res = service.users().messages().list(userId='me', q=args.q, maxResults=args.max).execute()
msgs = res.get('messages', [])
print('Found', len(msgs), 'messages')
for m in msgs:
    print(m['id'])

"""
OAuth CLI for Gmail (installed app flow)

Usage:
  python oauth_cli.py authorize
  python oauth_cli.py auth_url
  python oauth_cli.py finish --code <code>

Place your OAuth client secret at:
  C:/Users/<user>/.nanobot/workspace/client_secret.json

Authorize will open a browser to let you login and consent; token.json will be saved to the same workspace.
"""
import os
import sys
import json
import argparse
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

WORKSPACE = os.path.join(os.path.expanduser('~'), '.nanobot', 'workspace')
CLIENT_SECRET_PATH = os.path.join(WORKSPACE, 'client_secret.json')
TOKEN_PATH = os.path.join(WORKSPACE, 'token.json')
# scopes: adjust to least privilege required for your tasks
SCOPES = [
    'https://www.googleapis.com/auth/gmail.readonly',
    'https://www.googleapis.com/auth/gmail.send',
    'https://www.googleapis.com/auth/gmail.modify',
    'https://www.googleapis.com/auth/calendar.events'
]


def authorize():
    if not os.path.exists(CLIENT_SECRET_PATH):
        raise SystemExit(f'找不到 OAuth client_secret.json，請放到: {CLIENT_SECRET_PATH}')
    flow = InstalledAppFlow.from_client_secrets_file(CLIENT_SECRET_PATH, SCOPES)
    try:
        # try local server flow (opens browser and listens for callback)
        creds = flow.run_local_server(port=0)
    except Exception as e:
        # fallback to console flow when run_local_server isn't available
        print('無法使用 run_local_server（可能無法開啟瀏覽器），改用 console flow，請在終端貼上授權 url 並輸入 code。')
        creds = flow.run_console()
    with open(TOKEN_PATH, 'w', encoding='utf-8') as f:
        f.write(creds.to_json())
    print('授權完成，token 已儲存至', TOKEN_PATH)


def print_auth_url():
    if not os.path.exists(CLIENT_SECRET_PATH):
        raise SystemExit(f'找不到 OAuth client_secret.json，請放到: {CLIENT_SECRET_PATH}')
    flow = InstalledAppFlow.from_client_secrets_file(CLIENT_SECRET_PATH, SCOPES)
    auth_url, _ = flow.authorization_url(prompt='consent')
    print('請在本機瀏覽器開啟以下 URL，完成授權後會取得 code:')
    print(auth_url)


def finish_with_code(code: str):
    if not os.path.exists(CLIENT_SECRET_PATH):
        raise SystemExit(f'找不到 OAuth client_secret.json，請放到: {CLIENT_SECRET_PATH}')
    flow = InstalledAppFlow.from_client_secrets_file(CLIENT_SECRET_PATH, SCOPES)
    try:
        token_response = flow.fetch_token(code=code)
    except Exception as e:
        raise SystemExit(f'用 code 換 token 失敗: {e}')
    # InstalledAppFlow.fetch_token 回傳 dict，轉成 Credentials
    creds_obj = Credentials.from_authorized_user_info(token_response, SCOPES)
    with open(TOKEN_PATH, 'w', encoding='utf-8') as f:
        f.write(creds_obj.to_json())
    print('授權完成，token 已儲存至', TOKEN_PATH)


def load_credentials():
    creds = None
    if os.path.exists(TOKEN_PATH):
        creds = Credentials.from_authorized_user_file(TOKEN_PATH, SCOPES)
    if creds and creds.expired and creds.refresh_token:
        try:
            creds.refresh(Request())
            with open(TOKEN_PATH, 'w', encoding='utf-8') as f:
                f.write(creds.to_json())
        except Exception as e:
            print('刷新 token 失敗，需重新授權:', e)
            creds = None
    return creds


def build_service(creds):
    return build('gmail', 'v1', credentials=creds)


def main():
    p = argparse.ArgumentParser(prog='oauth_cli')
    sub = p.add_subparsers(dest='cmd')

    a = sub.add_parser('authorize')
    a.set_defaults(func=lambda args: authorize())

    au = sub.add_parser('auth_url')
    au.set_defaults(func=lambda args: print_auth_url())

    f = sub.add_parser('finish')
    f.add_argument('--code', required=True)
    f.set_defaults(func=lambda args: finish_with_code(args.code))

    args = p.parse_args()
    if not args.cmd:
        p.print_help()
        return
    try:
        args.func(args)
    except Exception as e:
        raise SystemExit(f'錯誤: {e}')


if __name__ == '__main__':
    main()

"""Send a simple test email using Gmail API
Usage: python send_test.py recipient@example.com
"""
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from email.mime.text import MIMEText
import base64
import sys
from pathlib import Path

TOKEN = Path(r'C:/Users/mz038/.nanobot/workspace/token.json')
SCOPES = ['https://www.googleapis.com/auth/gmail.send']

if not TOKEN.exists():
    print('token.json not found at', TOKEN)
    sys.exit(1)
creds = Credentials.from_authorized_user_file(str(TOKEN), SCOPES)
service = build('gmail', 'v1', credentials=creds)

recipient = sys.argv[1] if len(sys.argv) > 1 else None
if not recipient:
    print('Usage: python send_test.py recipient@example.com')
    sys.exit(1)

msg = MIMEText('這是一封測試郵件。','plain','utf-8')
msg['to'] = recipient
msg['from'] = 'me'
msg['subject'] = '測試郵件'
raw = base64.urlsafe_b64encode(msg.as_bytes()).decode()
res = service.users().messages().send(userId='me', body={'raw': raw}).execute()
print('Sent message id:', res.get('id'))