diff --git a/.github/workflows/python-release.yml b/.github/workflows/python-release.yml index 2a3954a48..42090f424 100644 --- a/.github/workflows/python-release.yml +++ b/.github/workflows/python-release.yml @@ -152,6 +152,7 @@ jobs: - name: Create Pull Request uses: peter-evans/create-pull-request@v4 + if: inputs.package == dbt-fal with: branch: bump-${{ inputs.package }}-${{ env.publishing_version }} delete-branch: true diff --git a/projects/adapter/pyproject.toml b/projects/adapter/pyproject.toml index 92d5ebb32..df5e956d9 100644 --- a/projects/adapter/pyproject.toml +++ b/projects/adapter/pyproject.toml @@ -1,6 +1,8 @@ [tool.poetry] name = "dbt-fal" version = "1.5.6a0" +# name = "fal" +# version = "0.9.4a0" description = "Run python scripts from any dbt project." readme = "README.md" homepage = "https://github.com/fal-ai/fal" @@ -95,5 +97,5 @@ build-backend = "poetry.core.masonry.api" [tool.poetry.scripts] dbt-fal = "fal.dbt.cli:cli" -# NOTE: deprecated use +# TODO: remove once `fal` is no longer a supported package fal = "fal.dbt.cli:fal_cli" diff --git a/projects/adapter/src/fal/__init__.py b/projects/adapter/src/fal/__init__.py index 26c1b1dd6..a10b9c5dd 100644 --- a/projects/adapter/src/fal/__init__.py +++ b/projects/adapter/src/fal/__init__.py @@ -1,3 +1,5 @@ +# TODO: remove once `fal` is no longer a supported package +# The whole file DBT_FAL_IMPORT_NOTICE = \ """The dbt tool `fal` and `dbt-fal` adapter have been merged into a single tool. Please import from the `fal.dbt` module instead. diff --git a/projects/adapter/src/fal/dbt/cli/args.py b/projects/adapter/src/fal/dbt/cli/args.py index ad6206d33..006e5b47f 100644 --- a/projects/adapter/src/fal/dbt/cli/args.py +++ b/projects/adapter/src/fal/dbt/cli/args.py @@ -208,7 +208,12 @@ def _build_cli_parser(): description="Run Python scripts on dbt models", ) - version = pkg_resources.get_distribution("dbt-fal").version + try: + version = pkg_resources.get_distribution("dbt-fal").version + except pkg_resources.DistributionNotFound: + # TODO: remove once `fal` is no longer a supported package + version = pkg_resources.get_distribution("fal").version + parser.add_argument( "-v", "--version", diff --git a/projects/adapter/src/fal/dbt/cli/cli.py b/projects/adapter/src/fal/dbt/cli/cli.py index 68d488ad3..5dee8a99a 100644 --- a/projects/adapter/src/fal/dbt/cli/cli.py +++ b/projects/adapter/src/fal/dbt/cli/cli.py @@ -11,6 +11,7 @@ from fal.dbt.integration.logger import log_manager +# TODO: remove once `fal` is no longer a supported package DBT_FAL_COMMAND_NOTICE = \ """The dbt tool `fal` and `dbt-fal` adapter have been merged into a single tool. Please use the new `dbt-fal` command line tool instead. @@ -20,6 +21,7 @@ dbt-fal """ +# TODO: remove once `fal` is no longer a supported package def fal_cli(argv: List[str] = sys.argv): print(DBT_FAL_COMMAND_NOTICE) cli(argv) diff --git a/projects/adapter/src/fal/dbt/packages/dependency_analysis.py b/projects/adapter/src/fal/dbt/packages/dependency_analysis.py index 42c7071da..76eb9e3b3 100644 --- a/projects/adapter/src/fal/dbt/packages/dependency_analysis.py +++ b/projects/adapter/src/fal/dbt/packages/dependency_analysis.py @@ -101,12 +101,25 @@ def _get_dbt_fal_package() -> Tuple[str, Optional[str]]: else: # We are going to install it from PyPI. dbt_fal_dep = "dbt-fal" - dbt_fal_version = importlib_metadata.version("dbt-fal") + try: + dbt_fal_version = importlib_metadata.version("dbt-fal") + except importlib_metadata.PackageNotFoundError: + # TODO: remove once `fal` is no longer a supported package + dbt_fal_version = importlib_metadata.version("fal") else: dbt_fal_dep = "dbt-fal" - dbt_fal_version = importlib_metadata.version("dbt-fal") + try: + dbt_fal_version = importlib_metadata.version("dbt-fal") + except importlib_metadata.PackageNotFoundError: + # TODO: remove once `fal` is no longer a supported package + dbt_fal_version = importlib_metadata.version("fal") + + try: + dbt_fal_extras = _find_fal_extras("dbt-fal") + except importlib_metadata.PackageNotFoundError: + # TODO: remove once `fal` is no longer a supported package + dbt_fal_extras = _find_fal_extras("fal") - dbt_fal_extras = _find_fal_extras("dbt-fal") if dbt_fal_extras: dbt_fal_dep += f"[{','.join(dbt_fal_extras)}]"