gfycat_helper.py

import requests
import datetime
import json

from django.core.cache import cache
from django.conf import settings

GFYCAT_GET_URL = 'https://api.gfycat.com/v1/gfycats/{}'
GFYCAT_TOKEN_URL = 'https://api.gfycat.com/v1/oauth/token/'
GFYCAT_TOKEN_CACHE_KEY = 'gfycat_token'

def get_gfycat_auth_header():
    token = cache.get(GFYCAT_TOKEN_CACHE_KEY)
    if not token:
        data = {
            'client_id': settings.GFYCAT_API_KEY,
            'client_secret': settings.GFYCAT_API_SECRET,
            'grant_type': 'client_credentials'
        }
        resp = requests.post(GFYCAT_TOKEN_URL, data=json.dumps(data))
        if resp.status_code < 300:
            resp_data = resp.json()
            token = resp_data['access_token']
            cache.set(GFYCAT_TOKEN_CACHE_KEY, token, resp_data['expires_in'])
        else:
            return None
    return {'Authorization': 'Bearer {}'.format(token)}

def get_gfycat_header(has_data=False):
    headers = {'Content-Type': 'application/json'} if has_data else {}
    headers.update(get_gfycat_auth_header())
    return headers
  
def get_by_id(gfycat_id):
    result = requests.get(GFYCAT_GET_URL.format(gfycat_id),
                          headers=get_gfycat_header(has_data=False)).json().get(
        'gfyItem')