twitter_influence.py

import sys
import json

def usage():
    print('Usage: ')
    print("python {} <username>".format(sys.argv[0]))

if __name__ == "__main__":
    if len(sys.argv) != 3:
        usage()
        sys.exit(1)

    screen_name1 = sys.argv[1]
    screen_name2 = sys.argv[2]

    #Build up a list of followers
    followers_file1 = 'Twitter_Profiles/{}/followers.jsonl'.format(screen_name1)
    followers_file2 = 'Twitter_Profiles/{}/followers.jsonl'.format(screen_name2)

    with open(followers_file1) as f1, open(followers_file2) as f2:
        reach1 = []
        reach2 = []
        for line in f1:
            profile = json.loads(line)
            reach1.append((profile['screen_name'], profile['followers_count']))
        for line in f2:
            profile = json.loads(line)
            reach2.append((profile['screen_name'], profile['followers_count']))


    #Load basic statistics
    profile_file1 = 'Twitter_Profiles/{}/user_profile.json'.format(screen_name1)
    profile_file2 = 'Twitter_Profiles/{}/user_profile.json'.format(screen_name2)
    with open(profile_file1) as f1, open(profile_file2) as f2:
        profile1 = json.load(f1)
        profile2 = json.load(f2)

        followers1 = profile1['followers_count']
        followers2 = profile2['followers_count']

        tweets1 = profile['statuses_count']
        tweets2 = profile['statuses_count']

    sum_reach1 = sum([x[1] for x in reach1]) #sum up all of a user's followers, followers
    sum_reach2 = sum([x[1] for x in reach2])
    avg_followers1 = round(sum_reach1/ followers1, 2)
    avg_followers2 = round(sum_reach2/ followers2, 2)

    #Load the timelines for the two users to observe the number of times their tweets have been favorited
    timeline_file1 = 'user_timeline_{}.jsonl'.format(screen_name1)
    timeline_file2 = 'user_timeline_{}.jsonl'.format(screen_name2)
    with open(timeline_file1) as f1, open(timeline_file2) as f2:
        favorited_count1, retweet_count1 = [], []
        favorited_count2, retweet_count2 = [], []
        for line in f1:
            tweet = json.loads(line)
            favorited_count1.append(tweet['favorite_count'])
            retweet_count1.append(tweet['retweet_count'])

        for line in f2:
            tweet = json.loads(line)
            favorited_count2.append(tweet['favorite_count'])
            retweet_count2.append(tweet['retweet_count'])

    #Average number of favorites and retweets

    avg_favorite1 = round(sum(favorited_count1)/ tweets1, 2)
    avg_favorite2 = round(sum(favorited_count2)/ tweets2, 2) #avg favorite per tweet
    avg_retweet1 = round(sum(retweet_count1)/ tweets1, 2)
    avg_retweet2 = round(sum(retweet_count2)/ tweets2, 2)
    favorite_per_user1 = round(sum(favorited_count1)/ followers1, 2) # your avg follower gives u x favorites in all of the likes u have accumulated
    favorite_per_user2 = round(sum(favorited_count2)/ followers2, 2)
    retweet_per_user1 = round(sum(retweet_count1)/ followers1, 2) # avg retweet per follower
    retweet_per_user2 = round(sum(retweet_count2)/ followers2, 2)

    print("------------ Stats for {} ---------------------".format(screen_name1))
    print("{} follwers".format(followers1))
    print("{} users can be reached by 1-degree connections.".format(sum_reach1))
    print("{}'s followers averagely have: {} followers each".format(screen_name1, avg_followers1))
    #print("Favorited {} times ({} favorites per tweet, and {} per user)".format(sum(favorited_count1), avg_favorite1, favorite_per_user1))
    print("Retweeted {} times ({} retweets per tweet, and {} per user)".format(sum(retweet_count1), avg_retweet1, retweet_per_user1))

    print("---------------------------- Next User ----------------------------")

    print("------------ Stats for {} ---------------------".format(screen_name2))
    print("{} follwers".format(followers2))
    print("{} users can be reached by 1-degree connections.".format(sum_reach2))
    print("{}'s followers averagely have: {} followers each".format(screen_name2, avg_followers2))
    #print("Favorited {} times ({} favorites per tweet, and {} per user)".format(sum(favorited_count2), avg_favorite2, favorite_per_user2))
    print("Retweeted {} times ({} retweets per tweet, and {} per user)".format(sum(retweet_count2), avg_retweet2, retweet_per_user2))