21 lines
643 B
Python
21 lines
643 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
from django.utils.functional import SimpleLazyObject
|
||
|
|
from ghapi.api import GitHub
|
||
|
|
from social_auth.db.django_models import UserSocialAuth
|
||
|
|
|
||
|
|
|
||
|
|
def get_github(request):
|
||
|
|
if not request.user.is_authenticated():
|
||
|
|
return GitHub()
|
||
|
|
try:
|
||
|
|
social = UserSocialAuth.objects.get(provider='github', user_id=request.user.pk)
|
||
|
|
except UserSocialAuth.DoesNotExist:
|
||
|
|
return GitHub()
|
||
|
|
token = social.tokens.get('access_token', None)
|
||
|
|
return GitHub(token)
|
||
|
|
|
||
|
|
|
||
|
|
class GithubAPIMiddleware(object):
|
||
|
|
def process_request(self, request):
|
||
|
|
request.user = SimpleLazyObject(lambda : get_github(request))
|