-
Notifications
You must be signed in to change notification settings - Fork 0
/
personsmatch.py
106 lines (80 loc) · 2.89 KB
/
personsmatch.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
import os
import sys
import operator
import requests
import json
import twitter
from watson_developer_cloud import PersonalityInsightsV2 as PersonalityInsights
def get_twitter_client():
try:
# Set the environment variables.
twitter_consumer_key = os.environ['PERSONALITY_MATCH_CONSUMER_KEY']
twitter_consumer_secret = os.environ['PERSONALITY_MATCH_CONSUMER_SECRET']
twitter_access_token = os.environ['PERSONALITY_MATCH_ACCESS_TOKEN']
twitter_access_secret = os.environ['PERSONALITY_MATCH_ACCESS_SECRET']
except KeyError:
sys.stderr.write("TWITTER_* environment variables not set\n")
sys.exit(1)
# Create an instance of the Twitter API.
twitter_api = twitter.Api(consumer_key=twitter_consumer_key, consumer_secret=twitter_consumer_secret, access_token_key=twitter_access_token, access_token_secret=twitter_access_secret)
return twitter_api
def get_pi_client():
try:
# IBM Bluemix credentials for Personality Insights
pi_username = os.environ['PI_USERNAME']
pi_password = os.environ['PI_PASSWORD']
except KeyError:
sys.stderr.write("PI_* environment variables not set\n")
sys.exit(1)
# Create Personality Insights instance
personality_insights = PersonalityInsights(username=pi_username, password=pi_password)
return personality_insights
def analyze(handle):
# Retrieve data from Twitter.
statuses = get_twitter_client().GetUserTimeline(screen_name=handle, count=200, include_rts=False)
# This variable saves tweets.
# Convert to bytes first.
text = b""
#text = "".encode() -> method 2
# View the results.
for status in statuses:
if (status.lang == 'en'):
text += status.text.encode('utf-8')
pi_result = get_pi_client().profile(text)
return pi_result
def flatten(orig):
data = {}
for c in orig['tree']['children']:
if 'children' in c:
for c2 in c['children']:
if 'children' in c2:
for c3 in c2['children']:
if 'children' in c3:
for c4 in c3['children']:
if (c4['category'] == 'personality'):
data[c4['id']] = c4['percentage']
if 'children' not in c3:
if (c3['category'] == 'personality'):
data[c3['id']] = c3['percentage']
return data
def compare(dict1, dict2):
compared_data = {}
for keys in dict1:
if dict1[keys] != dict2[keys]:
compared_data[keys] = abs(dict1[keys] - dict2[keys])
return compared_data
user_handle = "@khwilo"
match_handle = "@jadytrix"
user_result = analyze(user_handle)
match_result = analyze(match_handle)
user = flatten(user_result)
match = flatten(match_result)
compared_results = compare(user, match)
sorted_result = sorted(compared_results.items(), key=operator.itemgetter(1))
for keys, value in sorted_result[:5]:
print(keys),
print(user[keys]),
print('->'),
print(match[keys]),
print('->'),
print(compared_results[keys])