2024-12-19 11:02:37 -05:00
|
|
|
import pathlib
|
|
|
|
|
import shutil
|
|
|
|
|
|
|
|
|
|
PYTHON_BUILD_DIR = pathlib.Path(__file__).parent
|
|
|
|
|
ROOT_DIR = PYTHON_BUILD_DIR.parent
|
|
|
|
|
WHEEL_OUTPUT_DIR = PYTHON_BUILD_DIR / "wheels"
|
|
|
|
|
|
|
|
|
|
# These should line up with the build.gradle file.
|
|
|
|
|
wheel_dirs = [
|
|
|
|
|
ROOT_DIR / "metadata-ingestion/dist",
|
2025-04-21 17:19:25 -04:00
|
|
|
ROOT_DIR / "datahub-actions/dist",
|
2024-12-19 11:02:37 -05:00
|
|
|
ROOT_DIR / "metadata-ingestion-modules/airflow-plugin/dist",
|
|
|
|
|
ROOT_DIR / "metadata-ingestion-modules/dagster-plugin/dist",
|
|
|
|
|
ROOT_DIR / "metadata-ingestion-modules/prefect-plugin/dist",
|
|
|
|
|
ROOT_DIR / "metadata-ingestion-modules/gx-plugin/dist",
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
# Delete and recreate the output directory.
|
|
|
|
|
if WHEEL_OUTPUT_DIR.exists():
|
|
|
|
|
shutil.rmtree(WHEEL_OUTPUT_DIR)
|
|
|
|
|
WHEEL_OUTPUT_DIR.mkdir(parents=True)
|
|
|
|
|
|
|
|
|
|
# Copy things over.
|
|
|
|
|
for wheel_dir in wheel_dirs:
|
|
|
|
|
for wheel_file in wheel_dir.glob("*"):
|
|
|
|
|
shutil.copy(wheel_file, WHEEL_OUTPUT_DIR)
|
|
|
|
|
|
|
|
|
|
print("Copied wheels to", WHEEL_OUTPUT_DIR)
|
