fivetran-connector-sdk 0.12.12.1__py3-none-any.whl → 0.12.17.1__py3-none-any.whl

fivetran_connector_sdk/__init__.py

@@ -26,7 +26,7 @@ from fivetran_connector_sdk.protos import common_pb2
 from fivetran_connector_sdk.protos import connector_sdk_pb2
 from fivetran_connector_sdk.protos import connector_sdk_pb2_grpc
 
-__version__ = "0.12.12.1"
+__version__ = "0.12.17.1"
 
 MAC_OS = "mac"
 WIN_OS = "windows"
@@ -38,6 +38,7 @@ VERSION_FILENAME = "version.txt"
 UPLOAD_FILENAME = "code.zip"
 LAST_VERSION_CHECK_FILE = "_last_version_check"
 ROOT_LOCATION = ".ft_sdk_connector_tester"
+CONFIG_FILE = "_config.json"
 OUTPUT_FILES_DIR = "files"
 REQUIREMENTS_TXT = "requirements.txt"
 PYPI_PACKAGE_DETAILS_URL = "https://pypi.org/pypi/fivetran_connector_sdk/json"
@@ -57,6 +58,7 @@ COMMANDS_AND_SYNONYMS = {
 CONNECTION_SCHEMA_NAME_PATTERN = r'^[_a-z][_a-z0-9]*$'
 DEBUGGING = False
 EXECUTED_VIA_CLI = False
+PRODUCTION_BASE_URL = "https://api.fivetran.com"
 TABLES = {}
 
 JAVA_LONG_MAX_VALUE = 9223372036854775807
@@ -508,6 +510,18 @@ def get_available_port():
     return None
 
 
+def update_base_url_if_required():
+    config_file_path = os.path.join(_tester_root_dir(), CONFIG_FILE)
+    if os.path.isfile(config_file_path):
+        with open(config_file_path, 'r') as f:
+            data = json.load(f)
+            base_url = data.get('production_base_url')
+            if base_url is not None:
+                global PRODUCTION_BASE_URL
+                PRODUCTION_BASE_URL = base_url
+                print_library_log(f"Updating PRODUCTION_BASE_URL to: {base_url}")
+
+
 class Connector(connector_sdk_pb2_grpc.ConnectorServicer):
     def __init__(self, update, schema=None):
         """Initializes the Connector instance.
@@ -522,6 +536,8 @@ class Connector(connector_sdk_pb2_grpc.ConnectorServicer):
         self.configuration = None
         self.state = None
 
+        update_base_url_if_required()
+
     @staticmethod
     def fetch_requirements_from_file(file_path: str) -> list[str]:
         """Reads a requirements file and returns a list of dependencies.
@@ -746,7 +762,7 @@ class Connector(connector_sdk_pb2_grpc.ConnectorServicer):
         if not config["secrets_list"]:
             del config["secrets_list"]
 
-        resp = rq.patch(f"https://api.fivetran.com/v1/connectors/{id}",
+        resp = rq.patch(f"{PRODUCTION_BASE_URL}/v1/connectors/{id}",
                         headers={"Authorization": f"Basic {deploy_key}"},
                         json={
                             "config": config,
@@ -771,7 +787,7 @@ class Connector(connector_sdk_pb2_grpc.ConnectorServicer):
         Returns:
             str: The connection ID, or None
         """
-        resp = rq.get(f"https://api.fivetran.com/v1/groups/{group_id}/connectors",
+        resp = rq.get(f"{PRODUCTION_BASE_URL}/v1/groups/{group_id}/connectors",
                       headers={"Authorization": f"Basic {deploy_key}"},
                       params={"schema": name})
         if not resp.ok:
@@ -796,7 +812,7 @@ class Connector(connector_sdk_pb2_grpc.ConnectorServicer):
         Returns:
             rq.Response: The response object.
         """
-        response = rq.post(f"https://api.fivetran.com/v1/connectors",
+        response = rq.post(f"{PRODUCTION_BASE_URL}/v1/connectors",
                            headers={"Authorization": f"Basic {deploy_key}"},
                            json={
                                "group_id": group_id,
@@ -888,7 +904,7 @@ class Connector(connector_sdk_pb2_grpc.ConnectorServicer):
             bool: True if the upload was successful, False otherwise.
         """
         print_library_log("Uploading your project...")
-        response = rq.post(f"https://api.fivetran.com/v1/deploy/{group_id}/{connection}",
+        response = rq.post(f"{PRODUCTION_BASE_URL}/v1/deploy/{group_id}/{connection}",
                            files={'file': open(local_path, 'rb')},
                            headers={"Authorization": f"Basic {deploy_key}"})
         if response.ok:
@@ -911,7 +927,7 @@ class Connector(connector_sdk_pb2_grpc.ConnectorServicer):
             bool: True if the cleanup was successful, False otherwise.
         """
         print_library_log("INFO: Cleaning up your uploaded project ")
-        response = rq.post(f"https://api.fivetran.com/v1/cleanup_code/{group_id}/{connection}",
+        response = rq.post(f"{PRODUCTION_BASE_URL}/v1/cleanup_code/{group_id}/{connection}",
                            headers={"Authorization": f"Basic {deploy_key}"})
         if response.ok:
             print("✓")
@@ -947,7 +963,7 @@ class Connector(connector_sdk_pb2_grpc.ConnectorServicer):
         Returns:
             tuple[str, str]: A tuple containing the group ID and group name.
         """
-        groups_url = "https://api.fivetran.com/v1/groups"
+        groups_url = f"{PRODUCTION_BASE_URL}/v1/groups"
 
         params = {"limit": 500}
         headers = {"Authorization": f"Basic {deploy_key}"}
@@ -1096,8 +1112,7 @@ class Connector(connector_sdk_pb2_grpc.ConnectorServicer):
                 print("✓")
             except:
                 shutil.rmtree(tester_root_dir)
-                raise RuntimeError(f"\nSEVERE: Failed to install the connector tester. Error details: ",
-                                   traceback.format_exc())
+                raise RuntimeError(f"\nSEVERE: Failed to install the connector tester. Error details: {traceback.format_exc()}")
 
         project_path = os.getcwd() if project_path is None else project_path
         self.validate_requirements_file(project_path, False)
@@ -1114,7 +1129,7 @@ class Connector(connector_sdk_pb2_grpc.ConnectorServicer):
         # server.wait_for_termination()
 
         try:
-            print_library_log(f"Running connector tester...")
+            print_library_log("Running connector tester...")
             for log_msg in self.__run_tester(java_exe, tester_root_dir, project_path, available_port):
                 print(log_msg, end="")
         except:

{fivetran_connector_sdk-0.12.12.1.dist-info → fivetran_connector_sdk-0.12.17.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fivetran_connector_sdk
-Version: 0.12.12.1
+Version: 0.12.17.1
 Summary: Build custom connectors on Fivetran platform
 Author-email: Fivetran <developers@fivetran.com>
 Project-URL: Homepage, https://fivetran.com/docs/connectors/connector-sdk

{fivetran_connector_sdk-0.12.12.1.dist-info → fivetran_connector_sdk-0.12.17.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-fivetran_connector_sdk/__init__.py,sha256=7omt3-H7FF5Tq7ClT535hVAk6qoI7NGoSHNDXAl5t9w,63395
+fivetran_connector_sdk/__init__.py,sha256=DFQiZdtyqblI_7npsJzaZ7jicU0U1B5LXA1SZY-WeYI,63949
 fivetran_connector_sdk/protos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fivetran_connector_sdk/protos/common_pb2.py,sha256=kUwVcyZHgLigNR-KnHZn7dHrlxaMnUXqzprsRx6T72M,6831
 fivetran_connector_sdk/protos/common_pb2.pyi,sha256=S0hdIzoXyyOKD5cjiGeDDLYpQ9J3LjAvu4rCj1JvJWE,9038
@@ -6,8 +6,8 @@ fivetran_connector_sdk/protos/common_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXH
 fivetran_connector_sdk/protos/connector_sdk_pb2.py,sha256=9Ke_Ti1s0vAeXapfXT-EryrT2-TSGQb8mhs4gxTpUMk,7732
 fivetran_connector_sdk/protos/connector_sdk_pb2.pyi,sha256=FWYxRgshEF3QDYAE0TM_mv4N2gGvkxCH_uPpxnMc4oA,8406
 fivetran_connector_sdk/protos/connector_sdk_pb2_grpc.py,sha256=ZfJLp4DW7uP4pFOZ74s_wQ6tD3eIPi-08UfnLwe4tzo,7163
-fivetran_connector_sdk-0.12.12.1.dist-info/METADATA,sha256=ozFEF88RjuLidxGYoy-t8nSAdd6qrZ4hRdhKf93dYZg,2939
-fivetran_connector_sdk-0.12.12.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-fivetran_connector_sdk-0.12.12.1.dist-info/entry_points.txt,sha256=uQn0KPnFlQmXJfxlk0tifdNsSXWfVlnAFzNqjXZM_xM,57
-fivetran_connector_sdk-0.12.12.1.dist-info/top_level.txt,sha256=-_xk2MFY4psIh7jw1lJePMzFb5-vask8_ZtX-UzYWUI,23
-fivetran_connector_sdk-0.12.12.1.dist-info/RECORD,,
+fivetran_connector_sdk-0.12.17.1.dist-info/METADATA,sha256=_ORD9BvuGvj63YvicMhGKKi9N7m7QuXyN1v0dQ3GDIw,2939
+fivetran_connector_sdk-0.12.17.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+fivetran_connector_sdk-0.12.17.1.dist-info/entry_points.txt,sha256=uQn0KPnFlQmXJfxlk0tifdNsSXWfVlnAFzNqjXZM_xM,57
+fivetran_connector_sdk-0.12.17.1.dist-info/top_level.txt,sha256=-_xk2MFY4psIh7jw1lJePMzFb5-vask8_ZtX-UzYWUI,23
+fivetran_connector_sdk-0.12.17.1.dist-info/RECORD,,