95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
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)
|
|
|
|
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)
|
|
|
|
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)
|
|
|
|
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()
|
|
})
|