airbyte-source-google-drive 0.2.0.dev202501272304__tar.gz → 0.2.0.dev202501291717__tar.gz

{airbyte_source_google_drive-0.2.0.dev202501272304 → airbyte_source_google_drive-0.2.0.dev202501291717}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: airbyte-source-google-drive
-Version: 0.2.0.dev202501272304
+Version: 0.2.0.dev202501291717
 Summary: Source implementation for Google Drive.
 License: ELv2
 Author: Airbyte
@@ -10,7 +10,7 @@ Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Requires-Dist: airbyte-cdk[file-based] (==6.26.0.dev04105)
+Requires-Dist: airbyte-cdk[file-based] (==6.26.0.dev04106)
 Requires-Dist: google-api-python-client (==2.104.0)
 Requires-Dist: google-api-python-client-stubs (==1.18.0)
 Requires-Dist: google-auth-httplib2 (==0.1.1)

{airbyte_source_google_drive-0.2.0.dev202501272304 → airbyte_source_google_drive-0.2.0.dev202501291717}/pyproject.toml

@@ -5,7 +5,7 @@ requires = [
 build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
-version = "0.2.0.dev202501272304"
+version = "0.2.0.dev202501291717"
 name = "airbyte-source-google-drive"
 description = "Source implementation for Google Drive."
 authors = [
@@ -31,7 +31,7 @@ google-api-python-client-stubs = "==1.18.0"
 extras = [
     "file-based",
 ]
-version = "6.26.0.dev04105"
+version = "6.26.0.dev04106"
 
 [tool.poetry.scripts]
 source-google-drive = "source_google_drive.run:run"

{airbyte_source_google_drive-0.2.0.dev202501272304 → airbyte_source_google_drive-0.2.0.dev202501291717}/source_google_drive/stream_reader.py

@@ -27,9 +27,6 @@ from source_google_drive.utils import get_folder_id
 from .exceptions import ErrorDownloadingFile, ErrorFetchingMetadata
 from .spec import SourceGoogleDriveSpec
 
-# remove this, just using while test credentials are prepared
-from .temp_mock import _get_looping_google_api_list_response
-
 
 FOLDER_MIME_TYPE = "application/vnd.google-apps.folder"
 GOOGLE_DOC_MIME_TYPE = "application/vnd.google-apps.document"
@@ -163,9 +160,8 @@ class SourceGoogleDriveStreamReader(AbstractFileBasedStreamReader):
                 for item in items:
                     yield item
         except Exception as e:
-            logger.info(f"There was an error listing {key} with {args}: {str(e)}")
-            logger.info(f"Backing off to mocked data for development purposes")
-            yield from _get_looping_google_api_list_response(service, key, args)
+            logger.error(f"There was an error listing {key} with {args}: {str(e)}")
+            raise e
 
     def load_identity_groups(self, logger: logging.Logger) -> Dict[str, Any]:
         domain = self.config.delivery_method.domain
@@ -208,8 +204,6 @@ class SourceGoogleDriveStreamReader(AbstractFileBasedStreamReader):
 
             yield rfp.dict()
 
-        return ""
-
     def get_matching_files(self, globs: List[str], prefix: Optional[str], logger: logging.Logger) -> Iterable[RemoteFile]:
         """
         Get all files matching the specified glob patterns.

airbyte_source_google_drive-0.2.0.dev202501272304/source_google_drive/temp_mock.py

@@ -1,36 +0,0 @@
-# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
-
-from typing import Any, Iterator
-
-
-def _get_looping_google_api_list_response(service: Any, key: str, args: dict[str, Any]) -> Iterator[dict[str, Any]]:
-    domain = args.get("domain") or args.get("groupKey")
-
-    if key == "users":
-        # Mocking 5 users
-        for i in range(1, 6):
-            yield {
-                "primaryEmail": f"user{i}@{domain}",
-                "name": {"fullName": f"User {i}"},
-                "emails": [{"address": f"user{i}@{domain}"}],
-            }
-
-    elif key == "groups":
-        # Mocking 5 groups
-        for i in range(1, 6):
-            yield {
-                "email": f"group{i}@{domain}",
-                "name": f"Group {i}",
-                "id": f"group-id-{i}",
-            }
-
-    elif key == "members":
-        # Mocking 5 members per group
-        group_id = args.get("groupKey")
-        for i in range(1, 6):
-            yield {
-                "email": f"member{i}@{domain}",
-                "role": "MEMBER",
-                "type": "USER",
-                "groupId": group_id,
-            }