arcade-google 0.0.13__py3-none-any.whl

arcade_google/tools/utils.py

@@ -0,0 +1,280 @@
+import re
+from base64 import urlsafe_b64decode
+from datetime import datetime, timedelta
+from enum import Enum
+from typing import Any, Optional
+from zoneinfo import ZoneInfo
+
+from bs4 import BeautifulSoup
+from google.oauth2.credentials import Credentials
+from googleapiclient.discovery import build
+
+from arcade_google.tools.models import Day, TimeSlot
+
+
+def parse_datetime(datetime_str: str, time_zone: str) -> datetime:
+    """
+    Parse a datetime string in ISO 8601 format and ensure it is timezone-aware.
+
+    Args:
+        datetime_str (str): The datetime string to parse. Expected format: 'YYYY-MM-DDTHH:MM:SS'.
+        time_zone (str): The timezone to apply if the datetime string is naive.
+
+    Returns:
+        datetime: A timezone-aware datetime object.
+
+    Raises:
+        ValueError: If the datetime string is not in the correct format.
+    """
+    try:
+        dt = datetime.fromisoformat(datetime_str)
+        if dt.tzinfo is None:
+            dt = dt.replace(tzinfo=ZoneInfo(time_zone))
+    except ValueError as e:
+        raise ValueError(
+            f"Invalid datetime format: '{datetime_str}'. Expected ISO 8601 format, e.g., '2024-12-31T15:30:00'."
+        ) from e
+    return dt
+
+
+class DateRange(Enum):
+    TODAY = "today"
+    YESTERDAY = "yesterday"
+    LAST_7_DAYS = "last_7_days"
+    LAST_30_DAYS = "last_30_days"
+    THIS_MONTH = "this_month"
+    LAST_MONTH = "last_month"
+    THIS_YEAR = "this_year"
+
+    def to_date_query(self):
+        today = datetime.now()
+        result = "after:"
+        comparison_date = today
+
+        if self == DateRange.YESTERDAY:
+            comparison_date = today - timedelta(days=1)
+        elif self == DateRange.LAST_7_DAYS:
+            comparison_date = today - timedelta(days=7)
+        elif self == DateRange.LAST_30_DAYS:
+            comparison_date = today - timedelta(days=30)
+        elif self == DateRange.THIS_MONTH:
+            comparison_date = today.replace(day=1)
+        elif self == DateRange.LAST_MONTH:
+            comparison_date = (today.replace(day=1) - timedelta(days=1)).replace(day=1)
+        elif self == DateRange.THIS_YEAR:
+            comparison_date = today.replace(month=1, day=1)
+        elif self == DateRange.LAST_MONTH:
+            comparison_date = (today.replace(month=1, day=1) - timedelta(days=1)).replace(
+                month=1, day=1
+            )
+
+        return result + comparison_date.strftime("%Y/%m/%d")
+
+
+def parse_email(email_data: dict[str, Any]) -> Optional[dict[str, str]]:
+    """
+    Parse email data and extract relevant information.
+
+    Args:
+        email_data (Dict[str, Any]): Raw email data from Gmail API.
+
+    Returns:
+        Optional[Dict[str, str]]: Parsed email details or None if parsing fails.
+    """
+    try:
+        payload = email_data["payload"]
+        headers = {d["name"].lower(): d["value"] for d in payload["headers"]}
+
+        body_data = _get_email_body(payload)
+
+        return {
+            "id": email_data.get("id", ""),
+            "from": headers.get("from", ""),
+            "date": headers.get("date", ""),
+            "subject": headers.get("subject", "No subject"),
+            "body": _clean_email_body(body_data) if body_data else "",
+        }
+    except Exception as e:
+        print(f"Error parsing email {email_data.get('id', 'unknown')}: {e}")
+        return None
+
+
+def parse_draft_email(draft_email_data: dict[str, Any]) -> Optional[dict[str, str]]:
+    """
+    Parse draft email data and extract relevant information.
+
+    Args:
+        draft_email_data (Dict[str, Any]): Raw draft email data from Gmail API.
+
+    Returns:
+        Optional[Dict[str, str]]: Parsed draft email details or None if parsing fails.
+    """
+    try:
+        message = draft_email_data["message"]
+        payload = message["payload"]
+        headers = {d["name"].lower(): d["value"] for d in payload["headers"]}
+
+        body_data = _get_email_body(payload)
+
+        return {
+            "id": draft_email_data.get("id", ""),
+            "from": headers.get("from", ""),
+            "date": headers.get("internaldate", ""),
+            "subject": headers.get("subject", "No subject"),
+            "body": _clean_email_body(body_data) if body_data else "",
+        }
+    except Exception as e:
+        print(f"Error parsing draft email {draft_email_data.get('id', 'unknown')}: {e}")
+        return None
+
+
+def get_draft_url(draft_id):
+    return f"https://mail.google.com/mail/u/0/#drafts/{draft_id}"
+
+
+def get_sent_email_url(sent_email_id):
+    return f"https://mail.google.com/mail/u/0/#sent/{sent_email_id}"
+
+
+def get_email_in_trash_url(email_id):
+    return f"https://mail.google.com/mail/u/0/#trash/{email_id}"
+
+
+def _get_email_body(payload: dict[str, Any]) -> Optional[str]:
+    """
+    Extract email body from payload.
+
+    Args:
+        payload (Dict[str, Any]): Email payload data.
+
+    Returns:
+        Optional[str]: Decoded email body or None if not found.
+    """
+    if "body" in payload and payload["body"].get("data"):
+        return urlsafe_b64decode(payload["body"]["data"]).decode()
+
+    for part in payload.get("parts", []):
+        if part.get("mimeType") == "text/plain" and "data" in part["body"]:
+            return urlsafe_b64decode(part["body"]["data"]).decode()
+
+    return None
+
+
+def _clean_email_body(body: str) -> str:
+    """
+    Remove HTML tags and clean up email body text while preserving most content.
+
+    Args:
+        body (str): The raw email body text.
+
+    Returns:
+        str: Cleaned email body text.
+    """
+    try:
+        # Remove HTML tags using BeautifulSoup
+        soup = BeautifulSoup(body, "html.parser")
+        text = soup.get_text(separator=" ")
+
+        # Clean up the text
+        text = _clean_text(text)
+
+        return text.strip()
+    except Exception as e:
+        print(f"Error cleaning email body: {e}")
+        return body
+
+
+def _clean_text(text: str) -> str:
+    """
+    Clean up the text while preserving most content.
+
+    Args:
+        text (str): The input text.
+
+    Returns:
+        str: Cleaned text.
+    """
+    # Replace multiple newlines with a single newline
+    text = re.sub(r"\n+", "\n", text)
+
+    # Replace multiple spaces with a single space
+    text = re.sub(r"\s+", " ", text)
+
+    # Remove leading/trailing whitespace from each line
+    text = "\n".join(line.strip() for line in text.split("\n"))
+
+    return text
+
+
+def _update_datetime(day: Day | None, time: TimeSlot | None, time_zone: str) -> dict | None:
+    """
+    Update the datetime for a Google Calendar event.
+
+    Args:
+        day (Day | None): The day of the event.
+        time (TimeSlot | None): The time of the event.
+        time_zone (str): The time zone of the event.
+
+    Returns:
+        dict | None: The updated datetime for the event.
+    """
+    if day and time:
+        dt = datetime.combine(day.to_date(time_zone), time.to_time())
+        return {"dateTime": dt.isoformat(), "timeZone": time_zone}
+    return None
+
+
+def build_query_string(sender, recipient, subject, body, date_range):
+    """
+    Helper function to build a query string for Gmail list_emails_by_header tool.
+    """
+    query = []
+    if sender:
+        query.append(f"from:{sender}")
+    if recipient:
+        query.append(f"to:{recipient}")
+    if subject:
+        query.append(f"subject:{subject}")
+    if body:
+        query.append(body)
+    if date_range:
+        query.append(date_range.to_date_query())
+    return " ".join(query)
+
+
+def fetch_messages(service, query_string, limit):
+    """
+    Helper function to fetch messages from Gmail API for the list_emails_by_header tool.
+    """
+    response = (
+        service.users()
+        .messages()
+        .list(userId="me", q=query_string, maxResults=limit or 100)
+        .execute()
+    )
+    return response.get("messages", [])
+
+
+def remove_none_values(params: dict) -> dict:
+    """
+    Remove None values from a dictionary.
+    :param params: The dictionary to clean
+    :return: A new dictionary with None values removed
+    """
+    return {k: v for k, v in params.items() if v is not None}
+
+
+# Drive utils
+def build_drive_service(token: str):
+    """
+    Build a Drive service object.
+    """
+    return build("drive", "v3", credentials=Credentials(token))
+
+
+# Docs utils
+def build_docs_service(token: str):
+    """
+    Build a Drive service object.
+    """
+    return build("docs", "v1", credentials=Credentials(token))

