utils.py

#!/usr/bin/env python2
# -- coding: utf-8 --

import os
import sys
import requests
from getpass import getpass

API_URL = 'https://www.muckrock.com/api_v1/'

def get_api_key():
    # try getting an environment variable before requesting the auth information
    try:
        token = os.environ['MUCKROCK_TOKEN']
    except KeyError:
        user = raw_input('Username: ')
        pw = getpass()
        response = requests.post('https://www.muckrock.com/api_v1/token-auth/', data={'username': user, 'password': pw})
        if not response.raise_for_status():
            data = response.json()
            token = data['token']
            os.environ['MUCKROCK_TOKEN'] = token
            print '\nYour token is: %(token)s\n\nSet it for future scripts with this command:\n\nexport MUCKROCK_TOKEN="%(token)s"\n' % {
                'token': token
            }
    return token

def get_headers(token=None):
    if token:
        return {
            'Authorization': 'Token %s' % token,
            'content-type': 'application/json'
        }
    else:
        return {'content-type': 'application/json'}

def display_progress(current, total):
    percent = (current / total) * 100.00
    sys.stdout.write("\r%d%%" % percent)
    sys.stdout.flush()