qotnews/apiserver/misc/api.py

import logging
logging.basicConfig(
        format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
        level=logging.DEBUG)

import requests

GOOGLEBOT_USER_AGENT = "Mozilla/5.0 (compatible; Googlebot/2.1; +http://www.google.com/bot.html)"
GOOGLEBOT_IP = '66.249.66.1'
TIMEOUT = 30

def xml(route, ref=None, headers=dict(), use_googlebot=True):
    try:
        if use_googlebot:
            headers['User-Agent'] = GOOGLEBOT_USER_AGENT
            headers['X-Forwarded-For'] = GOOGLEBOT_IP
        r = requests.get(route(ref), headers=headers, timeout=TIMEOUT)
        if r.status_code != 200:
            raise Exception('Bad response code ' + str(r.status_code))
        return r.text
    except KeyboardInterrupt:
        raise
    except BaseException as e:
        logging.error('Problem hitting URL: {}'.format(str(e)))
        return False

def json(route, ref=None, headers=dict(), use_googlebot=True):
    try:
        if use_googlebot:
            headers['User-Agent'] = GOOGLEBOT_USER_AGENT
            headers['X-Forwarded-For'] = GOOGLEBOT_IP
        r = requests.get(route(ref), headers=headers, timeout=TIMEOUT)
        if r.status_code != 200:
            raise Exception('Bad response code ' + str(r.status_code))
        return r.json()
    except KeyboardInterrupt:
        raise
    except BaseException as e:
        logging.error('Problem hitting URL: {}'.format(str(e)))
        return False