32 lines
837 B
Python
32 lines
837 B
Python
import assets
|
|
import sensors
|
|
from dagster_polars import PolarsParquetIOManager
|
|
from icecream import install
|
|
from shared.config import APP, STORAGE_DIR
|
|
from shared.io_manager import JsonIOManager
|
|
|
|
import dagster as dg
|
|
|
|
install()
|
|
|
|
definitions = dg.Definitions(
|
|
assets=[
|
|
asset.with_attributes(
|
|
group_names_by_key={asset.key: APP},
|
|
tags_by_key={asset.key: {"app": APP}},
|
|
)
|
|
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),
|
|
},
|
|
sensors=[
|
|
sensors.list_locations,
|
|
sensors.list_latitudes,
|
|
sensors.list_longitudes,
|
|
sensors.retrieve_weather,
|
|
sensors.retrieve_weather,
|
|
],
|
|
)
|