radiens-data-catalog 0.1.0__py3-none-any.whl

radiens_data_catalog/__init__.py

@@ -0,0 +1,27 @@
+"""
+radiens-data-catalog
+---------------------
+Programmatic catalog and sync tool for xdat neural recordings stored on
+Google Drive. Recordings are uniquely identified by (drive_path, base_name)
+and queryable by drive_path and/or base_name, each via exact match, prefix,
+or substring (see :data:`MatchMode`).
+"""
+
+from .catalog import Catalog, MatchMode, PrefetchResult, ScanResult, match_series
+from .config import Config
+from .drive import DriveItemEntry, RecordingEntry
+from .exceptions import AmbiguousRecordingError, CatalogError, EntryNotFoundError
+
+__all__ = [
+    "AmbiguousRecordingError",
+    "Catalog",
+    "CatalogError",
+    "Config",
+    "DriveItemEntry",
+    "EntryNotFoundError",
+    "MatchMode",
+    "PrefetchResult",
+    "RecordingEntry",
+    "ScanResult",
+    "match_series",
+]

radiens_data_catalog/auth.py

@@ -0,0 +1,30 @@
+"""
+auth.py
+-------
+Service account authentication for the Google Drive API.
+"""
+
+from google.oauth2 import service_account
+from googleapiclient.discovery import build
+
+from .drive import DriveService
+
+SCOPES: list[str] = ["https://www.googleapis.com/auth/drive.readonly"]
+
+
+def build_drive_service(credentials_path: str) -> DriveService:
+    """Build an authenticated Google Drive API service client.
+
+    Uses a service account credentials file for authentication. The Drive
+    folders you want to access must be shared with the service account email
+    found in the credentials JSON.
+
+    Args:
+        credentials_path: Path to the Google service account credentials JSON
+            file.
+
+    Returns:
+        An authenticated Drive API service object (read-only scope).
+    """
+    creds = service_account.Credentials.from_service_account_file(credentials_path, scopes=SCOPES)
+    return build("drive", "v3", credentials=creds)