from ....base import AggregatesTransformBase
from ....constants import ARTIST_ALBUMS_SPOTIFY_AGG_DATA_PATH, ARTIST_ALBUMS_SPOTIFY_RAW_DATA_PATH, RAW_ARTIST_ID_KEY
from .extract import Extract


__all__ = ["Transform"]


class Transform(AggregatesTransformBase):
    depends_on = {Extract}
    raw_id_key: str = RAW_ARTIST_ID_KEY
    raw_data_path: str = str(ARTIST_ALBUMS_SPOTIFY_RAW_DATA_PATH)
    agg_data_path: str = str(ARTIST_ALBUMS_SPOTIFY_AGG_DATA_PATH)