From 23fc62e87e17e904d0e9f4e0b357c53511756426 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 15 Apr 2024 21:41:50 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- dbt_sugar/core/clients/dbt.py | 1 + dbt_sugar/core/connectors/base.py | 1 + dbt_sugar/core/connectors/clickhouse_connector.py | 1 + dbt_sugar/core/connectors/postgres_connector.py | 1 + dbt_sugar/core/connectors/redshift_connector.py | 1 + dbt_sugar/core/connectors/snowflake_connector.py | 1 + dbt_sugar/core/flags.py | 1 + dbt_sugar/core/main.py | 1 + dbt_sugar/core/task/audit.py | 1 + dbt_sugar/core/task/base.py | 1 + dbt_sugar/core/task/bootstrap.py | 1 - dbt_sugar/core/task/doc.py | 7 ++++--- dbt_sugar/core/ui/traceback_manager.py | 1 + 13 files changed, 15 insertions(+), 4 deletions(-) diff --git a/dbt_sugar/core/clients/dbt.py b/dbt_sugar/core/clients/dbt.py index 415c85da..30f2349f 100644 --- a/dbt_sugar/core/clients/dbt.py +++ b/dbt_sugar/core/clients/dbt.py @@ -1,4 +1,5 @@ """Holds methods to interact with dbt API (we mostly don't for now because not stable) and objects.""" + import os from pathlib import Path from typing import Any, Dict, Optional, Union diff --git a/dbt_sugar/core/connectors/base.py b/dbt_sugar/core/connectors/base.py index 3f2354b5..dc13d34b 100644 --- a/dbt_sugar/core/connectors/base.py +++ b/dbt_sugar/core/connectors/base.py @@ -3,6 +3,7 @@ Only use this class implemented by a child connector. """ + from abc import ABC from typing import Dict, Sequence diff --git a/dbt_sugar/core/connectors/clickhouse_connector.py b/dbt_sugar/core/connectors/clickhouse_connector.py index 2bbb9434..92d428c0 100644 --- a/dbt_sugar/core/connectors/clickhouse_connector.py +++ b/dbt_sugar/core/connectors/clickhouse_connector.py @@ -3,6 +3,7 @@ Module dependent of the base connector. """ + from typing import Dict import sqlalchemy diff --git a/dbt_sugar/core/connectors/postgres_connector.py b/dbt_sugar/core/connectors/postgres_connector.py index 5c2655dd..f63486f9 100644 --- a/dbt_sugar/core/connectors/postgres_connector.py +++ b/dbt_sugar/core/connectors/postgres_connector.py @@ -3,6 +3,7 @@ Module dependent of the base connector. """ + from typing import Dict import sqlalchemy diff --git a/dbt_sugar/core/connectors/redshift_connector.py b/dbt_sugar/core/connectors/redshift_connector.py index b412e0bd..d15b01e8 100644 --- a/dbt_sugar/core/connectors/redshift_connector.py +++ b/dbt_sugar/core/connectors/redshift_connector.py @@ -3,6 +3,7 @@ Extends BaseConnector. """ + from typing import Dict import sqlalchemy diff --git a/dbt_sugar/core/connectors/snowflake_connector.py b/dbt_sugar/core/connectors/snowflake_connector.py index a1f19001..a676c5ac 100644 --- a/dbt_sugar/core/connectors/snowflake_connector.py +++ b/dbt_sugar/core/connectors/snowflake_connector.py @@ -3,6 +3,7 @@ Module dependent of the base connector. """ + from typing import Dict, Sequence import sqlalchemy diff --git a/dbt_sugar/core/flags.py b/dbt_sugar/core/flags.py index 667949a7..7aff7f39 100644 --- a/dbt_sugar/core/flags.py +++ b/dbt_sugar/core/flags.py @@ -1,4 +1,5 @@ """Flags module containing the FlagParser "Factory".""" + import sys from argparse import ArgumentParser from pathlib import Path diff --git a/dbt_sugar/core/main.py b/dbt_sugar/core/main.py index a86f27ec..25bebf43 100644 --- a/dbt_sugar/core/main.py +++ b/dbt_sugar/core/main.py @@ -1,4 +1,5 @@ """Main module for dbt-sugar. Sets up CLI arguments and sets up task handlers.""" + import argparse import sys from typing import List diff --git a/dbt_sugar/core/task/audit.py b/dbt_sugar/core/task/audit.py index 4bc558b2..0d4e457c 100644 --- a/dbt_sugar/core/task/audit.py +++ b/dbt_sugar/core/task/audit.py @@ -1,4 +1,5 @@ """Audit Task module.""" + from pathlib import Path from typing import Any, Dict, List diff --git a/dbt_sugar/core/task/base.py b/dbt_sugar/core/task/base.py index d3ae7df0..0d7999c0 100644 --- a/dbt_sugar/core/task/base.py +++ b/dbt_sugar/core/task/base.py @@ -1,4 +1,5 @@ """API definition for Task-like objects.""" + import abc import os import re diff --git a/dbt_sugar/core/task/bootstrap.py b/dbt_sugar/core/task/bootstrap.py index 0d9e83c3..29e7b91e 100644 --- a/dbt_sugar/core/task/bootstrap.py +++ b/dbt_sugar/core/task/bootstrap.py @@ -1,6 +1,5 @@ """Bootstrap module. Generates placeholders for all models in a dbt project.""" - import functools import operator import os diff --git a/dbt_sugar/core/task/doc.py b/dbt_sugar/core/task/doc.py index c1d5c0f0..5807b4fb 100644 --- a/dbt_sugar/core/task/doc.py +++ b/dbt_sugar/core/task/doc.py @@ -1,4 +1,5 @@ """Document Task module.""" + import copy import re import subprocess @@ -219,9 +220,9 @@ def add_primary_key_tests( self.column_update_payload[primary_key_column] = {"tests": PRIMARY_KEYS_TESTS} else: tests = self.column_update_payload[primary_key_column].get("tests", []) - self.column_update_payload[primary_key_column][ - "tests" - ] = self.combine_two_list_without_duplicates(PRIMARY_KEYS_TESTS, tests) + self.column_update_payload[primary_key_column]["tests"] = ( + self.combine_two_list_without_duplicates(PRIMARY_KEYS_TESTS, tests) + ) def orchestrate_model_documentation( self, schema: str, model_name: str, columns_sql: List[str] diff --git a/dbt_sugar/core/ui/traceback_manager.py b/dbt_sugar/core/ui/traceback_manager.py index 6b63bfc8..522bd7dc 100644 --- a/dbt_sugar/core/ui/traceback_manager.py +++ b/dbt_sugar/core/ui/traceback_manager.py @@ -1,4 +1,5 @@ """Holds handlers to manage error tracebacks using.""" + import pretty_errors from dbt_sugar.core.flags import FlagParser