def get(self, username = None):
if username is None:
self.redirect('/user')
return
user = User.gql("WHERE username = :1", username)
if user.count() == 0:
self.redirect('/')
return
else:
user = user[0]
session = get_current_session()
authorized_tokens = session.get('authorized_tokens', None)
is_user = user.key().name() == authorized_tokens['user_id'] \
if authorized_tokens else False
statistic = UserStatistic.get_by_key_name(user.key().name())
if statistic.statistics:
sorted_dict = simplejson.loads(statistic.statistics)
template_values = {
'user': user,
'is_user': is_user,
'sorted_dict': sorted_dict,
'total': statistic.total,
'last_check': statistic.updated \
+ timedelta(hours=7),
'start_time': statistic.start_time \
+ timedelta(hours=7),
'end_time': statistic.end_time \
+ timedelta(hours=7),
}
else:
template_values = {
'user': user,
'is_user': is_user,
}
path = os.path.join(TEMPLATE, 'timeline.html')
self.response.out.write(template.render(path, template_values))
def post(self):
twitter_id = self.request.get('twitter_id', None)
if twitter_id is None:
return
user = User.get_by_key_name(twitter_id)
username = user.username
twitter = Twython(
twitter_token = CONSUMER_KEY,
twitter_secret = CONSUMER_SECRET,
oauth_token = user.oauth_token,
oauth_token_secret = user.oauth_token_secret,
)
statistic = UserStatistic.get_by_key_name(twitter_id)
stat = dict()
total = 0
page = 0
max_id = None
start_time = None
end_time = None
while True:
if max_id is None:
tweets = twitter.getFriendsTimeline(
count=200,
include_entities=1,
)
else:
tweets = twitter.getFriendsTimeline(
count=200,
include_entities=1,
max_id=max_id,
#page=page,
)
if len(tweets) == 0:
break
else:
page = page + 1
total = total + len(tweets)
if end_time is None:
end_time = datetime \
.strptime(tweets[0]['created_at'],
"%a %b %d %H:%M:%S +0000 %Y")
last_tweet = tweets[len(tweets) - 1]
start_time = datetime \
.strptime(last_tweet['created_at'],
"%a %b %d %H:%M:%S +0000 %Y")
for tweet in tweets:
user = tweet['user']['screen_name']
if not stat.has_key(user):
stat[user] = 0
stat[user] = stat[user] + 1
max_id = tweets[len(tweets) - 1]['id']
if total > MAX_TWEETS:
break
sorted_stat = sorted(stat, key=stat.get)
sorted_stat.reverse()
sorted_dict = []
for item in sorted_stat:
if stat[item] > 4:
sorted_dict.append(dict(
user = item,
count = stat[item],
))
statistic.start_time = start_time
statistic.end_time = end_time
statistic.total = total
statistic.statistics = simplejson.dumps(sorted_dict)
statistic.count += 1
statistic.put()
# notify user
try:
task = Task(
url = '/notify',
params = {'username': username}
)
queue = Queue(name='notify-user')
queue.add(task)
except:
# need to send to error page if error is happened
logging.exception('something bad happened')
def get(self):
session = get_current_session()
authorized_tokens = session.get('authorized_tokens', None)
if authorized_tokens is None:
self.redirect('/connect')
twitter = Twython(
twitter_token = CONSUMER_KEY,
twitter_secret = CONSUMER_SECRET,
oauth_token = authorized_tokens['oauth_token'],
oauth_token_secret = authorized_tokens['oauth_token_secret']
)
twitter_id = authorized_tokens['user_id']
username = authorized_tokens['screen_name']
statistic = UserStatistic.get_by_key_name(twitter_id)
if statistic is None:
statistic = UserStatistic(
key_name=twitter_id,
twitter_id=long(twitter_id),
)
statistic.put()
if statistic.statistics == None \
or (statistic.updated + timedelta(hours=1)) < datetime.now():
try:
task = Task(
url = '/fetch',
params = {'twitter_id': twitter_id}
)
queue = Queue(name='fetch-tweets')
queue.add(task)
statistic.updated = datetime.now()
statistic.put()
template_values = {
'username': username,
'updated': True,
'error': False,
}
except:
# need to send to error page if error is happened
logging.exception('something bad happened')
template_values = {
'username': username,
'updated': False,
'error': True,
}
else:
template_values = {
'username': username,
'updated': False,
'error': False,
}
path = os.path.join(TEMPLATE, 'message.html')
self.response.out.write(template.render(path, template_values))
