Mixonomer/music/tasks/refresh_lastfm_stats.py

from google.cloud import firestore

import logging
from datetime import datetime

import music.db.database as database

from spotfm.maths.counter import Counter
from spotframework.model.uri import Uri

db = firestore.Client()

logger = logging.getLogger(__name__)


def refresh_lastfm_track_stats(username, playlist_name):

    logger.info(f'refreshing {playlist_name} stats for {username}')

    fmnet = database.get_authed_lastfm_network(username=username)
    spotnet = database.get_authed_spotify_network(username=username)
    counter = Counter(fmnet=fmnet, spotnet=spotnet)

    playlist = database.get_playlist(username=username, name=playlist_name)

    if playlist is None:
        logger.critical(f'playlist {playlist_name} for {username} not found')
        return

    if playlist.uri is None:
        logger.critical(f'playlist {playlist_name} for {username} has no spotify uri')
        return

    spotify_playlist = spotnet.get_playlist(uri=Uri(playlist.uri))
    track_count = counter.count_playlist(playlist=spotify_playlist)

    user_count = fmnet.get_user_scrobble_count()
    if user_count > 0:
        percent = round((track_count * 100) / user_count, 2)
    else:
        percent = 0

    playlist.update_database({
        'lastfm_stat_count': track_count,
        'lastfm_stat_percent': percent,

        'lastfm_stat_last_refresh': datetime.utcnow()
    })


def refresh_lastfm_album_stats(username, playlist_name):

    logger.info(f'refreshing {playlist_name} stats for {username}')

    fmnet = database.get_authed_lastfm_network(username=username)
    spotnet = database.get_authed_spotify_network(username=username)
    counter = Counter(fmnet=fmnet, spotnet=spotnet)

    playlist = database.get_playlist(username=username, name=playlist_name)

    if playlist is None:
        logger.critical(f'playlist {playlist_name} for {username} not found')
        return

    if playlist.uri is None:
        logger.critical(f'playlist {playlist_name} for {username} has no spotify uri')
        return

    spotify_playlist = spotnet.get_playlist(uri=Uri(playlist.uri))
    album_count = counter.count_playlist(playlist=spotify_playlist, query_album=True)

    user_count = fmnet.get_user_scrobble_count()
    if user_count > 0:
        album_percent = round((album_count * 100) / user_count, 2)
    else:
        album_percent = 0

    playlist.update_database({
        'lastfm_stat_album_count': album_count,
        'lastfm_stat_album_percent': album_percent,

        'lastfm_stat_last_refresh': datetime.utcnow()
    })


def refresh_lastfm_artist_stats(username, playlist_name):

    logger.info(f'refreshing {playlist_name} stats for {username}')

    fmnet = database.get_authed_lastfm_network(username=username)
    spotnet = database.get_authed_spotify_network(username=username)
    counter = Counter(fmnet=fmnet, spotnet=spotnet)

    playlist = database.get_playlist(username=username, name=playlist_name)

    if playlist is None:
        logger.critical(f'playlist {playlist_name} for {username} not found')
        return

    if playlist.uri is None:
        logger.critical(f'playlist {playlist_name} for {username} has no spotify uri')
        return

    spotify_playlist = spotnet.get_playlist(uri=Uri(playlist.uri))
    artist_count = counter.count_playlist(playlist=spotify_playlist, query_artist=True)

    user_count = fmnet.get_user_scrobble_count()
    if user_count > 0:
        artist_percent = round((artist_count * 100) / user_count, 2)
    else:
        artist_percent = 0

    playlist.update_database({
        'lastfm_stat_artist_count': artist_count,
        'lastfm_stat_artist_percent': artist_percent,

        'lastfm_stat_last_refresh': datetime.utcnow()
    })