improve dbt plugin code

memiiso · Oct 16, 2024 · f529a11 · f529a11
1 parent b1771ce
commit f529a11
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/opendbt/dbt/__init__.py b/opendbt/dbt/__init__.py
@@ -10,10 +10,9 @@ def patch_dbt():
     if dbt_version >= Version("1.7.0") and dbt_version < Version("1.8.0"):
         from opendbt.dbt.v17.task.docs.generate import OpenDbtGenerateTask
         from opendbt.dbt.v17.adapters.factory import OpenDbtAdapterContainer
-        import opendbt.dbt.v17.cli.main
         dbt.task.generate.GenerateTask = OpenDbtGenerateTask
         dbt.adapters.factory.FACTORY = OpenDbtAdapterContainer()
-    if dbt_version >= Version("1.8.0") and dbt_version < Version("1.9.0"):
+    elif dbt_version >= Version("1.8.0") and dbt_version < Version("1.9.0"):
         from opendbt.dbt.v18.task.docs.generate import OpenDbtGenerateTask
         from opendbt.dbt.v18.adapters.factory import OpenDbtAdapterContainer
         dbt.task.docs.generate.GenerateTask = OpenDbtGenerateTask
@@ -23,6 +22,7 @@ def patch_dbt():
             f"Unsupported dbt version {dbt_version}, please make sure dbt version is supported/integrated by opendbt")
 
     # common code
+    import opendbt.dbt.shared.cli.main
     dbt.cli.main.sqlfluff = opendbt.dbt.shared.cli.main.sqlfluff
     dbt.cli.main.sqlfluff_lint = opendbt.dbt.shared.cli.main.sqlfluff_lint
     dbt.cli.main.sqlfluff_fix = opendbt.dbt.shared.cli.main.sqlfluff_fix