House cleaning.
This commit is contained in:
Chris Rossi
@@ -0,0 +1,112 @@
|
||||
"""
|
||||
Class for accessing LinkedIn's API.
|
||||
"""
|
||||
import hashlib
|
||||
import json
|
||||
import urllib2
|
||||
import urlparse
|
||||
import uuid
|
||||
|
||||
from django.conf import settings
|
||||
from django.core.management.base import CommandError
|
||||
|
||||
from ...models import LinkedInToken
|
||||
|
||||
|
||||
class LinkedinAPI(object):
|
||||
"""
|
||||
Encapsulates the LinkedIn API.
|
||||
"""
|
||||
def __init__(self):
|
||||
config = getattr(settings, "LINKEDIN_API", None)
|
||||
if not config:
|
||||
raise CommandError("LINKEDIN_API is not configured")
|
||||
self.config = config
|
||||
|
||||
try:
|
||||
self.tokens = LinkedInToken.objects.get()
|
||||
except LinkedInToken.DoesNotExist:
|
||||
self.tokens = None
|
||||
|
||||
self.state = str(uuid.uuid4())
|
||||
|
||||
def http_error(self, error, message):
|
||||
"""
|
||||
Handle an unexpected HTTP response.
|
||||
"""
|
||||
print "!!ERROR!!"
|
||||
print error
|
||||
print error.read()
|
||||
raise CommandError(message)
|
||||
|
||||
def authorization_url(self):
|
||||
"""
|
||||
Synthesize a URL for beginning the authorization flow.
|
||||
"""
|
||||
config = self.config
|
||||
return ("https://www.linkedin.com/uas/oauth2/authorization"
|
||||
"?response_type=code"
|
||||
"&client_id=%s&state=%s&redirect_uri=%s" % (
|
||||
config['CLIENT_ID'], self.state, config['REDIRECT_URI']))
|
||||
|
||||
def get_authorization_code(self, redirect):
|
||||
"""
|
||||
Extract the authorization code from the redirect URL at the end of
|
||||
the authorization flow.
|
||||
"""
|
||||
query = urlparse.parse_qs(urlparse.urlparse(redirect).query)
|
||||
assert query['state'][0] == self.state, (query['state'][0], self.state)
|
||||
return query['code'][0]
|
||||
|
||||
def get_access_token(self, code):
|
||||
"""
|
||||
Given an authorization code, get an access token.
|
||||
"""
|
||||
config = self.config
|
||||
url = ("https://www.linkedin.com/uas/oauth2/accessToken"
|
||||
"?grant_type=authorization_code"
|
||||
"&code=%s&redirect_uri=%s&client_id=%s&client_secret=%s" % (
|
||||
code, config['REDIRECT_URI'], config['CLIENT_ID'],
|
||||
config['CLIENT_SECRET']))
|
||||
|
||||
try:
|
||||
response = urllib2.urlopen(url).read()
|
||||
except urllib2.HTTPError, error:
|
||||
self.http_error(error, "Unable to retrieve access token")
|
||||
|
||||
access_token = json.loads(response)['access_token']
|
||||
try:
|
||||
tokens = LinkedInToken.objects.get()
|
||||
tokens.access_token = access_token
|
||||
tokens.authorization_code = code
|
||||
except LinkedInToken.DoesNotExist:
|
||||
tokens = LinkedInToken(access_token=access_token)
|
||||
tokens.save()
|
||||
self.tokens = tokens
|
||||
|
||||
return access_token
|
||||
|
||||
def batch(self, emails):
|
||||
"""
|
||||
Get the LinkedIn status for a batch of emails.
|
||||
"""
|
||||
if self.tokens is None:
|
||||
raise CommandError(
|
||||
"You must log in to LinkedIn in order to use this script. "
|
||||
"Please use the 'login' command to log in to LinkedIn.")
|
||||
|
||||
def md5(email):
|
||||
"Compute md5 hash for an email address."
|
||||
md5hash = hashlib.md5()
|
||||
md5hash.update(email)
|
||||
return md5hash.hexdigest()
|
||||
|
||||
hashes = ','.join(("email-hash=" + md5(email) for email in emails))
|
||||
url = "https://api.linkedin.com/v1/people::(%s):(id)" % hashes
|
||||
url += "?oauth2_access_token=%s" % self.tokens.access_token
|
||||
try:
|
||||
response = urllib2.urlopen(url).read()
|
||||
print "GOT IT!", response
|
||||
except urllib2.HTTPError, error:
|
||||
self.http_error(error, "Unable to access People API")
|
||||
return (True for email in emails)
|
||||
|
||||
@@ -1,99 +0,0 @@
|
||||
import hashlib
|
||||
import json
|
||||
import urllib2
|
||||
import urlparse
|
||||
import uuid
|
||||
|
||||
from django.conf import settings
|
||||
from django.core.management.base import CommandError
|
||||
|
||||
from ...models import LinkedInToken
|
||||
|
||||
class LinkedinAPI(object):
|
||||
"""
|
||||
Encapsulates the LinkedIn API.
|
||||
"""
|
||||
def __init__(self):
|
||||
config = getattr(settings, "LINKEDIN_API", None)
|
||||
if not config:
|
||||
raise CommandError("LINKEDIN_API is not configured")
|
||||
self.config = config
|
||||
|
||||
try:
|
||||
self.tokens = LinkedInToken.objects.get()
|
||||
except LinkedInToken.DoesNotExist:
|
||||
self.tokens = None
|
||||
|
||||
self.state = str(uuid.uuid4())
|
||||
|
||||
def http_error(error, message):
|
||||
print "!!ERROR!!"
|
||||
print error
|
||||
print error.read()
|
||||
raise CommandError(message)
|
||||
|
||||
def authorization_url(self):
|
||||
config = self.config
|
||||
return ("https://www.linkedin.com/uas/oauth2/authorization"
|
||||
"?response_type=code"
|
||||
"&client_id=%s&state=%s&redirect_uri=%s" % (
|
||||
config['CLIENT_ID'], self.state, config['REDIRECT_URI']))
|
||||
|
||||
def get_authorization_code(self, redirect):
|
||||
query = urlparse.parse_qs(urlparse.urlparse(redirect).query)
|
||||
assert query['state'][0] == self.state, (query['state'][0], self.state)
|
||||
return query['code'][0]
|
||||
|
||||
def get_access_token(self, code):
|
||||
config = self.config
|
||||
url = ("https://www.linkedin.com/uas/oauth2/accessToken"
|
||||
"?grant_type=authorization_code"
|
||||
"&code=%s&redirect_uri=%s&client_id=%s&client_secret=%s" % (
|
||||
code, config['REDIRECT_URI'], config['CLIENT_ID'],
|
||||
config['CLIENT_SECRET']))
|
||||
|
||||
try:
|
||||
response = urllib2.urlopen(url).read()
|
||||
except urllib2.HTTPError, error:
|
||||
self.http_error(error, "Unable to retrieve access token")
|
||||
|
||||
access_token = json.loads(response)['access_token']
|
||||
try:
|
||||
tokens = LinkedInToken.objects.get()
|
||||
tokens.access_token = access_token
|
||||
tokens.authorization_code = code
|
||||
except LinkedInToken.DoesNotExist:
|
||||
tokens = LinkedInToken(access_token=access_token)
|
||||
tokens.save()
|
||||
self.tokens = tokens
|
||||
|
||||
return access_token
|
||||
|
||||
def batch(self, emails):
|
||||
"""
|
||||
Get the LinkedIn status for a batch of emails.
|
||||
"""
|
||||
if self.tokens is None:
|
||||
raise CommandError(
|
||||
"You must log in to LinkedIn in order to use this script. "
|
||||
"Please use the 'login' command to log in to LinkedIn.")
|
||||
|
||||
def md5(email):
|
||||
"Compute md5 hash for an email address."
|
||||
hash = hashlib.md5()
|
||||
hash.update(email)
|
||||
return hash.hexdigest()
|
||||
|
||||
hashes = ','.join(("email-hash=" + md5(email) for email in emails))
|
||||
url = "https://api.linkedin.com/v1/people::(%s):(id)" % hashes
|
||||
url += "?oauth2_access_token=%s" % self.tokens.access_token
|
||||
try:
|
||||
response = urllib2.urlopen(url).read()
|
||||
except urllib2.HTTPError, error:
|
||||
print "!!ERROR!!"
|
||||
print error
|
||||
print error.read()
|
||||
|
||||
raise CommandError("Unable to access People API")
|
||||
|
||||
return (True for email in emails)
|
||||
@@ -13,7 +13,7 @@ from django.utils import timezone
|
||||
from optparse import make_option
|
||||
|
||||
from ...models import LinkedIn
|
||||
from .api import LinkedinAPI
|
||||
from . import LinkedinAPI
|
||||
|
||||
FRIDAY = 4
|
||||
|
||||
|
||||
@@ -3,7 +3,7 @@ Log into LinkedIn API.
|
||||
"""
|
||||
from django.core.management.base import BaseCommand
|
||||
|
||||
from .api import LinkedinAPI
|
||||
from . import LinkedinAPI
|
||||
|
||||
|
||||
class Command(BaseCommand):
|
||||
|
||||
Reference in New Issue
Block a user