format codebase

dlt/__init__.py

@@ -21,17 +21,20 @@
 For more detailed info, see https://dlthub.com/docs/walkthroughs
 """
 
-from dlt.version import __version__
+from dlt import sources
 from dlt.common.configuration.accessors import config, secrets
-from dlt.common.typing import TSecretValue as _TSecretValue
 from dlt.common.configuration.specs import CredentialsConfiguration as _CredentialsConfiguration
 from dlt.common.pipeline import source_state as state
 from dlt.common.schema import Schema
-
-from dlt import sources
-from dlt.extract.decorators import source, resource, transformer, defer
-from dlt.pipeline import pipeline as _pipeline, run, attach, Pipeline, dbt, current as _current, mark as _mark
-from dlt.pipeline import progress
+from dlt.common.typing import TSecretValue as _TSecretValue
+from dlt.extract.decorators import defer, resource, source, transformer
+from dlt.pipeline import Pipeline, attach
+from dlt.pipeline import current as _current
+from dlt.pipeline import dbt
+from dlt.pipeline import mark as _mark
+from dlt.pipeline import pipeline as _pipeline
+from dlt.pipeline import progress, run
+from dlt.version import __version__
 
 pipeline = _pipeline
 current = _current

dlt/cli/config_toml_writer.py

@@ -1,11 +1,16 @@
-from typing import Any, NamedTuple, Tuple, Iterable
+from collections.abc import Sequence as C_Sequence
+from typing import Any, Iterable, NamedTuple, Tuple
+
 import tomlkit
-from tomlkit.items import Table as TOMLTable
 from tomlkit.container import Container as TOMLContainer
-from collections.abc import Sequence as C_Sequence
+from tomlkit.items import Table as TOMLTable
 
 from dlt.common import pendulum
-from dlt.common.configuration.specs import BaseConfiguration, is_base_configuration_inner_hint, extract_inner_hint
+from dlt.common.configuration.specs import (
+    BaseConfiguration,
+    extract_inner_hint,
+    is_base_configuration_inner_hint,
+)
 from dlt.common.data_types import py_type_to_sc_type
 from dlt.common.typing import AnyType, is_final_type, is_optional_type
 
@@ -53,13 +58,15 @@ def write_value(
     hint: AnyType,
     overwrite_existing: bool,
     default_value: Any = None,
-    is_default_of_interest: bool = False
+    is_default_of_interest: bool = False,
 ) -> None:
     # skip if table contains the name already
     if name in toml_table and not overwrite_existing:
         return
     # do not dump final and optional fields if they are not of special interest
-    if (is_final_type(hint) or is_optional_type(hint) or default_value is not None) and not is_default_of_interest:
+    if (
+        is_final_type(hint) or is_optional_type(hint) or default_value is not None
+    ) and not is_default_of_interest:
         return
     # get the inner hint to generate cool examples
     hint = extract_inner_hint(hint)
@@ -84,10 +91,19 @@ def write_spec(toml_table: TOMLTable, config: BaseConfiguration, overwrite_exist
         default_value = getattr(config, name, None)
         # check if field is of particular interest and should be included if it has default
         is_default_of_interest = name in config.__config_gen_annotations__
-        write_value(toml_table, name, hint, overwrite_existing, default_value=default_value, is_default_of_interest=is_default_of_interest)
+        write_value(
+            toml_table,
+            name,
+            hint,
+            overwrite_existing,
+            default_value=default_value,
+            is_default_of_interest=is_default_of_interest,
+        )
 
 
-def write_values(toml: TOMLContainer, values: Iterable[WritableConfigValue], overwrite_existing: bool) -> None:
+def write_values(
+    toml: TOMLContainer, values: Iterable[WritableConfigValue], overwrite_existing: bool
+) -> None:
     for value in values:
         toml_table: TOMLTable = toml  # type: ignore
         for section in value.sections:
@@ -98,4 +114,11 @@ def write_values(toml: TOMLContainer, values: Iterable[WritableConfigValue], ove
             else:
                 toml_table = toml_table[section]  # type: ignore
 
-        write_value(toml_table, value.name, value.hint, overwrite_existing, default_value=value.default_value, is_default_of_interest=True)
+        write_value(
+            toml_table,
+            value.name,
+            value.hint,
+            overwrite_existing,
+            default_value=value.default_value,
+            is_default_of_interest=True,
+        )