mirror of
https://github.com/Dvorinka/swingmusic-extended.git
synced 2026-06-04 12:33:03 +00:00
cwilvx
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
from app.db.userdata import FavoritesTable, ScrobbleTable
|
|
from app.store.albums import AlbumStore
|
|
from app.store.artists import ArtistStore
|
|
from app.store.tracks import TrackStore
|
|
|
|
|
|
from typing import Any
|
|
|
|
|
|
def map_scrobble_data():
|
|
"""
|
|
Maps scrobble data to the in-memory stores.
|
|
|
|
The scrobble data is loaded from the database and grouped by trackhash.
|
|
The album and artist scrobble data (for those tracks) are then incremented based on the data.
|
|
"""
|
|
records = ScrobbleTable.get_all(0, None)
|
|
|
|
# group records by trackhash
|
|
grouped: dict[str, dict[str, Any]] = {}
|
|
|
|
for record in records:
|
|
# aggregate playcount, playduration and lastplayed
|
|
item = grouped.setdefault(record.trackhash, {})
|
|
item["playcount"] = item.get("playcount", 0) + 1
|
|
item["playduration"] = item.get("playduration", 0) + record.duration
|
|
item["lastplayed"] = max(item.get("lastplayed", 0), record.timestamp)
|
|
|
|
# increment playcount, playduration and lastplayed for albums and artists
|
|
for trackhash, data in grouped.items():
|
|
track = TrackStore.trackhashmap.get(trackhash)
|
|
|
|
if track is None:
|
|
continue
|
|
|
|
track.increment_playcount(data["playduration"], data["lastplayed"])
|
|
|
|
album = AlbumStore.albummap.get(track.tracks[0].albumhash)
|
|
if album:
|
|
album.increment_playcount(data["playduration"], data["lastplayed"])
|
|
|
|
for artisthash in track.tracks[0].artisthashes:
|
|
artist = ArtistStore.artistmap.get(artisthash)
|
|
if artist:
|
|
artist.increment_playcount(data["playduration"], data["lastplayed"])
|
|
|
|
|
|
def map_favorites():
|
|
"""
|
|
Maps favorites data to the in-memory stores.
|
|
"""
|
|
favorites = FavoritesTable.get_all()
|
|
|
|
for entry in favorites:
|
|
if entry.type == "album":
|
|
album = AlbumStore.albummap.get(entry.hash)
|
|
if album:
|
|
album.toggle_favorite_user(entry.userid)
|
|
|
|
elif entry.type == "artist":
|
|
artist = ArtistStore.artistmap.get(entry.hash)
|
|
if artist:
|
|
artist.toggle_favorite_user(entry.userid)
|
|
|
|
elif entry.type == "track":
|
|
track = TrackStore.trackhashmap.get(entry.hash)
|
|
if track:
|
|
track.toggle_favorite_user(entry.userid)
|