lumera 0.9.1__py3-none-any.whl → 0.9.2__py3-none-any.whl

lumera/integrations/__init__.py

@@ -0,0 +1,34 @@
+"""
+Lumera SDK Integrations
+
+Third-party service integrations with Lumera credential management.
+
+Each integration module provides:
+- A `get_*_client()` or `get_*_service()` function that returns an authenticated client
+- Optional helper functions for common Lumera patterns
+
+Example:
+    from lumera.integrations import google, get_access_token
+
+    # Google Sheets with Lumera-managed OAuth
+    sheets = google.get_sheets_service()
+    data = sheets.spreadsheets().values().get(...)
+
+    # Google Drive
+    drive = google.get_drive_service()
+    files = drive.files().list().execute()
+
+    # Get raw access token for any provider
+    token = get_access_token("slack")
+
+Available integrations:
+- `google` - Google APIs (Sheets, Drive)
+
+Utilities:
+- `get_access_token(provider)` - Get OAuth token for any Lumera-connected provider
+"""
+
+from .._utils import get_access_token
+from . import google
+
+__all__ = ["get_access_token", "google"]

lumera/integrations/google.py

@@ -0,0 +1,338 @@
+"""
+Google API Integration
+
+Provides authenticated Google API clients using Lumera-managed OAuth credentials.
+
+Example:
+    from lumera.integrations import google
+
+    # Get authenticated Sheets service
+    sheets = google.get_sheets_service()
+    data = sheets.spreadsheets().values().get(
+        spreadsheetId="...",
+        range="Sheet1!A1:D10"
+    ).execute()
+
+    # Get authenticated Drive service
+    drive = google.get_drive_service()
+    files = drive.files().list().execute()
+
+    # Share credentials between services
+    creds = google.get_credentials()
+    sheets = google.get_sheets_service(credentials=creds)
+    drive = google.get_drive_service(credentials=creds)
+"""
+
+import io
+import logging
+import os
+import re
+from typing import TYPE_CHECKING, Optional, Tuple
+
+# When type checking we want access to the concrete ``Resource`` class that
+# ``googleapiclient.discovery.build`` returns. Importing it unconditionally
+# would require ``googleapiclient`` to be available in every execution
+# environment – something we cannot guarantee.  By guarding the import with
+# ``TYPE_CHECKING`` we give static analysers (ruff, mypy, etc.) the
+# information they need without introducing a hard runtime dependency.
+# During static analysis we want to import ``Resource`` so that it is a known
+# name for type checkers, but we don't require this import at runtime. Guard
+# it with ``TYPE_CHECKING`` to avoid hard dependencies.
+if TYPE_CHECKING:  # pragma: no cover
+    from googleapiclient.discovery import Resource  # noqa: F401
+
+# Always ensure that the symbol ``Resource`` exists at runtime to placate static
+# analysers like ruff (F821) that inspect the AST without executing the code.
+try:  # pragma: no cover – optional runtime import
+    from googleapiclient.discovery import Resource  # type: ignore
+except ModuleNotFoundError:  # pragma: no cover – provide a stub fallback
+
+    class Resource:  # noqa: D401
+        """Stub replacement for ``googleapiclient.discovery.Resource``."""
+
+        pass
+
+
+from google.oauth2.credentials import Credentials
+from googleapiclient.discovery import build
+from googleapiclient.http import MediaFileUpload, MediaIoBaseDownload
+
+from .._utils import get_access_token
+
+# Module logger
+logger = logging.getLogger(__name__)
+
+# =====================================================================================
+# Configuration
+# =====================================================================================
+
+MIME_GOOGLE_SHEET = "application/vnd.google-apps.spreadsheet"
+MIME_EXCEL = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+
+# =====================================================================================
+# Authentication & Service Initialization
+# =====================================================================================
+
+
+def get_credentials() -> Credentials:
+    """
+    Retrieves a Google OAuth token from Lumera and
+    converts it into a Credentials object usable by googleapiclient.
+
+    Returns:
+        google.oauth2.credentials.Credentials: Authenticated credentials object.
+
+    Raises:
+        RuntimeError: If LUMERA_TOKEN is not set or token fetch fails.
+    """
+    logger.debug("Fetching Google access token from Lumera…")
+    access_token = get_access_token("google")
+    logger.debug("Access token received.")
+    creds = Credentials(token=access_token)
+    logger.debug("Credentials object created.")
+    return creds
+
+
+# Backward compatibility alias
+get_google_credentials = get_credentials
+
+
+def get_sheets_service(credentials: Optional[Credentials] = None) -> "Resource":
+    """
+    Initializes and returns the Google Sheets API service.
+
+    If no credentials are provided, this function will automatically fetch a
+    Google access token from Lumera and construct the appropriate
+    ``google.oauth2.credentials.Credentials`` instance.
+
+    Args:
+        credentials: Optional pre-fetched credentials. If None, fetches from Lumera.
+
+    Returns:
+        Google Sheets API service resource.
+    """
+    if credentials is None:
+        logger.info("No credentials provided; fetching Google token…")
+        credentials = get_credentials()
+    logger.info("Google Sheets API service being initialized…")
+    return build("sheets", "v4", credentials=credentials)
+
+
+def get_drive_service(credentials: Optional[Credentials] = None) -> "Resource":
+    """
+    Initializes and returns the Google Drive API service.
+
+    If no credentials are provided, this function will automatically fetch a
+    Google access token from Lumera and construct the appropriate
+    ``google.oauth2.credentials.Credentials`` instance.
+
+    Args:
+        credentials: Optional pre-fetched credentials. If None, fetches from Lumera.
+
+    Returns:
+        Google Drive API service resource.
+    """
+    if credentials is None:
+        logger.info("No credentials provided; fetching Google token…")
+        credentials = get_credentials()
+    logger.info("Google Drive API service being initialized…")
+    return build("drive", "v3", credentials=credentials)
+
+
+# =====================================================================================
+# Google Sheets & Drive Utility Functions
+# =====================================================================================
+
+
+def get_spreadsheet_and_sheet_id(
+    service: "Resource", spreadsheet_url: str, tab_name: str
+) -> Tuple[Optional[str], Optional[int]]:
+    """
+    Given a Google Sheets URL and a tab (sheet) name, returns a tuple:
+    (spreadsheet_id, sheet_id)
+    """
+    spreadsheet_id = _extract_spreadsheet_id(spreadsheet_url)
+    if not spreadsheet_id:
+        return None, None
+
+    sheet_id = _get_sheet_id_from_name(service, spreadsheet_id, tab_name)
+    return spreadsheet_id, sheet_id
+
+
+def _extract_spreadsheet_id(spreadsheet_url: str) -> Optional[str]:
+    """Extracts the spreadsheet ID from a Google Sheets URL."""
+    logger.debug(f"Extracting spreadsheet ID from URL: {spreadsheet_url}")
+    pattern = r"/d/([a-zA-Z0-9-_]+)"
+    match = re.search(pattern, spreadsheet_url)
+    if match:
+        spreadsheet_id = match.group(1)
+        logger.debug(f"Spreadsheet ID extracted: {spreadsheet_id}")
+        return spreadsheet_id
+    logger.warning("Could not extract Spreadsheet ID.")
+    return None
+
+
+def _get_sheet_id_from_name(
+    service: "Resource", spreadsheet_id: str, tab_name: str
+) -> Optional[int]:
+    """Uses the Google Sheets API to fetch the sheet ID corresponding to 'tab_name'."""
+    logger.debug(f"Requesting sheet metadata for spreadsheet ID: {spreadsheet_id}")
+    response = (
+        service.spreadsheets()
+        .get(spreadsheetId=spreadsheet_id, fields="sheets.properties")
+        .execute()
+    )
+    logger.debug("Metadata received. Searching for tab…")
+
+    for sheet in response.get("sheets", []):
+        properties = sheet.get("properties", {})
+        if properties.get("title") == tab_name:
+            sheet_id = properties.get("sheetId")
+            logger.debug(f"Match found for tab '{tab_name}'. Sheet ID is {sheet_id}")
+            return sheet_id
+    logger.warning(f"No sheet found with tab name '{tab_name}'.")
+    return None
+
+
+def sheet_name_from_gid(service: "Resource", spreadsheet_id: str, gid: int) -> Optional[str]:
+    """Resolve a sheet's human-readable name (title) from its gid."""
+    logger.debug(f"Resolving sheet name from gid={gid} …")
+    meta = (
+        service.spreadsheets()
+        .get(
+            spreadsheetId=spreadsheet_id,
+            includeGridData=False,
+            fields="sheets(properties(sheetId,title))",
+        )
+        .execute()
+    )
+
+    for sheet in meta.get("sheets", []):
+        props = sheet.get("properties", {})
+        if props.get("sheetId") == gid:
+            title = props["title"]
+            logger.debug(f"Sheet gid={gid} corresponds to sheet name='{title}'.")
+            return title
+    logger.warning(f"No sheet found with gid={gid}")
+    return None
+
+
+def read_cell(service: "Resource", spreadsheet_id: str, range_a1: str) -> Optional[str]:
+    """Fetch a single cell value (as string); returns None if empty."""
+    logger.debug(f"Reading cell '{range_a1}' …")
+    resp = (
+        service.spreadsheets()
+        .values()
+        .get(spreadsheetId=spreadsheet_id, range=range_a1, majorDimension="ROWS")
+        .execute()
+    )
+
+    values = resp.get("values", [])
+    return values[0][0] if values and values[0] else None
+
+
+# NOTE: The function performs I/O side-effects and does not return a value.
+def download_file_direct(drive_service: "Resource", file_id: str, dest_path: str) -> None:
+    """
+    Downloads a file directly from Google Drive using files().get_media
+    without any format conversion.
+    """
+    logger.info(f"Initiating direct download for file ID: {file_id}")
+
+    request = drive_service.files().get_media(fileId=file_id)
+    fh = io.BytesIO()
+    downloader = MediaIoBaseDownload(fh, request)
+
+    done = False
+    while not done:
+        status, done = downloader.next_chunk()
+        if status:
+            logger.debug(f"Download progress: {int(status.progress() * 100)}%")
+
+    with open(dest_path, "wb") as f:
+        f.write(fh.getvalue())
+    logger.info(f"File saved to: {dest_path}")
+
+
+def upload_excel_as_google_sheet(
+    drive_service: "Resource", local_path: str, desired_name: str
+) -> Tuple[Optional[str], Optional[str]]:
+    """
+    Uploads a local XLSX file to Google Drive, converting it to Google Sheets format.
+    Returns the file ID and web link.
+    """
+    logger.info(f"Preparing to upload '{local_path}' as Google Sheet named '{desired_name}'")
+
+    if not os.path.isfile(local_path):
+        logger.error(f"Local file not found at '{local_path}'. Aborting.")
+        return None, None
+
+    media = MediaFileUpload(local_path, mimetype=MIME_EXCEL, resumable=True)
+    file_metadata = {"name": desired_name, "mimeType": MIME_GOOGLE_SHEET}
+
+    logger.info("Initiating Google Drive upload & conversion…")
+    request = drive_service.files().create(
+        body=file_metadata, media_body=media, fields="id, webViewLink"
+    )
+
+    response = None
+    while response is None:
+        status, response = request.next_chunk()
+        if status:
+            logger.debug(f"Upload progress: {int(status.progress() * 100)}%")
+
+    file_id = response.get("id")
+    web_view_link = response.get("webViewLink")
+    logger.info(f"Upload completed. File ID: {file_id}")
+    return file_id, web_view_link
+
+
+# Remove rows from a sheet.  All parameters are 1-based (both *start_row* and
+# *end_row* are inclusive) mirroring the UI behaviour in Google Sheets.
+def delete_rows_api_call(
+    service: "Resource",
+    spreadsheet_id: str,
+    sheet_gid: int,
+    start_row: int,
+    end_row: int,
+) -> None:
+    """Executes the API call to delete rows."""
+    logger.info(f"Deleting rows {start_row}-{end_row} (1-based inclusive)…")
+
+    body = {
+        "requests": [
+            {
+                "deleteDimension": {
+                    "range": {
+                        "sheetId": sheet_gid,
+                        "dimension": "ROWS",
+                        "startIndex": start_row - 1,  # 0-based
+                        "endIndex": end_row,  # end-exclusive
+                    }
+                }
+            }
+        ]
+    }
+    service.spreadsheets().batchUpdate(spreadsheetId=spreadsheet_id, body=body).execute()
+    logger.info("Rows deleted.")
+
+
+__all__ = [
+    # Authentication
+    "get_credentials",
+    "get_google_credentials",  # backward compat alias
+    # Services
+    "get_sheets_service",
+    "get_drive_service",
+    # Sheets helpers
+    "get_spreadsheet_and_sheet_id",
+    "sheet_name_from_gid",
+    "read_cell",
+    "delete_rows_api_call",
+    # Drive helpers
+    "download_file_direct",
+    "upload_excel_as_google_sheet",
+    # Constants
+    "MIME_GOOGLE_SHEET",
+    "MIME_EXCEL",
+]

