mirror of
https://github.com/kovidgoyal/calibre.git
synced 2025-07-08 10:44:09 -04:00
Remove unused code
This commit is contained in:
parent
dbcb63c290
commit
20b9cc3a81
@ -1,168 +0,0 @@
|
|||||||
#!/usr/bin/env python2
|
|
||||||
# vim:fileencoding=UTF-8:ts=4:sw=4:sta:et:sts=4:ai
|
|
||||||
from __future__ import (unicode_literals, division, absolute_import,
|
|
||||||
print_function)
|
|
||||||
|
|
||||||
__license__ = 'GPL v3'
|
|
||||||
__copyright__ = '2011, Kovid Goyal <kovid@kovidgoyal.net>'
|
|
||||||
__docformat__ = 'restructuredtext en'
|
|
||||||
|
|
||||||
import os, cPickle, struct
|
|
||||||
from threading import Thread
|
|
||||||
from Queue import Queue, Empty
|
|
||||||
from multiprocessing.connection import arbitrary_address, Listener
|
|
||||||
from functools import partial
|
|
||||||
|
|
||||||
from calibre import as_unicode, prints
|
|
||||||
from calibre.constants import iswindows, DEBUG
|
|
||||||
from calibre.utils.ipc import eintr_retry_call
|
|
||||||
from polyglot.builtins import unicode_type
|
|
||||||
|
|
||||||
|
|
||||||
def _encode(msg):
|
|
||||||
raw = cPickle.dumps(msg, -1)
|
|
||||||
size = len(raw)
|
|
||||||
header = struct.pack('!Q', size)
|
|
||||||
return header + raw
|
|
||||||
|
|
||||||
|
|
||||||
def _decode(raw):
|
|
||||||
sz = struct.calcsize('!Q')
|
|
||||||
if len(raw) < sz:
|
|
||||||
return 'invalid', None
|
|
||||||
header, = struct.unpack('!Q', raw[:sz])
|
|
||||||
if len(raw) != sz + header or header == 0:
|
|
||||||
return 'invalid', None
|
|
||||||
return cPickle.loads(raw[sz:])
|
|
||||||
|
|
||||||
|
|
||||||
class Writer(Thread):
|
|
||||||
|
|
||||||
TIMEOUT = 60 # seconds
|
|
||||||
|
|
||||||
def __init__(self, conn):
|
|
||||||
Thread.__init__(self)
|
|
||||||
self.daemon = True
|
|
||||||
self.dataq, self.resultq = Queue(), Queue()
|
|
||||||
self.conn = conn
|
|
||||||
self.start()
|
|
||||||
self.data_written = False
|
|
||||||
|
|
||||||
def close(self):
|
|
||||||
self.dataq.put(None)
|
|
||||||
|
|
||||||
def flush(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def write(self, raw_data):
|
|
||||||
self.dataq.put(raw_data)
|
|
||||||
|
|
||||||
try:
|
|
||||||
ex = self.resultq.get(True, self.TIMEOUT)
|
|
||||||
except Empty:
|
|
||||||
raise IOError('Writing to socket timed out')
|
|
||||||
else:
|
|
||||||
if ex is not None:
|
|
||||||
raise IOError('Writing to socket failed with error: %s' % ex)
|
|
||||||
|
|
||||||
def run(self):
|
|
||||||
while True:
|
|
||||||
x = self.dataq.get()
|
|
||||||
if x is None:
|
|
||||||
break
|
|
||||||
try:
|
|
||||||
self.data_written = True
|
|
||||||
eintr_retry_call(self.conn.send_bytes, x)
|
|
||||||
except Exception as e:
|
|
||||||
self.resultq.put(as_unicode(e))
|
|
||||||
else:
|
|
||||||
self.resultq.put(None)
|
|
||||||
|
|
||||||
def __enter__(self):
|
|
||||||
return self
|
|
||||||
|
|
||||||
def __exit__(self, exc_type, exc_val, exc_tb):
|
|
||||||
self.close()
|
|
||||||
|
|
||||||
|
|
||||||
class Server(Thread):
|
|
||||||
|
|
||||||
def __init__(self, dispatcher):
|
|
||||||
Thread.__init__(self)
|
|
||||||
self.daemon = True
|
|
||||||
|
|
||||||
self.auth_key = os.urandom(32)
|
|
||||||
self.address = arbitrary_address('AF_PIPE' if iswindows else 'AF_UNIX')
|
|
||||||
if iswindows and self.address[1] == ':':
|
|
||||||
self.address = self.address[2:]
|
|
||||||
self.listener = Listener(address=self.address,
|
|
||||||
authkey=self.auth_key, backlog=4)
|
|
||||||
|
|
||||||
self.keep_going = True
|
|
||||||
self.dispatcher = dispatcher
|
|
||||||
|
|
||||||
@property
|
|
||||||
def connection_information(self):
|
|
||||||
if not self.is_alive():
|
|
||||||
self.start()
|
|
||||||
return (self.address, self.auth_key)
|
|
||||||
|
|
||||||
def stop(self):
|
|
||||||
self.keep_going = False
|
|
||||||
try:
|
|
||||||
self.listener.close()
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
|
|
||||||
def run(self):
|
|
||||||
while self.keep_going:
|
|
||||||
try:
|
|
||||||
conn = eintr_retry_call(self.listener.accept)
|
|
||||||
self.handle_client(conn)
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
|
|
||||||
def handle_client(self, conn):
|
|
||||||
t = Thread(target=partial(self._handle_client, conn))
|
|
||||||
t.daemon = True
|
|
||||||
t.start()
|
|
||||||
|
|
||||||
def _handle_client(self, conn):
|
|
||||||
while True:
|
|
||||||
try:
|
|
||||||
func_name, args, kwargs = eintr_retry_call(conn.recv)
|
|
||||||
except EOFError:
|
|
||||||
try:
|
|
||||||
conn.close()
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
return
|
|
||||||
else:
|
|
||||||
try:
|
|
||||||
self.call_func(func_name, args, kwargs, conn)
|
|
||||||
except:
|
|
||||||
try:
|
|
||||||
conn.close()
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
prints('Proxy function: %s with args: %r and'
|
|
||||||
' kwargs: %r failed')
|
|
||||||
if DEBUG:
|
|
||||||
import traceback
|
|
||||||
traceback.print_exc()
|
|
||||||
break
|
|
||||||
|
|
||||||
def call_func(self, func_name, args, kwargs, conn):
|
|
||||||
with Writer(conn) as f:
|
|
||||||
try:
|
|
||||||
self.dispatcher(f, func_name, args, kwargs)
|
|
||||||
except Exception as e:
|
|
||||||
if not f.data_written:
|
|
||||||
import traceback
|
|
||||||
# Try to tell the client process what error happened
|
|
||||||
try:
|
|
||||||
eintr_retry_call(conn.send_bytes, (_encode(('failed', (unicode_type(e),
|
|
||||||
as_unicode(traceback.format_exc()))))))
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
raise
|
|
Loading…
x
Reference in New Issue
Block a user