From 7316afb1a5ae97bb1648e32fab1087f3dfc0dd2f Mon Sep 17 00:00:00 2001 From: Thomas Wade Date: Sun, 30 Sep 2018 15:07:26 +0930 Subject: [PATCH] Add favourite tag, artist, and most produced tag to user nodes --- scrape.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/scrape.py b/scrape.py index 6ba29f9..7ff05ef 100755 --- a/scrape.py +++ b/scrape.py @@ -442,6 +442,8 @@ for _, user in user_dict.items(): continue 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( \ user.url_username, \ @@ -450,10 +452,16 @@ for _, user in user_dict.items(): track_count=str(user.track_count), \ like_count=str(user.like_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(' 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.like_count) + ' likes') print(' Following ' + str(user.following_count))