Sync all skills and memories 2026-04-14 07:27
This commit is contained in:
327
skills/productivity/google-workspace/scripts/google_api.py
Normal file
327
skills/productivity/google-workspace/scripts/google_api.py
Normal file
@@ -0,0 +1,327 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Google Workspace API CLI for Hermes Agent.
|
||||
|
||||
Thin wrapper that delegates to gws (googleworkspace/cli) via gws_bridge.py.
|
||||
Maintains the same CLI interface for backward compatibility with Hermes skills.
|
||||
|
||||
Usage:
|
||||
python google_api.py gmail search "is:unread" [--max 10]
|
||||
python google_api.py gmail get MESSAGE_ID
|
||||
python google_api.py gmail send --to user@example.com --subject "Hi" --body "Hello"
|
||||
python google_api.py gmail reply MESSAGE_ID --body "Thanks"
|
||||
python google_api.py calendar list [--start DATE] [--end DATE] [--calendar primary]
|
||||
python google_api.py calendar create --summary "Meeting" --start DATETIME --end DATETIME
|
||||
python google_api.py calendar delete EVENT_ID
|
||||
python google_api.py drive search "budget report" [--max 10]
|
||||
python google_api.py contacts list [--max 20]
|
||||
python google_api.py sheets get SHEET_ID RANGE
|
||||
python google_api.py sheets update SHEET_ID RANGE --values '[[...]]'
|
||||
python google_api.py sheets append SHEET_ID RANGE --values '[[...]]'
|
||||
python google_api.py docs get DOC_ID
|
||||
"""
|
||||
|
||||
import argparse
|
||||
import json
|
||||
import os
|
||||
import subprocess
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
BRIDGE = Path(__file__).parent / "gws_bridge.py"
|
||||
PYTHON = sys.executable
|
||||
|
||||
|
||||
def gws(*args: str) -> None:
|
||||
"""Call gws via the bridge and exit with its return code."""
|
||||
result = subprocess.run(
|
||||
[PYTHON, str(BRIDGE)] + list(args),
|
||||
env={**os.environ, "HERMES_HOME": os.environ.get("HERMES_HOME", str(Path.home() / ".hermes"))},
|
||||
)
|
||||
sys.exit(result.returncode)
|
||||
|
||||
|
||||
# -- Gmail --
|
||||
|
||||
def gmail_search(args):
|
||||
cmd = ["gmail", "+triage", "--query", args.query, "--max", str(args.max), "--format", "json"]
|
||||
gws(*cmd)
|
||||
|
||||
def gmail_get(args):
|
||||
gws("gmail", "+read", "--id", args.message_id, "--headers", "--format", "json")
|
||||
|
||||
def gmail_send(args):
|
||||
cmd = ["gmail", "+send", "--to", args.to, "--subject", args.subject, "--body", args.body, "--format", "json"]
|
||||
if args.cc:
|
||||
cmd += ["--cc", args.cc]
|
||||
if args.html:
|
||||
cmd.append("--html")
|
||||
gws(*cmd)
|
||||
|
||||
def gmail_reply(args):
|
||||
gws("gmail", "+reply", "--message-id", args.message_id, "--body", args.body, "--format", "json")
|
||||
|
||||
def gmail_labels(args):
|
||||
gws("gmail", "users", "labels", "list", "--params", json.dumps({"userId": "me"}), "--format", "json")
|
||||
|
||||
def gmail_modify(args):
|
||||
body = {}
|
||||
if args.add_labels:
|
||||
body["addLabelIds"] = args.add_labels.split(",")
|
||||
if args.remove_labels:
|
||||
body["removeLabelIds"] = args.remove_labels.split(",")
|
||||
gws(
|
||||
"gmail", "users", "messages", "modify",
|
||||
"--params", json.dumps({"userId": "me", "id": args.message_id}),
|
||||
"--json", json.dumps(body),
|
||||
"--format", "json",
|
||||
)
|
||||
|
||||
|
||||
# -- Calendar --
|
||||
|
||||
def calendar_list(args):
|
||||
if args.start or args.end:
|
||||
# Specific date range — use raw Calendar API for precise timeMin/timeMax
|
||||
from datetime import datetime, timedelta, timezone as tz
|
||||
now = datetime.now(tz.utc)
|
||||
time_min = args.start or now.isoformat()
|
||||
time_max = args.end or (now + timedelta(days=7)).isoformat()
|
||||
gws(
|
||||
"calendar", "events", "list",
|
||||
"--params", json.dumps({
|
||||
"calendarId": args.calendar,
|
||||
"timeMin": time_min,
|
||||
"timeMax": time_max,
|
||||
"maxResults": args.max,
|
||||
"singleEvents": True,
|
||||
"orderBy": "startTime",
|
||||
}),
|
||||
"--format", "json",
|
||||
)
|
||||
else:
|
||||
# No date range — use +agenda helper (defaults to 7 days)
|
||||
cmd = ["calendar", "+agenda", "--days", "7", "--format", "json"]
|
||||
if args.calendar != "primary":
|
||||
cmd += ["--calendar", args.calendar]
|
||||
gws(*cmd)
|
||||
|
||||
def calendar_create(args):
|
||||
cmd = [
|
||||
"calendar", "+insert",
|
||||
"--summary", args.summary,
|
||||
"--start", args.start,
|
||||
"--end", args.end,
|
||||
"--format", "json",
|
||||
]
|
||||
if args.location:
|
||||
cmd += ["--location", args.location]
|
||||
if args.description:
|
||||
cmd += ["--description", args.description]
|
||||
if args.attendees:
|
||||
for email in args.attendees.split(","):
|
||||
cmd += ["--attendee", email.strip()]
|
||||
if args.calendar != "primary":
|
||||
cmd += ["--calendar", args.calendar]
|
||||
gws(*cmd)
|
||||
|
||||
def calendar_delete(args):
|
||||
gws(
|
||||
"calendar", "events", "delete",
|
||||
"--params", json.dumps({"calendarId": args.calendar, "eventId": args.event_id}),
|
||||
"--format", "json",
|
||||
)
|
||||
|
||||
|
||||
# -- Drive --
|
||||
|
||||
def drive_search(args):
|
||||
query = args.query if args.raw_query else f"fullText contains '{args.query}'"
|
||||
gws(
|
||||
"drive", "files", "list",
|
||||
"--params", json.dumps({
|
||||
"q": query,
|
||||
"pageSize": args.max,
|
||||
"fields": "files(id,name,mimeType,modifiedTime,webViewLink)",
|
||||
}),
|
||||
"--format", "json",
|
||||
)
|
||||
|
||||
|
||||
# -- Contacts --
|
||||
|
||||
def contacts_list(args):
|
||||
gws(
|
||||
"people", "people", "connections", "list",
|
||||
"--params", json.dumps({
|
||||
"resourceName": "people/me",
|
||||
"pageSize": args.max,
|
||||
"personFields": "names,emailAddresses,phoneNumbers",
|
||||
}),
|
||||
"--format", "json",
|
||||
)
|
||||
|
||||
|
||||
# -- Sheets --
|
||||
|
||||
def sheets_get(args):
|
||||
gws(
|
||||
"sheets", "+read",
|
||||
"--spreadsheet", args.sheet_id,
|
||||
"--range", args.range,
|
||||
"--format", "json",
|
||||
)
|
||||
|
||||
def sheets_update(args):
|
||||
values = json.loads(args.values)
|
||||
gws(
|
||||
"sheets", "spreadsheets", "values", "update",
|
||||
"--params", json.dumps({
|
||||
"spreadsheetId": args.sheet_id,
|
||||
"range": args.range,
|
||||
"valueInputOption": "USER_ENTERED",
|
||||
}),
|
||||
"--json", json.dumps({"values": values}),
|
||||
"--format", "json",
|
||||
)
|
||||
|
||||
def sheets_append(args):
|
||||
values = json.loads(args.values)
|
||||
gws(
|
||||
"sheets", "+append",
|
||||
"--spreadsheet", args.sheet_id,
|
||||
"--json-values", json.dumps(values),
|
||||
"--format", "json",
|
||||
)
|
||||
|
||||
|
||||
# -- Docs --
|
||||
|
||||
def docs_get(args):
|
||||
gws(
|
||||
"docs", "documents", "get",
|
||||
"--params", json.dumps({"documentId": args.doc_id}),
|
||||
"--format", "json",
|
||||
)
|
||||
|
||||
|
||||
# -- CLI parser (backward-compatible interface) --
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description="Google Workspace API for Hermes Agent (gws backend)")
|
||||
sub = parser.add_subparsers(dest="service", required=True)
|
||||
|
||||
# --- Gmail ---
|
||||
gmail = sub.add_parser("gmail")
|
||||
gmail_sub = gmail.add_subparsers(dest="action", required=True)
|
||||
|
||||
p = gmail_sub.add_parser("search")
|
||||
p.add_argument("query", help="Gmail search query (e.g. 'is:unread')")
|
||||
p.add_argument("--max", type=int, default=10)
|
||||
p.set_defaults(func=gmail_search)
|
||||
|
||||
p = gmail_sub.add_parser("get")
|
||||
p.add_argument("message_id")
|
||||
p.set_defaults(func=gmail_get)
|
||||
|
||||
p = gmail_sub.add_parser("send")
|
||||
p.add_argument("--to", required=True)
|
||||
p.add_argument("--subject", required=True)
|
||||
p.add_argument("--body", required=True)
|
||||
p.add_argument("--cc", default="")
|
||||
p.add_argument("--html", action="store_true", help="Send body as HTML")
|
||||
p.add_argument("--thread-id", default="", help="Thread ID (unused with gws, kept for compat)")
|
||||
p.set_defaults(func=gmail_send)
|
||||
|
||||
p = gmail_sub.add_parser("reply")
|
||||
p.add_argument("message_id", help="Message ID to reply to")
|
||||
p.add_argument("--body", required=True)
|
||||
p.set_defaults(func=gmail_reply)
|
||||
|
||||
p = gmail_sub.add_parser("labels")
|
||||
p.set_defaults(func=gmail_labels)
|
||||
|
||||
p = gmail_sub.add_parser("modify")
|
||||
p.add_argument("message_id")
|
||||
p.add_argument("--add-labels", default="", help="Comma-separated label IDs to add")
|
||||
p.add_argument("--remove-labels", default="", help="Comma-separated label IDs to remove")
|
||||
p.set_defaults(func=gmail_modify)
|
||||
|
||||
# --- Calendar ---
|
||||
cal = sub.add_parser("calendar")
|
||||
cal_sub = cal.add_subparsers(dest="action", required=True)
|
||||
|
||||
p = cal_sub.add_parser("list")
|
||||
p.add_argument("--start", default="", help="Start time (ISO 8601)")
|
||||
p.add_argument("--end", default="", help="End time (ISO 8601)")
|
||||
p.add_argument("--max", type=int, default=25)
|
||||
p.add_argument("--calendar", default="primary")
|
||||
p.set_defaults(func=calendar_list)
|
||||
|
||||
p = cal_sub.add_parser("create")
|
||||
p.add_argument("--summary", required=True)
|
||||
p.add_argument("--start", required=True, help="Start (ISO 8601 with timezone)")
|
||||
p.add_argument("--end", required=True, help="End (ISO 8601 with timezone)")
|
||||
p.add_argument("--location", default="")
|
||||
p.add_argument("--description", default="")
|
||||
p.add_argument("--attendees", default="", help="Comma-separated email addresses")
|
||||
p.add_argument("--calendar", default="primary")
|
||||
p.set_defaults(func=calendar_create)
|
||||
|
||||
p = cal_sub.add_parser("delete")
|
||||
p.add_argument("event_id")
|
||||
p.add_argument("--calendar", default="primary")
|
||||
p.set_defaults(func=calendar_delete)
|
||||
|
||||
# --- Drive ---
|
||||
drv = sub.add_parser("drive")
|
||||
drv_sub = drv.add_subparsers(dest="action", required=True)
|
||||
|
||||
p = drv_sub.add_parser("search")
|
||||
p.add_argument("query")
|
||||
p.add_argument("--max", type=int, default=10)
|
||||
p.add_argument("--raw-query", action="store_true", help="Use query as raw Drive API query")
|
||||
p.set_defaults(func=drive_search)
|
||||
|
||||
# --- Contacts ---
|
||||
con = sub.add_parser("contacts")
|
||||
con_sub = con.add_subparsers(dest="action", required=True)
|
||||
|
||||
p = con_sub.add_parser("list")
|
||||
p.add_argument("--max", type=int, default=50)
|
||||
p.set_defaults(func=contacts_list)
|
||||
|
||||
# --- Sheets ---
|
||||
sh = sub.add_parser("sheets")
|
||||
sh_sub = sh.add_subparsers(dest="action", required=True)
|
||||
|
||||
p = sh_sub.add_parser("get")
|
||||
p.add_argument("sheet_id")
|
||||
p.add_argument("range")
|
||||
p.set_defaults(func=sheets_get)
|
||||
|
||||
p = sh_sub.add_parser("update")
|
||||
p.add_argument("sheet_id")
|
||||
p.add_argument("range")
|
||||
p.add_argument("--values", required=True, help="JSON array of arrays")
|
||||
p.set_defaults(func=sheets_update)
|
||||
|
||||
p = sh_sub.add_parser("append")
|
||||
p.add_argument("sheet_id")
|
||||
p.add_argument("range")
|
||||
p.add_argument("--values", required=True, help="JSON array of arrays")
|
||||
p.set_defaults(func=sheets_append)
|
||||
|
||||
# --- Docs ---
|
||||
docs = sub.add_parser("docs")
|
||||
docs_sub = docs.add_subparsers(dest="action", required=True)
|
||||
|
||||
p = docs_sub.add_parser("get")
|
||||
p.add_argument("doc_id")
|
||||
p.set_defaults(func=docs_get)
|
||||
|
||||
args = parser.parse_args()
|
||||
args.func(args)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user