Add favourite tag, artist, and most produced tag to user nodes

This commit is contained in:
Thomas Wade 2018-09-30 15:07:26 +09:30
parent 1440e060e4
commit 7316afb1a5

View File

@ -442,6 +442,8 @@ for _, user in user_dict.items():
continue continue
favourite_tag, favourite_artist = Favourite_Tag_And_Artist(user.likes, track_dict) favourite_tag, favourite_artist = Favourite_Tag_And_Artist(user.likes, track_dict)
most_produced_tag, _ = Favourite_Tag_And_Artist(user.tracks, track_dict)
graph.add_node( \ graph.add_node( \
user.url_username, \ user.url_username, \
@ -450,10 +452,16 @@ for _, user in user_dict.items():
track_count=str(user.track_count), \ track_count=str(user.track_count), \
like_count=str(user.like_count), \ like_count=str(user.like_count), \
following_count=str(user.following_count), \ following_count=str(user.following_count), \
follower_count=str(user.follower_count) \ follower_count=str(user.follower_count), \
most_produced_tag=str(most_produced_tag), \
favourite_tag=str(favourite_tag), \
favourite_artist=str(favourite_artist) \
) )
print(user.username) print(user.username)
print(' Most produced tag: ' + str(most_produced_tag))
print(' Favourite tag: ' + str(favourite_tag))
print(' Favourite artist: ' + str(favourite_artist))
print(' ' + str(user.track_count) + ' tracks') print(' ' + str(user.track_count) + ' tracks')
print(' ' + str(user.like_count) + ' likes') print(' ' + str(user.like_count) + ' likes')
print(' Following ' + str(user.following_count)) print(' Following ' + str(user.following_count))