import logging import httplib import traceback import string import errno from socket import error as socket_error ############################################################################### log = logging.getLogger("PLEX."+__name__) ############################################################################### class RequestMgr: def __init__(self): self.conns = {} def getConnection(self, protocol, host, port): conn = self.conns.get(protocol+host+str(port), False) if not conn: if protocol == "https": conn = httplib.HTTPSConnection(host, port) else: conn = httplib.HTTPConnection(host, port) self.conns[protocol+host+str(port)] = conn return conn def closeConnection(self, protocol, host, port): conn = self.conns.get(protocol+host+str(port), False) if conn: conn.close() self.conns.pop(protocol+host+str(port), None) def dumpConnections(self): for conn in self.conns.values(): conn.close() self.conns = {} def post(self, host, port, path, body, header={}, protocol="http"): conn = None try: conn = self.getConnection(protocol, host, port) header['Connection'] = "keep-alive" conn.request("POST", path, body, header) data = conn.getresponse() if int(data.status) >= 400: log.error("HTTP response error: %s" % str(data.status)) # this should return false, but I'm hacking it since iOS # returns 404 no matter what return data.read() or True else: return data.read() or True except socket_error as serr: # Ignore remote close and connection refused (e.g. shutdown PKC) if serr.errno in (errno.WSAECONNABORTED, errno.WSAECONNREFUSED): pass else: log.error("Unable to connect to %s\nReason:" % host) log.error(traceback.print_exc()) self.conns.pop(protocol+host+str(port), None) if conn: conn.close() return False except Exception as e: log.error("Exception encountered: %s" % e) # Close connection just in case try: conn.close() except: pass return False def getwithparams(self, host, port, path, params, header={}, protocol="http"): newpath = path + '?' pairs = [] for key in params: pairs.append(str(key)+'='+str(params[key])) newpath += string.join(pairs, '&') return self.get(host, port, newpath, header, protocol) def get(self, host, port, path, header={}, protocol="http"): try: conn = self.getConnection(protocol, host, port) header['Connection'] = "keep-alive" conn.request("GET", path, headers=header) data = conn.getresponse() if int(data.status) >= 400: log.error("HTTP response error: %s" % str(data.status)) return False else: return data.read() or True except socket_error as serr: # Ignore remote close and connection refused (e.g. shutdown PKC) if serr.errno in (errno.WSAECONNABORTED, errno.WSAECONNREFUSED): pass else: log.error("Unable to connect to %s\nReason:" % host) log.error(traceback.print_exc()) self.conns.pop(protocol+host+str(port), None) conn.close() return False