Initial import

ebook_converter/utils/ipc/__init__.py

@@ -0,0 +1,83 @@
+#!/usr/bin/env python2
+# vim:fileencoding=UTF-8:ts=4:sw=4:sta:et:sts=4:ai
+from __future__ import absolute_import, division, print_function, unicode_literals
+
+__license__   = 'GPL v3'
+__copyright__ = '2009, Kovid Goyal <kovid@kovidgoyal.net>'
+__docformat__ = 'restructuredtext en'
+
+import os, errno, sys
+from threading import Thread
+
+from calibre import force_unicode
+from calibre.constants import iswindows, get_windows_username, islinux, filesystem_encoding, ispy3
+from calibre.utils.filenames import ascii_filename
+from polyglot.functools import lru_cache
+
+VADDRESS = None
+
+
+def eintr_retry_call(func, *args, **kwargs):
+    while True:
+        try:
+            return func(*args, **kwargs)
+        except EnvironmentError as e:
+            if getattr(e, 'errno', None) == errno.EINTR:
+                continue
+            raise
+
+
+@lru_cache()
+def socket_address(which):
+    if iswindows:
+        ans = r'\\.\pipe\Calibre' + which
+        try:
+            user = get_windows_username()
+        except Exception:
+            user = None
+        if user:
+            user = ascii_filename(user).replace(' ', '_')
+            if user:
+                ans += '-' + user[:100] + 'x'
+    else:
+        user = force_unicode(os.environ.get('USER') or os.path.basename(os.path.expanduser('~')), filesystem_encoding)
+        sock_name = '{}-calibre-{}.socket'.format(ascii_filename(user).replace(' ', '_'), which)
+        if islinux:
+            ans = '\0' + sock_name
+        else:
+            from tempfile import gettempdir
+            tmp = force_unicode(gettempdir(), filesystem_encoding)
+            ans = os.path.join(tmp, sock_name)
+    if not ispy3 and not isinstance(ans, bytes):
+        ans = ans.encode(filesystem_encoding)
+    return ans
+
+
+def gui_socket_address():
+    return socket_address('GUI' if iswindows else 'gui')
+
+
+def viewer_socket_address():
+    return socket_address('Viewer' if iswindows else 'viewer')
+
+
+class RC(Thread):
+
+    def __init__(self, print_error=True, socket_address=None):
+        self.print_error = print_error
+        self.socket_address = socket_address or gui_socket_address()
+        Thread.__init__(self)
+        self.conn = None
+        self.daemon = True
+
+    def run(self):
+        from multiprocessing.connection import Client
+        self.done = False
+        try:
+            self.conn = Client(self.socket_address)
+            self.done = True
+        except Exception:
+            if self.print_error:
+                print('Failed to connect to address {}', file=sys.stderr).format(repr(self.socket_address))
+                import traceback
+                traceback.print_exc()