{lumera-0.9.1.dist-info → lumera-0.9.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lumera
-Version: 0.9.1
+Version: 0.9.2
 Summary: SDK for building on Lumera platform
 Requires-Python: >=3.11
 Requires-Dist: requests

{lumera-0.9.1.dist-info → lumera-0.9.2.dist-info}/RECORD

@@ -9,7 +9,9 @@ lumera/pb.py,sha256=fTGjcclIGmXxQsBCDhXVrPyKBkdPJ3DRwNZfICT5X8E,20434
 lumera/sdk.py,sha256=0Q4b0LXbr7Vl0SpstH1IKEZwGek33Jz-4ICMRe_Z60A,31217
 lumera/storage.py,sha256=b0W6JNSGfmhJIcmK3vrATXAwxIr_bfrj-hPuQRVLTYU,8206
 lumera/webhooks.py,sha256=L_Q5YHBJKQNpv7G9Nq0QqlGMRch6x9ptlwu1xD2qwUc,8661
-lumera-0.9.1.dist-info/METADATA,sha256=gbS5AHAHUaw4sh8EHbKKb1ONZnzD2ph-lOidiUv49iE,1611
-lumera-0.9.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lumera-0.9.1.dist-info/top_level.txt,sha256=HgfK4XQkpMTnM2E5iWM4kB711FnYqUY9dglzib3pWlE,7
-lumera-0.9.1.dist-info/RECORD,,
+lumera/integrations/__init__.py,sha256=LnJmAnFB_p3YMKyeGVdDP4LYlJ85XFNQFAxGo6zF7CI,937
+lumera/integrations/google.py,sha256=QkbBbbDh3I_OToPDFqcivU6sWy2UieHBxZ_TPv5rqK0,11862
+lumera-0.9.2.dist-info/METADATA,sha256=Z26CWbc_x9lmH4nKp_7FQ4JWd0QSsgxiLQC-K6O_QgA,1611
+lumera-0.9.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lumera-0.9.2.dist-info/top_level.txt,sha256=HgfK4XQkpMTnM2E5iWM4kB711FnYqUY9dglzib3pWlE,7
+lumera-0.9.2.dist-info/RECORD,,