benchmark-runner 1.0.702__py3-none-any.whl → 1.0.704__py3-none-any.whl

benchmark_runner/common/google_drive/google_drive_operations.py

@@ -1,6 +1,5 @@
 
 import os
-import json
 
 from googleapiclient.discovery import build
 from google.oauth2.credentials import Credentials
@@ -20,40 +19,32 @@ class GoogleDriveOperations:
     1. credentials.json - Obtain from the "Google Cloud Console" under "APIs & Services" > "Credentials" > "Create Credentials" and select "OAuth 2.0 Client ID".
     2. token.json - This is generated automatically. If a browser is not available, copy the token from a machine with a browser.
     """
-    def __init__(self, google_drive_path: str, google_drive_credentials: str, google_drive_token: str, google_drive_shared_drive_id: str):
+    def __init__(self, google_drive_path: str, google_drive_token_file: str, google_drive_credentials_file: str, google_drive_shared_drive_id: str):
         """
         Initializes GoogleDriveOperations with authentication.
         """
-        self._google_drive_path = google_drive_path
-        self._google_drive_credentials = google_drive_credentials
-        self._google_drive_token = google_drive_token
+        self._google_drive_token = google_drive_token_file  # Read token from file
+        self._google_drive_credentials = google_drive_credentials_file  # Read credentials from file
         self._google_drive_shared_drive_id = google_drive_shared_drive_id
+        self._google_drive_path = google_drive_path
         self.creds = self.authenticate()
         self.service = build('drive', 'v3', credentials=self.creds)
 
     def authenticate(self):
-        """
-        Authenticates and returns credentials using token or interactive login.
-        :return: Credentials object
-        """
         creds = None
-
-        # Load credentials from the token string if it exists
-        if self._google_drive_token:
-            creds = Credentials.from_authorized_user_info(json.loads(self._google_drive_token), SCOPES)
-
-        # Check if the credentials are invalid or expired
+        # Load credentials from token.json if it exists
+        if os.path.exists(self._google_drive_token):
+            creds = Credentials.from_authorized_user_file(self._google_drive_token, SCOPES)
+        # If credentials are not valid or do not exist, authenticate
         if not creds or not creds.valid:
             if creds and creds.expired and creds.refresh_token:
                 creds.refresh(Request())
             else:
-                # Load client secrets from the credentials string and run the local server for auth
-                flow = InstalledAppFlow.from_client_config(json.loads(self._google_drive_credentials), SCOPES)
+                flow = InstalledAppFlow.from_client_secrets_file(self._google_drive_credentials, SCOPES)
                 creds = flow.run_local_server(port=0)
-
-            # Save the new token to the variable as a string
-            self._google_drive_token = creds.to_json()
-
+            # Save the credentials for the next run
+            with open(self._google_drive_credentials, 'w') as token_file:
+                token_file.write(creds.to_json())
         return creds
 
     def list_folders_and_files_in_shared_drive(self, shared_drive_id):

benchmark_runner/main/environment_variables.py

@@ -208,10 +208,10 @@ class EnvironmentVariables:
         self._environment_variables_dict['bucket'] = EnvironmentVariables.get_env('IBM_BUCKET', '')
         self._environment_variables_dict['key'] = EnvironmentVariables.get_env('IBM_KEY', '')
 
-        # Google drive
+        # Google drive, use file for credentials (jenkins limitation)
         self._environment_variables_dict['google_drive_path'] = EnvironmentVariables.get_env('GOOGLE_DRIVE_PATH', '')
-        self._environment_variables_dict['google_drive_credentials'] = EnvironmentVariables.get_env('GOOGLE_DRIVE_CREDENTIALS', '')
-        self._environment_variables_dict['google_drive_token'] = EnvironmentVariables.get_env('GOOGLE_DRIVE_TOKEN', '')
+        self._environment_variables_dict['google_drive_credentials_file'] = EnvironmentVariables.get_env('GOOGLE_DRIVE_CREDENTIALS_FILE', '')
+        self._environment_variables_dict['google_drive_token_file'] = EnvironmentVariables.get_env('GOOGLE_DRIVE_TOKEN_FILE', '')
         self._environment_variables_dict['google_drive_shared_drive_id'] = EnvironmentVariables.get_env('GOOGLE_DRIVE_SHARED_DRIVE_ID', '')
 
         # Grafana

benchmark_runner/workloads/workloads_operations.py

@@ -110,13 +110,13 @@ class WorkloadsOperations:
             self._windows_os = os.path.splitext(file_name)[0]
         # google drive
         self._google_drive_path = self._environment_variables_dict.get('google_drive_path', '')
-        self._google_drive_credentials = self._environment_variables_dict.get('google_drive_credentials', '')
-        self._google_drive_token = self._environment_variables_dict.get('google_drive_token', '')
+        self._google_drive_credentials_file = self._environment_variables_dict.get('google_drive_credentials_file', '')
+        self._google_drive_token_file = self._environment_variables_dict.get('google_drive_token_file', '')
         self._google_drive_shared_drive_id = self._environment_variables_dict.get('google_drive_shared_drive_id', '')
         if self._google_drive_path:
             self._google_drive_operation = GoogleDriveOperations(google_drive_path=self._google_drive_path,
-                                                                 google_drive_credentials=self._google_drive_credentials,
-                                                                 google_drive_token=self._google_drive_token,
+                                                                 google_drive_credentials_file=self._google_drive_credentials_file,
+                                                                 google_drive_token_file=self._google_drive_token_file,
                                                                  google_drive_shared_drive_id=self._google_drive_shared_drive_id)
         self._upgrade_ocp_version = self._environment_variables_dict.get('upgrade_ocp_version', '')
         self._run_strategy = self._environment_variables_dict.get('run_strategy', '')

{benchmark_runner-1.0.702.dist-info → benchmark_runner-1.0.704.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: benchmark-runner
-Version: 1.0.702
+Version: 1.0.704
 Summary: Benchmark Runner Tool
 Home-page: https://github.com/redhat-performance/benchmark-runner
 Author: Red Hat

{benchmark_runner-1.0.702.dist-info → benchmark_runner-1.0.704.dist-info}/RECORD

@@ -37,7 +37,7 @@ benchmark_runner/common/elasticsearch/elasticsearch_operations.py,sha256=fXckd9u
 benchmark_runner/common/github/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 benchmark_runner/common/github/github_operations.py,sha256=wp4zH0_4Qzb_c-GtQlP3VnN374H0aa_icGPk0pwkAZc,1111
 benchmark_runner/common/google_drive/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-benchmark_runner/common/google_drive/google_drive_operations.py,sha256=FLQT5E3xQkJUhtVEBdFUaYIMNbjkflYZTTQYCCVLc94,15632
+benchmark_runner/common/google_drive/google_drive_operations.py,sha256=PuL7pWKJRTTIwnRPM6XRBNyHe4Hg6g3ee9cYZnKrdJc,15513
 benchmark_runner/common/grafana/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 benchmark_runner/common/grafana/grafana_operations.py,sha256=Emt-p44OSe_XgVp6c94EVcDEaniyYtUDQMWCGIP0GR8,4877
 benchmark_runner/common/grafana/update_grafana_latest_value_mappings.py,sha256=aBKXx1spB8O2kjVsOl5heLl5lKcU07Hx8oHJLasJpSQ,5523
@@ -158,7 +158,7 @@ benchmark_runner/krkn_hub/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 benchmark_runner/krkn_hub/krknhub_exceptions.py,sha256=Hk7Co6zZ0u2RSmBmS4ZAi21ffZaQ2ITTfl6tGLtiAdY,180
 benchmark_runner/krkn_hub/krknhub_workloads.py,sha256=qNRZA-FBQZiT6h013WbP6zBRINh3c6YMnnlksN9fssA,2851
 benchmark_runner/main/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-benchmark_runner/main/environment_variables.py,sha256=T8A3n-UInN9wiKH5ovh4TCgXaMQ9uw_XThCG71v2JP0,27907
+benchmark_runner/main/environment_variables.py,sha256=-4ZJCC6lHmv777xXp_0W4X7KAbZZ183N9qwoFyBVhCQ,27974
 benchmark_runner/main/environment_variables_exceptions.py,sha256=UR0Ith0P0oshsDZdJRlRq8ZUTt0h8jFvUtrnP4m4AIY,437
 benchmark_runner/main/main.py,sha256=SB-rD4U_978gP8ojGUtIsYzhSsmzeEDsUfOiJQJdz9c,14097
 benchmark_runner/main/temporary_environment_variables.py,sha256=ODSHkfhgvdr_b2e3XyvykW21MVjSdyqimREyMc2klRE,957
@@ -169,9 +169,9 @@ benchmark_runner/workloads/vdbench_vm.py,sha256=Yhoz-GbvZwA8q6qGIeSUsYhEIERj8SmJ
 benchmark_runner/workloads/windows_vm.py,sha256=eHK79ueAkSlNC1uamz19o7CO20wzJi-UIqRuTByTVxg,2373
 benchmark_runner/workloads/workloads.py,sha256=F9fnk4h715tq7ANSCbDH0jktB8fpr_u3YG61Kdi5_os,1422
 benchmark_runner/workloads/workloads_exceptions.py,sha256=u7VII95iPRF_YhfpGH1U1RmgiIYESMOtbSF1dz7_ToE,1858
-benchmark_runner/workloads/workloads_operations.py,sha256=tAgvdMRMu5Ybck81UBCBE6H76PN0L2GptjNRnYoFv2M,25221
-benchmark_runner-1.0.702.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-benchmark_runner-1.0.702.dist-info/METADATA,sha256=lTYMLBqE9xGwf6N9_zvfiYTd3ICm4Vw9NoQzQnIa3nA,11309
-benchmark_runner-1.0.702.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-benchmark_runner-1.0.702.dist-info/top_level.txt,sha256=MP7UbTCzu59D53uKCZl5VsQeM_vheyMc7FmryczJQbk,17
-benchmark_runner-1.0.702.dist-info/RECORD,,
+benchmark_runner/workloads/workloads_operations.py,sha256=zhMAL-Zc2JtdI-LG4kxGwbGwHaLY2DfklpzM4bBG-eo,25261
+benchmark_runner-1.0.704.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+benchmark_runner-1.0.704.dist-info/METADATA,sha256=Fmd9LZkRbLjk6lLYvYdil9j8xJur42ccEPL-MAuSxrM,11309
+benchmark_runner-1.0.704.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+benchmark_runner-1.0.704.dist-info/top_level.txt,sha256=MP7UbTCzu59D53uKCZl5VsQeM_vheyMc7FmryczJQbk,17
+benchmark_runner-1.0.704.dist-info/RECORD,,