feat: add libs library

README.md

@@ -64,18 +64,19 @@ odd_dbt_test test --profile=my_profile
 
 ### Run commands programmatically
 You could run that scrip to read, parse and ingest test results to the platform.
+
 ```python
 # ingest_test_result.py
 
 from odd_dbt import config
 from odd_dbt.domain.cli_args import CliArgs
-from odd_dbt.lib.dbt import get_context
-from odd_dbt.lib.odd import create_dbt_generator_from_oddrn
+from odd_dbt.libs.dbt import get_context
+from odd_dbt.libs.odd import create_dbt_generator_from_oddrn
 from odd_dbt.service.odd import ingest_entities
 from odd_dbt.mapper.test_results import DbtTestMapper
 from odd_dbt.mapper.lineage import DbtLineageMapper
 
-cfg = config.Config() # All fields can be set manually or read from ENV variables
+cfg = config.Config()  # All fields can be set manually or read from ENV variables
 client = config.create_odd_client(host=cfg.odd_platform_host, token=cfg.odd_platform_token)
 generator = create_dbt_generator_from_oddrn(oddrn=cfg.dbt_data_source_oddrn)
 

odd_dbt/app.py

@@ -10,7 +10,7 @@
 from odd_dbt import get_version
 from odd_dbt.logger import logger
 from odd_dbt.mapper.test_results import DbtTestMapper
-from odd_dbt.lib import odd, dbt
+from odd_dbt.libs import odd, dbt
 from odd_dbt.service import odd as odd_api
 from odd_dbt.service.dbt import run_tests, CliArgs
 

odd_dbt/libs/dbt.py

@@ -0,0 +1,28 @@
+import dbt.events.functions as events_functions
+from dbt import flags
+from dbt.contracts.graph.nodes import ParsedNode, ModelNode
+
+from odd_dbt.domain.cli_args import CliArgs, FlagsArgs
+from odd_dbt.domain.context import DbtContext
+
+
+def collect_flags(cli_args: CliArgs):
+    flag_args = FlagsArgs(
+        project_dir=cli_args.project_dir,
+        profiles_dir=cli_args.profiles_dir,
+        target=cli_args.target,
+        profile=cli_args.profile,
+    )
+    flags.set_from_args(flag_args, None)
+    events_functions.set_invocation_id()
+
+    return flags.get_flags()
+
+
+def get_context(cli_args: CliArgs) -> DbtContext:
+    collect_flags(cli_args)
+    return DbtContext(cli_args=cli_args)
+
+
+def is_a_model_node(node: ParsedNode) -> bool:
+    return isinstance(node, ModelNode)

odd_dbt/libs/odd.py

@@ -0,0 +1,18 @@
+from odd_models import DataEntity
+from oddrn_generator import DbtGenerator
+
+
+def create_dbt_generator_from_oddrn(oddrn: str) -> DbtGenerator:
+    return DbtGenerator(host_settings=extract_host_from_oddrn(oddrn))
+
+
+def create_dbt_generator(host: str) -> DbtGenerator:
+    return DbtGenerator(host_settings=host)
+
+
+def extract_host_from_oddrn(oddrn: str) -> str:
+    return oddrn.split("//dbt/host/")[-1]
+
+
+def is_data_transformer(data_entity: DataEntity) -> bool:
+    return data_entity.data_transformer is not None

odd_dbt/mapper/lineage.py

@@ -13,7 +13,7 @@
 
 from odd_dbt import logger
 from odd_dbt.domain.context import DbtContext
-from odd_dbt.lib.dbt import is_a_model_node
+from odd_dbt.libs.dbt import is_a_model_node
 from odd_dbt.mapper.generator import create_generator
 from odd_dbt.mapper.metadata import get_model_metadata
 

odd_dbt/service/odd.py

@@ -2,7 +2,7 @@
 from odd_models import DataEntityList
 from odd_models.api_client.v2.odd_api_client import Client
 
-from odd_dbt.lib.odd import create_dbt_generator
+from odd_dbt.libs.odd import create_dbt_generator
 from odd_dbt.logger import logger