diff options
Diffstat (limited to 'DeDRM_plugin/__init__.py')
-rw-r--r-- | DeDRM_plugin/__init__.py | 25 |
1 files changed, 2 insertions, 23 deletions
diff --git a/DeDRM_plugin/__init__.py b/DeDRM_plugin/__init__.py index ba9e2ec..eaf505a 100644 --- a/DeDRM_plugin/__init__.py +++ b/DeDRM_plugin/__init__.py @@ -134,29 +134,8 @@ except: config_dir = "" -# Wrap a stream so that output gets flushed immediately -# and also make sure that any unicode strings get safely -# encoded using "replace" before writing them. -class SafeUnbuffered: - def __init__(self, stream): - self.stream = stream - self.encoding = stream.encoding - if self.encoding == None: - self.encoding = "utf-8" - def write(self, data): - if isinstance(data,str) or isinstance(data,unicode): - # str for Python3, unicode for Python2 - data = data.encode(self.encoding,"replace") - try: - buffer = getattr(self.stream, 'buffer', self.stream) - # self.stream.buffer for Python3, self.stream for Python2 - buffer.write(data) - buffer.flush() - except: - # We can do nothing if a write fails - raise - def __getattr__(self, attr): - return getattr(self.stream, attr) +from utilities import SafeUnbuffered + PLUGIN_NAME = __version.PLUGIN_NAME PLUGIN_VERSION = __version.PLUGIN_VERSION |