89 lines
2.4 KiB
Python
89 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
# GPL 2008
|
|
import os
|
|
import gzip
|
|
import StringIO
|
|
import urllib
|
|
import urllib2
|
|
|
|
from chardet.universaldetector import UniversalDetector
|
|
|
|
|
|
# Default headers for HTTP requests.
|
|
DEFAULT_HEADERS = {
|
|
'User-Agent': 'Mozilla/5.0 (X11; U; Linux i386; en-US; rv:1.9.1.1) Gecko/20090716 Firefox/3.5',
|
|
'Accept-Encoding': 'gzip'
|
|
}
|
|
|
|
def status(url, data=None, headers=DEFAULT_HEADERS):
|
|
try:
|
|
f = openUrl(url, data, headers)
|
|
s = f.code
|
|
except urllib2.HTTPError, e:
|
|
s = e.code
|
|
return s
|
|
|
|
def exists(url, data=None, headers=DEFAULT_HEADERS):
|
|
s = status(url, data, headers)
|
|
if s >= 200 and s < 400:
|
|
return True
|
|
return False
|
|
|
|
def getHeaders(url, data=None, headers=DEFAULT_HEADERS):
|
|
try:
|
|
f = openUrl(url, data, headers)
|
|
f.headers['Status'] = "%s" % f.code
|
|
headers = f.headers
|
|
f.close()
|
|
except urllib2.HTTPError, e:
|
|
e.headers['Status'] = "%s" % e.code
|
|
headers = e.headers
|
|
return dict(headers)
|
|
|
|
def openUrl(url, data=None, headers=DEFAULT_HEADERS):
|
|
url = url.replace(' ', '%20')
|
|
req = urllib2.Request(url, data, headers)
|
|
return urllib2.urlopen(req)
|
|
|
|
def readUrl(url, data=None, headers=DEFAULT_HEADERS, returnHeaders=False):
|
|
f = openUrl(url, data, headers)
|
|
data = f.read()
|
|
f.close()
|
|
if f.headers.get('content-encoding', None) == 'gzip':
|
|
data = gzip.GzipFile(fileobj=StringIO.StringIO(data)).read()
|
|
if returnHeaders:
|
|
f.headers['Status'] = "%s" % f.code
|
|
return dict(f.headers), data
|
|
return data
|
|
|
|
def readUrlUnicode(url):
|
|
data = readUrl(url)
|
|
encoding = getEncoding(data)
|
|
if not encoding:
|
|
encoding = 'latin-1'
|
|
return unicode(data, encoding)
|
|
|
|
def getEncoding(data):
|
|
if 'content="text/html; charset=utf-8"' in data:
|
|
return 'utf-8'
|
|
elif 'content="text/html; charset=iso-8859-1"' in data:
|
|
return 'iso-8859-1'
|
|
detector = UniversalDetector()
|
|
for line in data.split('\n'):
|
|
detector.feed(line)
|
|
if detector.done:
|
|
break
|
|
detector.close()
|
|
return detector.result['encoding']
|
|
|
|
def saveUrl(url, filename, overwrite=False):
|
|
if not os.path.exists(filename) or overwrite:
|
|
dirname = os.path.dirname(filename)
|
|
if not os.path.exists(dirname):
|
|
os.makedirs(dirname)
|
|
data = readUrl(url)
|
|
f = open(filename, 'w')
|
|
f.write(data)
|
|
f.close()
|
|
|