import logging from datetime import datetime import music.db.database as database from music.model.user import User from music.model.playlist import Playlist from spotfm.maths.counter import Counter from spotframework.net.network import SpotifyNetworkException from fmframework.net.network import LastFMNetworkException logger = logging.getLogger(__name__) def refresh_lastfm_track_stats(username, playlist_name): logger.info(f'refreshing {playlist_name} stats for {username}') user = User.collection.filter('username', '==', username.strip().lower()).get() if user is None: logger.error(f'user {username} not found') fmnet = database.get_authed_lastfm_network(user) spotnet = database.get_authed_spotify_network(user) counter = Counter(fmnet=fmnet, spotnet=spotnet) playlist = user.get_playlist(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 try: spotify_playlist = spotnet.playlist(uri=playlist.uri) except SpotifyNetworkException: logger.exception(f'error retrieving spotify playlist {username} / {playlist_name}') return track_count = counter.count_playlist(playlist=spotify_playlist) try: user_count = fmnet.user_scrobble_count() if user_count > 0: percent = round((track_count * 100) / user_count, 2) else: percent = 0 except LastFMNetworkException: logger.exception(f'error while retrieving user scrobble count {username} / {playlist_name}') percent = 0 playlist.lastfm_stat_count = track_count playlist.lastfm_stat_percent = percent playlist.lastfm_stat_last_refresh = datetime.utcnow() playlist.update() def refresh_lastfm_album_stats(username, playlist_name): logger.info(f'refreshing {playlist_name} stats for {username}') user = User.collection.filter('username', '==', username.strip().lower()).get() if user is None: logger.error(f'user {username} not found') fmnet = database.get_authed_lastfm_network(user) spotnet = database.get_authed_spotify_network(user) counter = Counter(fmnet=fmnet, spotnet=spotnet) playlist = user.get_playlist(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 try: spotify_playlist = spotnet.playlist(uri=playlist.uri) except SpotifyNetworkException: logger.exception(f'error retrieving spotify playlist {username} / {playlist_name}') return album_count = counter.count_playlist(playlist=spotify_playlist, query_album=True) try: user_count = fmnet.user_scrobble_count() if user_count > 0: album_percent = round((album_count * 100) / user_count, 2) else: album_percent = 0 except LastFMNetworkException: logger.exception(f'error while retrieving user scrobble count {username} / {playlist_name}') album_percent = 0 playlist.lastfm_stat_album_count = album_count playlist.lastfm_stat_album_percent = album_percent playlist.lastfm_stat_last_refresh = datetime.utcnow() playlist.update() def refresh_lastfm_artist_stats(username, playlist_name): logger.info(f'refreshing {playlist_name} stats for {username}') user = User.collection.filter('username', '==', username.strip().lower()).get() if user is None: logger.error(f'user {username} not found') fmnet = database.get_authed_lastfm_network(user) spotnet = database.get_authed_spotify_network(user) counter = Counter(fmnet=fmnet, spotnet=spotnet) playlist = user.get_playlist(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 try: spotify_playlist = spotnet.playlist(uri=playlist.uri) except SpotifyNetworkException: logger.exception(f'error retrieving spotify playlist {username} / {playlist_name}') return artist_count = counter.count_playlist(playlist=spotify_playlist, query_artist=True) try: user_count = fmnet.user_scrobble_count() if user_count > 0: artist_percent = round((artist_count * 100) / user_count, 2) else: artist_percent = 0 except LastFMNetworkException: logger.exception(f'error while retrieving user scrobble count {username} / {playlist_name}') artist_percent = 0 playlist.lastfm_stat_artist_count = artist_count playlist.lastfm_stat_artist_percent = artist_percent playlist.lastfm_stat_last_refresh = datetime.utcnow() playlist.update()