def __init__(self, maxconnects, qsize=0, **kwargs):
self._multi = Multi(**kwargs)
self._multi.maxconnects = maxconnects
# TODO: fix bug with pipelining Pool not work
self._multi.pipelining = 0
self._queue = Queue(qsize)
self._size = maxconnects
self._data = {}
# registry curl, sockets, event bitmask
self._registry = {} # key is curl, value is (socket, event bitmask)
def socket_cb(curl, socket, event):
self._registry[curl] = (socket, event)
# TODO: timer
def timer_cb(m, timeout):
if timeout == 0:
code, running = m.socket_action(const.CURL_SOCKET_TIMEOUT, 0)
self._multi.socketfunction = socket_cb
self._multi.timerfunction = timer_cb
super(Pool, self).__init__()
def __init__(self, maxconnects, qsize=0, **kwargs):
self._multi = Multi(**kwargs)
self._multi.maxconnects = maxconnects
# TODO: fix bug with pipelining Pool not work
self._multi.pipelining = 0
self._queue = Queue(qsize)
self._size = maxconnects
self._data = {}
# registry curl, sockets, event bitmask
self._registry = {} # key is curl, value is (socket, event bitmask)
def socket_cb(curl, socket, event):
self._registry[curl] = (socket, event)
# TODO: timer
def timer_cb(m, timeout):
if timeout == 0:
code, running = m.socket_action(const.CURL_SOCKET_TIMEOUT, 0)
self._multi.socketfunction = socket_cb
self._multi.timerfunction = timer_cb
super(Pool, self).__init__()
class Pool(threading.Thread):
"""
Pool work as like Thread and Queue
"""
default_curl_settings = dict(
useragent='Mozilla 6',
autoreferer=1,
followlocation=1,
maxredirs=20,
encoding='',
)
def __init__(self, maxconnects, qsize=0, **kwargs):
self._multi = Multi(**kwargs)
self._multi.maxconnects = maxconnects
# TODO: fix bug with pipelining Pool not work
self._multi.pipelining = 0
self._queue = Queue(qsize)
self._size = maxconnects
self._data = {}
# registry curl, sockets, event bitmask
self._registry = {} # key is curl, value is (socket, event bitmask)
def socket_cb(curl, socket, event):
self._registry[curl] = (socket, event)
# TODO: timer
def timer_cb(m, timeout):
if timeout == 0:
code, running = m.socket_action(const.CURL_SOCKET_TIMEOUT, 0)
self._multi.socketfunction = socket_cb
self._multi.timerfunction = timer_cb
super(Pool, self).__init__()
def _do_socket_action(self, socket, event=0):
code = const.CURLM_CALL_MULTI_SOCKET
while code == const.CURLM_CALL_MULTI_SOCKET:
code, running = self._multi.socket_action(socket, event)
return running
def _do_remove(self):
# remove complete curls and sockets
complete = [(curl, item[0]) for curl, item in self._registry.items()
if item[1] == const.CURL_POLL_REMOVE]
for curl, sock in complete:
self._multi.remove_handle(curl)
# call callback
url, header, body, callback = self._data[curl]
body = ''.join(body)
if self._Thread__started.is_set():
del self._data[curl]
else:
self._data[curl][2] = body
if callback:
callback(curl, url, header, body)
# task done
self._queue.task_done()
removed = len(complete) > 0
if removed:
for curl, socket in complete:
del self._registry[curl]
return removed
def _do_add(self):
start_event = self._Thread__started
added = False
while len(self._multi._handles) < self._size:
block = False
# thread mode and daemon and empty
if self.isDaemon() and start_event.is_set(
) and not self._multi._handles:
block = True
try:
data = self._queue.get(block=block)
except Empty:
break
else:
added = True
url, kwargs, callback = data
c = self._create_curl(url, callback, **kwargs)
self._multi.add_handle(c)
if added:
code = const.CURLM_CALL_MULTI_SOCKET
while code:
code, running = self._multi.socket_action(
const.CURL_SOCKET_TIMEOUT, 0)
return added
def _create_curl(self, url, callback, **kwargs):
settings = self.default_curl_settings.copy()
settings.update(kwargs)
header = []
body = []
def headerfunction(v):
v = v.strip()
if v:
header.append(v)
def writefunction(v):
body.append(v)
settings['headerfunction'] = headerfunction
settings['writefunction'] = writefunction
c = Curl(url, **settings)
self._data[c] = [url, header, body, callback]
return c
def run(self):
"""
for threading mode use "start" method
for non-threading mode use "run" method
"""
start_event = self._Thread__started
self._do_add()
running = 1
do_action = self._do_socket_action
while True:
while running:
r = []
w = []
e = []
for socket, event in self._registry.values():
if event == const.CURL_POLL_IN:
r.append(socket)
elif event == const.CURL_POLL_OUT:
w.append(socket)
elif event == const.CURL_POLL_INOUT:
r.append(socket)
w.append(socket)
rr, ww, ee = select.select(r, w, e, 1.0)
#print self._registry
#print r, w, e
for socket in rr:
running = do_action(socket, const.CURL_CSELECT_IN)
for socket in ww:
running = do_action(socket, const.CURL_CSELECT_OUT)
for socket in ee:
running = do_action(socket, const.CURL_CSELECT_ERR)
self._do_remove()
self._do_add()
# check loop
started = start_event.is_set()
if not started and not self.isDaemon():
break
# for non-threading mode
if not start_event.is_set():
data = [(curl, v[0], v[1], v[2]) for curl, v in self._data.items()]
self._data.clear()
return data
def add(self, url, callback=None, **kwargs):
"""
put values to queue
callback prototype:
def callback(curl, start_url, header, body):
[code]
"""
self._queue.put((url, kwargs, callback))
def join(self, timeout=None):
"""
if thread is daemon then run loop forever
"""
if self.isDaemon():
self._queue.join()
else:
super(Pool, self).join(timeout)
def wrap(multi, timeout, userp):
func(Multi(multi), timeout)
return 0
class Pool(threading.Thread):
"""
Pool work as like Thread and Queue
"""
default_curl_settings = dict(
useragent='Mozilla 6',
autoreferer=1,
followlocation=1,
maxredirs=20,
encoding='',
)
def __init__(self, maxconnects, qsize=0, **kwargs):
self._multi = Multi(**kwargs)
self._multi.maxconnects = maxconnects
# TODO: fix bug with pipelining Pool not work
self._multi.pipelining = 0
self._queue = Queue(qsize)
self._size = maxconnects
self._data = {}
# registry curl, sockets, event bitmask
self._registry = {} # key is curl, value is (socket, event bitmask)
def socket_cb(curl, socket, event):
self._registry[curl] = (socket, event)
# TODO: timer
def timer_cb(m, timeout):
if timeout == 0:
code, running = m.socket_action(const.CURL_SOCKET_TIMEOUT, 0)
self._multi.socketfunction = socket_cb
self._multi.timerfunction = timer_cb
super(Pool, self).__init__()
def _do_socket_action(self, socket, event=0):
code = const.CURLM_CALL_MULTI_SOCKET
while code == const.CURLM_CALL_MULTI_SOCKET:
code, running = self._multi.socket_action(socket, event)
return running
def _do_remove(self):
# remove complete curls and sockets
complete = [(curl, item[0]) for curl, item in self._registry.items() if item[1] == const.CURL_POLL_REMOVE]
for curl, sock in complete:
self._multi.remove_handle(curl)
# call callback
url, header, body, callback = self._data[curl]
body = ''.join(body)
if self._Thread__started.is_set():
del self._data[curl]
else:
self._data[curl][2] = body
if callback:
callback(curl, url, header, body)
# task done
self._queue.task_done()
removed = len(complete) > 0
if removed:
for curl, socket in complete:
del self._registry[curl]
return removed
def _do_add(self):
start_event = self._Thread__started
added = False
while len(self._multi._handles) < self._size:
block = False
# thread mode and daemon and empty
if self.isDaemon() and start_event.is_set() and not self._multi._handles:
block = True
try:
data = self._queue.get(block=block)
except Empty:
break
else:
added = True
url, kwargs, callback = data
c = self._create_curl(url, callback, **kwargs)
self._multi.add_handle(c)
if added:
code = const.CURLM_CALL_MULTI_SOCKET
while code:
code, running = self._multi.socket_action(const.CURL_SOCKET_TIMEOUT, 0)
return added
def _create_curl(self, url, callback, **kwargs):
settings = self.default_curl_settings.copy()
settings.update(kwargs)
header = []
body = []
def headerfunction(v):
v = v.strip()
if v:
header.append(v)
def writefunction(v):
body.append(v)
settings['headerfunction'] = headerfunction
settings['writefunction'] = writefunction
c = Curl(url, **settings)
self._data[c] = [url, header, body, callback]
return c
def run(self):
"""
for threading mode use "start" method
for non-threading mode use "run" method
"""
start_event = self._Thread__started
self._do_add()
running = 1
do_action = self._do_socket_action
while True:
while running:
r = []
w = []
e = []
for socket, event in self._registry.values():
if event == const.CURL_POLL_IN:
r.append(socket)
elif event == const.CURL_POLL_OUT:
w.append(socket)
elif event == const.CURL_POLL_INOUT:
r.append(socket)
w.append(socket)
rr, ww, ee = select.select(r, w, e, 1.0)
#print self._registry
#print r, w, e
for socket in rr:
running = do_action(socket, const.CURL_CSELECT_IN)
for socket in ww:
running = do_action(socket, const.CURL_CSELECT_OUT)
for socket in ee:
running = do_action(socket, const.CURL_CSELECT_ERR)
self._do_remove()
self._do_add()
# check loop
started = start_event.is_set()
if not started and not self.isDaemon():
break
# for non-threading mode
if not start_event.is_set():
data = [(curl, v[0], v[1], v[2])
for curl, v in self._data.items()]
self._data.clear()
return data
def add(self, url, callback=None, **kwargs):
"""
put values to queue
callback prototype:
def callback(curl, start_url, header, body):
[code]
"""
self._queue.put((url, kwargs, callback))
def join(self, timeout=None):
"""
if thread is daemon then run loop forever
"""
if self.isDaemon():
self._queue.join()
else:
super(Pool, self).join(timeout)
