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))