96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
#http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/146306
|
|
# Submitter: Wade Leftwich
|
|
# Licensing:
|
|
# according to http://aspn.activestate.com/ASPN/Cookbook/Python
|
|
# "Except where otherwise noted, recipes in the Python Cookbook are published under the Python license ."
|
|
# This recipe is covered under the Python license: http://www.python.org/license
|
|
|
|
import httplib, mimetypes, urllib2
|
|
import socket
|
|
from socket import error, herror, gaierror, timeout
|
|
socket.setdefaulttimeout(None)
|
|
import urlparse
|
|
|
|
def link_exists(host, selector):
|
|
url = "http://" + host + selector
|
|
host, path = urlparse.urlsplit(url)[1:3]
|
|
found = 0
|
|
msg = "ping"
|
|
try:
|
|
h = httplib.HTTP(host) ## Make HTTPConnection Object
|
|
h.putrequest('HEAD', selector)
|
|
h.putheader('content-type', "text/plain")
|
|
h.putheader('content-length', str(len(msg)))
|
|
h.putheader('Host', host)
|
|
h.endheaders()
|
|
h.send(msg)
|
|
|
|
errcode, errmsg, headers = h.getreply()
|
|
if errcode == 200:
|
|
found = 1
|
|
else:
|
|
print "FAIL: graph server Status %d %s : %s" % (errcode, errmsg, url)
|
|
except Exception, e:
|
|
print "FAIL: graph server ", e.__class__, e, url
|
|
return found
|
|
|
|
def post_multipart(host, selector, fields, files):
|
|
"""
|
|
Post fields and files to an http host as multipart/form-data.
|
|
fields is a sequence of (name, value) elements for regular form fields.
|
|
files is a sequence of (name, filename, value) elements for data to be uploaded as files
|
|
Return the server's response page.
|
|
"""
|
|
try:
|
|
host = host.replace('http://', '')
|
|
index = host.find('/')
|
|
if index > 0:
|
|
selector = '/'.join([host[index:], selector.lstrip('/')])
|
|
host = host[0:index]
|
|
content_type, body = encode_multipart_formdata(fields, files)
|
|
h = httplib.HTTP(host)
|
|
h.putrequest('POST', selector)
|
|
h.putheader('content-type', content_type)
|
|
h.putheader('content-length', str(len(body)))
|
|
h.putheader('Host', host)
|
|
h.endheaders()
|
|
h.send(body)
|
|
errcode, errmsg, headers = h.getreply()
|
|
return h.file.read()
|
|
except (httplib.HTTPException, error, herror, gaierror, timeout), e:
|
|
print "FAIL: graph server unreachable"
|
|
print "FAIL: " + str(e)
|
|
raise
|
|
except:
|
|
print "FAIL: graph server unreachable"
|
|
raise
|
|
|
|
def encode_multipart_formdata(fields, files):
|
|
"""
|
|
fields is a sequence of (name, value) elements for regular form fields.
|
|
files is a sequence of (name, filename, value) elements for data to be uploaded as files
|
|
Return (content_type, body) ready for httplib.HTTP instance
|
|
"""
|
|
BOUNDARY = '----------ThIs_Is_tHe_bouNdaRY_$'
|
|
CRLF = '\r\n'
|
|
L = []
|
|
for (key, value) in fields:
|
|
L.append('--' + BOUNDARY)
|
|
L.append('Content-Disposition: form-data; name="%s"' % key)
|
|
L.append('')
|
|
L.append(value)
|
|
for (key, filename, value) in files:
|
|
L.append('--' + BOUNDARY)
|
|
L.append('Content-Disposition: form-data; name="%s"; filename="%s"' % (key, filename))
|
|
L.append('Content-Type: %s' % get_content_type(filename))
|
|
L.append('')
|
|
L.append(value)
|
|
L.append('--' + BOUNDARY + '--')
|
|
L.append('')
|
|
body = CRLF.join(L)
|
|
content_type = 'multipart/form-data; boundary=%s' % BOUNDARY
|
|
return content_type, body
|
|
|
|
def get_content_type(filename):
|
|
return mimetypes.guess_type(filename)[0] or 'application/octet-stream'
|