Ruff check --fix

Author: matyas-jirat-keboola

src/keboola/component/base.py

@@ -8,7 +8,7 @@
 from abc import ABC
 from dataclasses import dataclass
 from enum import Enum
-from typing import Union, List, Optional
+from typing import List, Optional, Union
 
 
 @dataclass
@@ -23,15 +23,14 @@ def __post_init__(self):
         In other cases exception is thrown and printed via stderr.
 
         """
-        self.status = 'success'
+        self.status = "success"
 
     def __str__(self):
         # the None values / attributes will be ignored.
-        dict_obj = dataclasses.asdict(self, dict_factory=lambda x: {k: v for (k, v) in x if
-                                                                    v is not None})
+        dict_obj = dataclasses.asdict(self, dict_factory=lambda x: {k: v for (k, v) in x if v is not None})
         # hack to add default status
         if self.status:
-            dict_obj['status'] = self.status
+            dict_obj["status"] = self.status
         return json.dumps(dict_obj)
 
 
@@ -56,6 +55,7 @@ class SelectElement(SyncActionResult):
     """
     For select elements. Label is optional and value will be used
     """
+
     value: str
     label: Optional[str] = None
 
@@ -77,13 +77,14 @@ def process_sync_action_result(result: Union[None, List[dict], dict, SyncActionR
     if isinstance(result, SyncActionResult):
         result_str = str(result)
     elif isinstance(result, list):
-        result_str = f'[{", ".join([json.dumps(r) if isinstance(r, dict) else str(r) for r in result])}]'
+        result_str = f"[{', '.join([json.dumps(r) if isinstance(r, dict) else str(r) for r in result])}]"
     elif result is None:
-        result_str = json.dumps({'status': 'success'})
+        result_str = json.dumps({"status": "success"})
     elif isinstance(result, dict):
         # for backward compatibility
         result_str = json.dumps(result)
     else:
-        raise ValueError("Result of sync action must be either None or an instance of SyncActionResult "
-                         "or a List[SyncActionResult]")
+        raise ValueError(
+            "Result of sync action must be either None or an instance of SyncActionResult or a List[SyncActionResult]"
+        )
     return result_str

src/keboola/component/dao.py

@@ -1,14 +1,15 @@
-from typing import List, Dict
-from typing import Optional, Union
-from keboola.component.dao import SupportedDataTypes
 from dataclasses import dataclass
+from typing import Dict, List, Optional, Union
+
+from keboola.component.dao import SupportedDataTypes
 
 
 @dataclass
 class FieldSchema:
     """
     Defines the name and type specifications of a single field in a table
     """
+
     name: str
     base_type: Optional[Union[SupportedDataTypes, str]] = None
     description: Optional[str] = None
@@ -22,6 +23,7 @@ class TableSchema:
     """
     TableSchema class is used to define the schema and metadata of a table.
     """
+
     name: str
     fields: List[FieldSchema]
     primary_keys: Optional[List[str]] = None
@@ -79,10 +81,12 @@ def init_table_schema_from_dict(json_table_schema: Dict) -> TableSchema:
         json_table_schema["fields"] = [FieldSchema(**field) for field in json_table_schema["fields"]]
     except TypeError as type_error:
         raise KeyError(
-            f"When creating the table schema the definition of columns failed : {type_error}") from type_error
+            f"When creating the table schema the definition of columns failed : {type_error}"
+        ) from type_error
     try:
         ts = TableSchema(**json_table_schema)
     except TypeError as type_error:
         raise KeyError(
-            f"When creating the table schema the definition of the table failed : {type_error}") from type_error
+            f"When creating the table schema the definition of the table failed : {type_error}"
+        ) from type_error
     return ts

src/keboola/component/interface.py

@@ -2,4 +2,4 @@
 import sys
 
 # just in case include in path
-sys.path.insert(0, str(pathlib.Path(__file__).resolve().parents[1].joinpath('src')))
+sys.path.insert(0, str(pathlib.Path(__file__).resolve().parents[1].joinpath("src")))

src/keboola/component/sync_actions.py

@@ -10,7 +10,7 @@
 
 class MockComponent(ComponentBase):
     def run(self):
-        return 'run_executed'
+        return "run_executed"
 
 
 class MockComponentFail(ComponentBase):
@@ -19,44 +19,40 @@ def run(self):
 
 
 class TestCommonInterface(unittest.TestCase):
-
     def setUp(self):
-        path = os.path.join(os.path.dirname(os.path.realpath(__file__)),
-                            'data_examples', 'data1')
+        path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "data_examples", "data1")
         os.environ["KBC_DATADIR"] = path
 
     def test_default_arguments_pass(self):
         MockComponent()
 
     def test_missing_config_parameters_fail(self):
         with self.assertRaises(UserException):
-            MockComponent(required_parameters=['missing'])
+            MockComponent(required_parameters=["missing"])
 
     def test_missing_image_parameters_fail(self):
         with self.assertRaises(UserException):
-            c = MockComponent(required_image_parameters=['missing'])
+            c = MockComponent(required_image_parameters=["missing"])
             c.execute_action()
 
     def test_missing_action_fail(self):
-        path = os.path.join(os.path.dirname(os.path.realpath(__file__)),
-                            'data_examples', 'data_custom_action')
+        path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "data_examples", "data_custom_action")
         os.environ["KBC_DATADIR"] = path
         with self.assertRaises(AttributeError):
             MockComponent().execute_action()
 
     def test_run_action_passes(self):
-        self.assertEqual(MockComponent().execute_action(), 'run_executed')
+        self.assertEqual(MockComponent().execute_action(), "run_executed")
 
     def test_custom_action_passes(self):
-        path = os.path.join(os.path.dirname(os.path.realpath(__file__)),
-                            'data_examples', 'data_custom_action')
+        path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "data_examples", "data_custom_action")
         os.environ["KBC_DATADIR"] = path
 
         class CustomActionComponent(ComponentBase):
             def run(self):
                 pass
 
-            @sync_action('custom_action')
+            @sync_action("custom_action")
             def test_action(self):
                 return [SelectElement("test")]
 
@@ -68,40 +64,37 @@ def test_run_action_fails_with_user_error(self):
 
     def test_system_action_name_fail(self):
         with self.assertRaises(ValueError):
+
             class ComponentInvalidActionName(ComponentBase):
                 def run(self):
                     pass
 
-                @sync_action('run')
+                @sync_action("run")
                 def test_action(self):
                     pass
 
-            path = os.path.join(os.path.dirname(os.path.realpath(__file__)),
-                                'data_examples', 'data_custom_action')
+            path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "data_examples", "data_custom_action")
             os.environ["KBC_DATADIR"] = path
 
             ComponentInvalidActionName().execute_action()
 
-    @patch('sys.stdout', new_callable=StringIO)
+    @patch("sys.stdout", new_callable=StringIO)
     def test_sync_action_prints_valid_message(self, stdout):
-        path = os.path.join(os.path.dirname(os.path.realpath(__file__)),
-                            'data_examples', 'data_custom_action')
+        path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "data_examples", "data_custom_action")
         os.environ["KBC_DATADIR"] = path
 
         class CustomActionComponent(ComponentBase):
             def run(self):
                 pass
 
-            @sync_action('custom_action')
+            @sync_action("custom_action")
             def get_columns(self):
-                return [SelectElement("value_a", "label_a"),
-                        SelectElement("value_b")
-                        ]
+                return [SelectElement("value_a", "label_a"), SelectElement("value_b")]
 
         CustomActionComponent().execute_action()
         expected = '[{"value": "value_a", "label": "label_a"}, {"value": "value_b", "label": "value_b"}]'
         self.assertEqual(stdout.getvalue(), expected)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()