27 lines
735 B
Python
27 lines
735 B
Python
import assets
|
|
import schedules
|
|
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
|
|
from dagster import load_assets_from_modules
|
|
|
|
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 load_assets_from_modules([assets])
|
|
],
|
|
resources={
|
|
"json_io_manager": JsonIOManager(base_dir=STORAGE_DIR),
|
|
"polars_parquet_io_manager": PolarsParquetIOManager(base_dir=STORAGE_DIR),
|
|
},
|
|
schedules=[schedules.raw_schedule],
|
|
)
|