sync action for the run method

Author: soustruh

src/component.py

@@ -8,7 +8,7 @@
 from os import mkdir, path
 
 from keboola.component.base import ComponentBase, sync_action
-from keboola.component.dao import SupportedDataTypes, BaseType, ColumnDefinition
+from keboola.component.dao import BaseType, ColumnDefinition, SupportedDataTypes
 from keboola.component.exceptions import UserException
 from keboola.component.sync_actions import MessageType, SelectElement, ValidationResult
 from keboola.utils.header_normalizer import NormalizerStrategy, get_normalizer
@@ -86,7 +86,7 @@ class Component(ComponentBase):
     def __init__(self):
         super().__init__()
 
-    def run(self):
+    def run(self, return_data=False):
         self.validate_configuration_parameters(REQUIRED_PARAMETERS)
         self.validate_image_parameters(REQUIRED_IMAGE_PARS)
 
@@ -130,6 +130,9 @@ def run(self):
         logging.debug([result for result in results])
         logging.info(f"Downloaded {total_records} records in total")
 
+        if return_data:
+            return results
+
         # remove headers and get columns
         output_columns = self._fix_header_from_csv(results)
         output_columns = self.normalize_column_names(output_columns)
@@ -419,9 +422,9 @@ def _get_login_method(self) -> LoginType:
     @staticmethod
     def process_salesforce_domain(url):
         if url.startswith("http://"):
-            url = url[len("http://"):]
+            url = url[len("http://") :]
         if url.startswith("https://"):
-            url = url[len("https://"):]
+            url = url[len("https://") :]
         if url.endswith(".salesforce.com"):
             url = url[: -len(".salesforce.com")]
 
@@ -619,6 +622,13 @@ def load_possible_primary_keys(self) -> list[SelectElement]:
         else:
             raise UserException(f"Invalid {KEY_QUERY_TYPE}")
 
+    # @sync_action("runComponent")
+    # def sync_run_component(self):
+    #     """
+    #     Run the component as a sync action
+    #     """
+    #     return self.run(return_data=True)
+
     def _get_object_name_from_custom_query(self) -> str:
         params = self.configuration.parameters
         salesforce_client = self.get_salesforce_client(params)