import assets
from dagster_polars import PolarsParquetIOManager
from icecream import install
from shared.config import APP, STORAGE_DIR

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={
        "polars_parquet_io_manager": PolarsParquetIOManager(base_dir=STORAGE_DIR)
    },
)