Skip to content
This repository has been archived by the owner on Feb 8, 2018. It is now read-only.

Commit

Permalink
Here's a one-off script for #1936
Browse files Browse the repository at this point in the history
  • Loading branch information
chadwhitacre committed Feb 7, 2014
1 parent 22557e7 commit 69ed7b4
Showing 1 changed file with 57 additions and 0 deletions.
57 changes: 57 additions & 0 deletions scripts/update_user_info.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,57 @@
#!/usr/bin/env python
"""This is a one-off script to update user_info for #1936.
This could be generalized for #900.
"""
import os
import time

import requests
from gittip import wireup
from requests_oauthlib import OAuth1


db = wireup.db()
oauth = OAuth1( os.environ['TWITTER_CONSUMER_KEY']
, os.environ['TWITTER_CONSUMER_SECRET']
, os.environ['TWITTER_ACCESS_TOKEN']
, os.environ['TWITTER_ACCESS_TOKEN_SECRET']
)
elsewhere = db.all("SELECT user_id FROM ELSEWHERE WHERE platform='twitter';")
url = "https://api.twitter.com/1.1/users/show.json?user_id=%s"

for user_id in elsewhere:
response = requests.get(url % user_id, auth=oauth)

if response.status_code != 200:
# Who knows what happened? Bail.
# (Supposedly we shouldn't hit 429, at least).
raise SystemExit


# Update!
# =======

user_info = response.json()

# flatten per upsert method in gittip/elsewhere/__init__.py
for k, v in user_info.items():
user_info[k] = unicode(v)

db.run("UPDATE elsewhere SET user_info=%s WHERE user_id=%s", (user_info, user_id))


# Stay under our rate limit.
# =========================
# We get 180 per 15 minutes for the users/show endpoint, per:
#
# https://dev.twitter.com/docs/rate-limiting/1.1/limits

print response.headers['X-RATE-LIMIT-REMAINING']
nremaining = int(response.headers['X-RATE-LIMIT-REMAINING'])
sleep_for = 5
if nremaining < 180:
reset = int(response.headers['X-RATE-LIMIT-RESET'])
sleep_for = reset - time.time()
time.sleep(sleep_for)

0 comments on commit 69ed7b4

Please sign in to comment.