1
0
mirror of https://github.com/krateng/maloja.git synced 2023-08-10 21:12:55 +03:00

Reworked scrobble handling

This commit is contained in:
krateng 2022-04-06 17:42:48 +02:00
parent f4e42f9256
commit 1257768e33
3 changed files with 63 additions and 27 deletions

View File

@ -251,13 +251,22 @@ def post_scrobble(artist:Multi=None,**keys):
:param int time: UNIX timestamp of the scrobble. Optional, not needed if scrobble is at time of request.
:param boolean nofix: Skip server-side metadata parsing. Optional.
"""
#artists = "/".join(artist)
# url multi args automatically become list
keys['artists'] = artist if artist is not None else keys.get("artists")
keys['fix'] = keys.get("nofix") is None
if keys.get('time') is not None: keys['time'] = int(keys.get('time'))
return database.incoming_scrobble(**keys,client=request.malojaclient)
rawscrobble = {
'track_artists':artist if artist is not None else keys.get("artists"),
'track_title':keys.get('title'),
'album_name':keys.get('album'),
'album_artists':keys.get('albumartists'),
'scrobble_duration':keys.get('duration'),
'track_length':keys.get('length'),
'scrobble_time':int(keys.get('time')) if (keys.get('time') is not None) else None
}
return database.incoming_scrobble(
rawscrobble,
client=request.malojaclient,
fix=(keys.get("nofix") is None)
)
# TODO: malojaclient needs to be converted to proper argument in doreah

View File

@ -68,44 +68,68 @@ coa = CollectorAgent()
def incoming_scrobble(artists,title,album=None,albumartists=None,duration=None,length=None,time=None,fix=True,client=None,dbconn=None,**kwargs):
## this function accepts a flat dict - all info of the scrobble should be top level key
## but can contain a list as value
## the following keys are valid:
## scrobble_duration int
## scrobble_time int
## track_title str, mandatory
## track_artists list, mandatory
## track_length int
## album_name str
## album_artists list
##
##
##
##
##
##
def incoming_scrobble(rawscrobble,fix=True,client=None,dbconn=None,**kwargs):
# TODO: just collecting all extra kwargs now. at some point, rework the authenticated api with alt function
# to actually look at the converted args instead of the request object and remove the key
# so that this function right here doesnt get the key passed to it
if time is None:
time = int(datetime.datetime.now(tz=datetime.timezone.utc).timestamp())
log("Incoming scrobble (" + str(client) + "): ARTISTS: " + str(artists) + ", TRACK: " + title,module="debug")
if fix:
(artists,title) = cla.fullclean(artists,title)
if len(artists) == 0 or title == "":
if (not "track_artists" in rawscrobble) or (len(rawscrobble['track_artists']) == 0) or (not "track_title" in rawscrobble):
log(f"Incoming scrobble {rawscrobble} [Source: {client}] is not valid")
return {"status":"failure"}
if albumartists is None:
albumartists = artists
log(f"Incoming scrobble [{client}]: {rawscrobble}")
# raw scrobble to processed info
scrobbleinfo = {**rawscrobble}
if fix:
scrobbleinfo['track_artists'],scrobbleinfo['track_title'] = cla.fullclean(scrobbleinfo['track_artists'],scrobbleinfo['track_title'])
scrobbleinfo['scrobble_time'] = scrobbleinfo.get('scrobble_time') or int(datetime.datetime.now(tz=datetime.timezone.utc).timestamp())
# processed info to internal scrobble dict
scrobbledict = {
"time":time,
"time":scrobbleinfo.get('scrobble_time'),
"track":{
"artists":artists,
"title":title,
"artists":scrobbleinfo.get('track_artists'),
"title":scrobbleinfo.get('track_title'),
"album":{
"name":album,
"artists":albumartists
"name":scrobbleinfo.get('album_name'),
"artists":scrobbleinfo.get('album_artists')
},
"length":None
"length":scrobbleinfo.get('track_length')
},
"duration":duration,
"origin":"client:" + client if client else "generic"
"duration":scrobbleinfo.get('scrobble_duration'),
"origin":f"client: {client}" if client else "generic",
"extra":{
k:scrobbleinfo[k] for k in scrobbleinfo if k not in
['scrobble_time','track_artists','track_title','track_length','scrobble_duration','album_name','album_artists']
},
"rawscrobble":rawscrobble
}
sqldb.add_scrobble(scrobbledict,dbconn=dbconn)
proxy_scrobble_all(artists,title,time)
proxy_scrobble_all(scrobbledict['track']['artists'],scrobbledict['track']['title'],scrobbledict['time'])
dbcache.invalidate_caches(time)
dbcache.invalidate_caches(scrobbledict['time'])
return {"status":"success","scrobble":scrobbledict}

View File

@ -143,6 +143,9 @@ def connection_provider(func):
# "origin":string,
# "extra":{string-keyed mapping for all flags with the scrobble}
# }
#
# The dict sent to the DB can have one extra field 'rawscrobble', but this isn't
# returned on read
@ -207,7 +210,7 @@ def scrobble_dict_to_db(info):
"duration":info['duration'],
"track_id":get_track_id(info['track']),
"extra":json.dumps(info.get('extra',{})),
"rawscrobble":json.dumps(info)
"rawscrobble":json.dumps(info.get('rawscrobble'))
}
def track_dict_to_db(info):