from datetime import datetime

import factory
from factory import fuzzy

from apollo_main_db.spotify.models import SpotifyPlaylistTrackListHistory2, SpotifyTrack2, \
    SpotifyTrackPlaylistLastWeekHistory


class SpotifyTrack2Factory(factory.alchemy.SQLAlchemyModelFactory):
    id = fuzzy.FuzzyText(length=25)
    name = fuzzy.FuzzyText(length=25)
    isrc = fuzzy.FuzzyText(length=100)

    class Meta:
        model = SpotifyTrack2
        sqlalchemy_session_persistence = 'commit'


class SpotifyPlaylistTrackListHistory2Factory(factory.alchemy.SQLAlchemyModelFactory):
    playlist_id = fuzzy.FuzzyText(length=50)
    playlist_index = factory.Faker("pyint")
    track_id = fuzzy.FuzzyText(length=50)
    date = factory.Faker("date")

    class Meta:
        model = SpotifyPlaylistTrackListHistory2
        sqlalchemy_session_persistence = "commit"


class SpotifyTrackPlaylistLastWeekHistoryFactory(factory.alchemy.SQLAlchemyModelFactory):
    date = factory.Faker("date")
    isrc = fuzzy.FuzzyText(length=12)
    playlist_id = fuzzy.FuzzyText(length=22)

    class Meta:
        model = SpotifyTrackPlaylistLastWeekHistory
        sqlalchemy_session_persistence = "commit"