Allow passing in a QImage to save_cover_data_to()

This commit is contained in:
Kovid Goyal 2016-06-19 18:08:07 +05:30
parent 3b3b0795bb
commit edcc38fbb5

View File

@ -155,9 +155,13 @@ def save_cover_data_to(data, path=None, bgcolor='#ffffff', resize_to=None, compr
The image will be resized to fit into this target size. If None the The image will be resized to fit into this target size. If None the
value from the tweak is used. value from the tweak is used.
''' '''
img, fmt = image_and_format_from_data(data)
orig_fmt = normalize_format_name(fmt)
fmt = normalize_format_name(data_fmt if path is None else os.path.splitext(path)[1][1:]) fmt = normalize_format_name(data_fmt if path is None else os.path.splitext(path)[1][1:])
if isinstance(data, QImage):
img = data
changed = True
else:
img, orig_fmt = image_and_format_from_data(data)
orig_fmt = normalize_format_name(orig_fmt)
changed = fmt != orig_fmt changed = fmt != orig_fmt
if resize_to is not None: if resize_to is not None:
changed = True changed = True