forked from tanner/qotnews
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
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 |