shared config

This commit is contained in:
2025-07-29 11:38:41 +02:00
parent cc6beeffed
commit f59b30b9ea
6 changed files with 17 additions and 44 deletions

View File

@@ -1,16 +1,13 @@
import os
import assets
from dagster_polars import PolarsParquetIOManager
from icecream import install
from shared.config import APP, STORAGE_DIR
import dagster as dg
from dagster import load_assets_from_modules
install()
APP = os.environ["APP"]
definitions = dg.Definitions(
assets=[
asset.with_attributes(
@@ -20,8 +17,6 @@ definitions = dg.Definitions(
for asset in load_assets_from_modules([assets])
],
resources={
"polars_parquet_io_manager": PolarsParquetIOManager(
base_dir=os.environ.get("STORAGE_DIR", "/storage") + f"/{APP}"
),
"polars_parquet_io_manager": PolarsParquetIOManager(base_dir=STORAGE_DIR),
},
)