arcade_google-0.0.13.dist-info/METADATA

@@ -0,0 +1,20 @@
+Metadata-Version: 2.1
+Name: arcade_google
+Version: 0.0.13
+Summary: Arcade tools for the entire google suite
+Author: Arcade AI
+Author-email: dev@arcade-ai.com
+Requires-Python: >=3.10,<4.0
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: arcade-ai (==0.0.13)
+Requires-Dist: beautifulsoup4 (>=4.10.0,<5.0.0)
+Requires-Dist: google-api-core (==2.19.1)
+Requires-Dist: google-api-python-client (==2.137.0)
+Requires-Dist: google-auth (==2.32.0)
+Requires-Dist: google-auth-httplib2 (==0.2.0)
+Requires-Dist: google-auth-oauthlib (==1.2.1)
+Requires-Dist: googleapis-common-protos (==1.63.2)

arcade_google-0.0.13.dist-info/RECORD

@@ -0,0 +1,11 @@
+arcade_google/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arcade_google/tools/__init__.py,sha256=MXM0xOEislCvl1UgwJC_Nqme83KjkQsaiUj4VKP0Q9s,49
+arcade_google/tools/calendar.py,sha256=YuMEapqYOiPU59hsXyVy70E-6i5Jt2xMCRRkuamEh1c,11782
+arcade_google/tools/docs.py,sha256=uu3-Nc0T0WQ8fzbqETjNBxkdy5FEZjt71Slu5NAgPRU,5075
+arcade_google/tools/drive.py,sha256=mVMaToUOgxXyjP96t2o1X72PEtD7628J5wBS9F6hcjg,3243
+arcade_google/tools/gmail.py,sha256=TtC_6S7-KVg9N2c7Xq8l1uYKhD5HUWYcDmZZce9613k,11421
+arcade_google/tools/models.py,sha256=um8mKisitNPimCx1HYX9spnKp1UzBWfRF4H2ducGn-8,9592
+arcade_google/tools/utils.py,sha256=jO9RpRm8N_NRhnGtpHDLKf_he1u2-3r8iSqR5JxaPfU,8355
+arcade_google-0.0.13.dist-info/METADATA,sha256=vKRnnhTXAOrEsrYy_6XUkmq3Q6XAH6a-76qlzImUZJM,798
+arcade_google-0.0.13.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+arcade_google-0.0.13.dist-info/RECORD,,

arcade_google-0.0.13.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 1.9.1
+Root-Is-Purelib: true
+Tag: py3-none-any