IGN:Proper error messages when mimagemagick fails to load. Also workaround occasional IOError when sendingfiles to devices on linux

This commit is contained in:
Kovid Goyal 2009-08-14 11:47:05 -06:00
parent 00c32f1a2a
commit 207a9728c0
2 changed files with 29 additions and 15 deletions

View File

@ -24,6 +24,12 @@ class File(object):
self.name = os.path.basename(path)
def check_transfer(infile, dest):
infile.seek(0)
dest.seek(0)
return infile.read() == dest.read()
class CLI(object):
def get_file(self, path, outfile, end_session=True):
@ -45,7 +51,15 @@ class CLI(object):
if not os.path.exists(d):
os.makedirs(d)
with open(path, 'wb') as dest:
shutil.copyfileobj(infile, dest)
try:
shutil.copyfileobj(infile, dest)
except IOError:
print 'WARNING: First attempt to send file to device failed'
infile.seek(0)
dest.seek(0)
dest.truncate()
shutil.copyfileobj(infile, dest)
#if not check_transfer(infile, dest): raise Exception('Transfer failed')
if close:
infile.close()

View File

@ -62,7 +62,7 @@ and save it to a new file.
>>> MagickWriteImage(wand,"out.png") #doctest: +ELLIPSIS
<MagickBooleanType object at ...>
>>>
>>>
"""
@ -82,14 +82,15 @@ else:
_lib = util.find_library('MagickWand')
if _lib is None:
_lib = util.find_library('Wand')
_magick = _magick_error = None
try:
_magick = ctypes.CDLL(_lib)
except Exception, err:
_magick_erorr = str(err)
global _magick_error
_magick_error = str(err)
_initialized = False
_initialized = False
def initialize():
global _initialized
if not _initialized:
@ -97,22 +98,22 @@ def initialize():
_magick.MagickWandGenesis()
else:
raise RuntimeError('Failed to load ImageMagick: %s'%_magick_error)
def finalize():
global _initialized
if _initialized:
_magick.MagickWandTerminus()
_initialized = False
class ImageMagick(object):
def __enter__(self):
initialize()
def __exit__(self, *args):
finalize()
class MetricType(ctypes.c_int): pass
UndefinedMetric = MetricType(0)
AbsoluteErrorMetric = MetricType(1)
@ -489,7 +490,7 @@ SetEvaluateOperator = MagickEvaluateOperator(10)
SubtractEvaluateOperator = MagickEvaluateOperator(11)
XorEvaluateOperator = MagickEvaluateOperator(12)
class ExceptionType(ctypes.c_int):
class ExceptionType(ctypes.c_int):
pass
UndefinedException = ExceptionType(0)
@ -4360,9 +4361,8 @@ try:
except AttributeError,e:
pass
else:
PixelSetColor = _magick.PixelSetColor
PixelSetColor = _magick.PixelSetColor
if __name__=='__main__':
import doctest
doctest.testmod()