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,18 +1,14 @@
import os
import assets
import sensors
from config import APP
from dagster_polars import PolarsParquetIOManager
from icecream import install
from resources import JsonIOManager
from shared.config import APP, STORAGE_DIR
import dagster as dg
install()
storage_dir = os.environ.get("STORAGE_DIR", "/storage")
definitions = dg.Definitions(
assets=[
asset.with_attributes(
@@ -22,8 +18,8 @@ definitions = dg.Definitions(
for asset in dg.load_assets_from_modules([assets])
],
resources={
"json_io_manager": JsonIOManager(base_dir=storage_dir),
"polars_parquet_io_manager": PolarsParquetIOManager(base_dir=storage_dir),
"json_io_manager": JsonIOManager(base_dir=STORAGE_DIR),
"polars_parquet_io_manager": PolarsParquetIOManager(base_dir=STORAGE_DIR),
},
sensors=[sensors.list_locations, sensors.list_latitudes, sensors.retrieve_weather],
schedules=[