"""Fake factories."""

from datetime import datetime, timedelta

import factory
from apollo_main_db.apollo import models as apollo_models
from apollo_main_db.spotify import models as spotify_models
from factory import fuzzy


class SpotifyViewPlaylistFactory(factory.alchemy.SQLAlchemyModelFactory):
    """Fake model for SpotifyViewPlaylist.

    """
    category_id = fuzzy.FuzzyInteger(1, 10)
    playlist_id = fuzzy.FuzzyText(length=22)
    type = fuzzy.FuzzyChoice(
        (
            spotify_models.ViewPlaylistTypeEnum.GTP,
            spotify_models.ViewPlaylistTypeEnum.HH,
            spotify_models.ViewPlaylistTypeEnum.NMF,
        )
    )

    class Meta:
        model = spotify_models.SpotifyViewPlaylist
        sqlalchemy_session_persistence = "commit"


class ApolloGlobalTrackPrioritiesHistoryFactory(factory.alchemy.SQLAlchemyModelFactory):
    """Fake model for ApolloGlobalTrackPrioritiesHistory.

    """
    id = fuzzy.FuzzyInteger(1, 1000)
    date = fuzzy.FuzzyDate((datetime.utcnow() - timedelta(days=10)).date(), datetime.utcnow().date())
    type_id = fuzzy.FuzzyInteger(1, 10)
    playlist_id = fuzzy.FuzzyText(length=22)

    class Meta:
        model = apollo_models.ApolloGlobalTrackPrioritiesHistory
        sqlalchemy_session_persistence = "commit"


class ApolloGlobalTrackPrioritiesHistoryTrackFactory(factory.alchemy.SQLAlchemyModelFactory):
    """Fake model for ApolloGlobalTrackPrioritiesHistoryTrack.

    """
    id = fuzzy.FuzzyInteger(100, 9999)
    history_id = fuzzy.FuzzyInteger(1, 1000)
    track_id = fuzzy.FuzzyText(length=22)
    isrc = fuzzy.FuzzyText(length=15)

    class Meta:
        model = apollo_models.ApolloGlobalTrackPrioritiesHistoryTrack
        sqlalchemy_session_persistence = "commit"