brynq-sdk-google-drive 1.0.0__tar.gz → 2.0.0__tar.gz

{brynq_sdk_google_drive-1.0.0 → brynq_sdk_google_drive-2.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq_sdk_google_drive
-Version: 1.0.0
+Version: 2.0.0
 Summary: Google Drive wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_google_drive-1.0.0 → brynq_sdk_google_drive-2.0.0}/brynq_sdk_google_drive/google_drive.py

@@ -1,5 +1,5 @@
 import os
-from typing import List, Union
+from typing import List, Union, Literal, Optional
 import requests
 import json
 from io import BytesIO
@@ -9,24 +9,31 @@ from googleapiclient.http import MediaFileUpload
 from brynq_sdk_brynq import BrynQ
 
 class GoogleDrive(BrynQ):
-    def __init__(self, label: Union[str, List], debug: bool = False):
+    def __init__(self, system_type: Optional[Literal['source', 'target']] = None, debug: bool = False):
         """
         :param label: label of the Google Drive system in BrynQ
         :param debug: set to True to enable debug logging
         """
         super().__init__()
         api_version = 'v3'
+        self.system_type = system_type
         self.base_url = f'https://www.googleapis.com/drive/{api_version}/'
-        credentials = self.get_system_credential(system='google-drive', label=label)
+        credentials = self.interfaces.credentials.get(system="google-drive", system_type=system_type)
+        self.credentials = credentials.get('data')
         self.debug = debug
         if self.debug:
             print(f"credentials: {credentials}")
-        self.access_token = credentials['auth']['access_token']
-        self.brynq_system_id = credentials['id']
+        self.access_token = credentials['access_token']
+        self.timeout = 3600
+
+    def _refresh_credentials(self):
+        credentials = self.interfaces.credentials.get(system="sharepoint", system_type=self.system_type)
+        credentials = credentials.get('data')
+        self.access_token = credentials['access_token']
 
     def _get_google_drive_headers(self):
-        access_token = self.refresh_system_credential(system='google-drive', system_id=self.brynq_system_id)['access_token']
-        headers = {'Authorization': f'Bearer {access_token}'}
+        self._refresh_credentials()
+        headers = {'Authorization': f'Bearer {self.access_token}'}
         if self.debug:
             print(headers)
         return headers
@@ -45,7 +52,7 @@ class GoogleDrive(BrynQ):
         if drive_id:
             params['driveId'] = drive_id
             params['corpora'] = 'drive'
-        response = requests.get(url=url, headers=headers, params=params)
+        response = requests.get(url=url, headers=headers, params=params, timeout=self.timeout)
         response.raise_for_status()
         return response
 
@@ -55,8 +62,7 @@ class GoogleDrive(BrynQ):
         local_file_path: local path of the file you want to upload
         remote_file_path: remote path of the folder and filename where you want to place the file
         """
-        credentials = self.refresh_system_credential(system='google-drive', system_id=self.brynq_system_id)
-        service = build('drive', 'v3', credentials=credentials)
+        service = build('drive', 'v3', credentials=self.credentials)
 
         file_metadata = {'name': os.path.basename(remote_file_path)}
         media = MediaFileUpload(local_file_path, resumable=True)
@@ -81,7 +87,7 @@ class GoogleDrive(BrynQ):
         """
         url = f'{self.base_url}files/{file_id}/export?mimeType={mime_type}'
         headers = self._get_google_drive_headers()
-        response = requests.get(url=url, headers=headers)
+        response = requests.get(url=url, headers=headers, timeout=self.timeout)
         response.raise_for_status()
         with open(local_file_path, 'wb') as f:
             f.write(response.content)
@@ -93,8 +99,7 @@ class GoogleDrive(BrynQ):
         local_folder_path: local folder where the files will be downloaded to
         remote_folder_path: remote path of the folder you want to get on Google Drive
         """
-        credentials = self.refresh_system_credential(system='google-drive', system_id=self.brynq_system_id)
-        service = build('drive', 'v3', credentials=credentials)
+        service = build('drive', 'v3', credentials=self.credentials)
 
         # List files in the specified remote folder
         query = f"'{remote_folder_path}' in parents"
@@ -129,8 +134,7 @@ class GoogleDrive(BrynQ):
         remote_file_path: complete path including filename
         :return: response from Google Drive
         """
-        credentials = self.refresh_system_credential(system='google-drive', system_id=self.brynq_system_id)
-        service = build('drive', 'v3', credentials=credentials)
+        service = build('drive', 'v3', credentials=self.credentials)
 
         # Find the file ID
         query = f"name = '{remote_file_path}'"
@@ -152,8 +156,7 @@ class GoogleDrive(BrynQ):
         Remove a folder from Google Drive
         folder: folder id that you want to delete
         """
-        credentials = self.refresh_system_credential(system='google-drive', system_id=self.brynq_system_id)
-        service = build('drive', 'v3', credentials=credentials)
+        service = build('drive', 'v3', credentials=self.credentials)
 
         response = service.files().delete(fileId=folder_id).execute()
         if self.debug:

{brynq_sdk_google_drive-1.0.0 → brynq_sdk_google_drive-2.0.0}/brynq_sdk_google_drive.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq-sdk-google-drive
-Version: 1.0.0
+Version: 2.0.0
 Summary: Google Drive wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_google_drive-1.0.0 → brynq_sdk_google_drive-2.0.0}/brynq_sdk_google_drive.egg-info/requires.txt

@@ -1,3 +1,3 @@
-brynq-sdk-brynq>=2
+brynq-sdk-brynq<5,>=4
 google-api-python-client<3,>=2
 requests<=3,>=2

{brynq_sdk_google_drive-1.0.0 → brynq_sdk_google_drive-2.0.0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 
 setup(
     name='brynq_sdk_google_drive',
-    version='1.0.0',
+    version='2.0.0',
     description='Google Drive wrapper from BrynQ',
     long_description='Groogle Drive wrapper from BrynQ',
     author='BrynQ',
@@ -10,7 +10,7 @@ setup(
     packages=find_namespace_packages(include=['brynq_sdk*']),
     license='BrynQ License',
     install_requires=[
-        'brynq-sdk-brynq>=2',
+        'brynq-sdk-brynq>=4,<5',
         'google-api-python-client>=2,<3',
         'requests>=2,<=3'
     ],