node_helpers.py 691 Bytes
Newer Older
1
from PIL import Image, ImageFile, UnidentifiedImageError
2
3
4
5
6
7
8
9
10
11

def conditioning_set_values(conditioning, values={}):
    c = []
    for t in conditioning:
        n = [t[0], t[1].copy()]
        for k in values:
            n[1][k] = values[k]
        c.append(n)

    return c
12
13

def open_image(path):
14
15
16
    prev_value = None

    try:
17
        img = Image.open(path)
18
19
    except (UnidentifiedImageError, ValueError): #PIL issues #4472 and #2445
        prev_value = ImageFile.LOAD_TRUNCATED_IMAGES
20
21
22
        ImageFile.LOAD_TRUNCATED_IMAGES = True
        img = Image.open(path)
    finally:
23
24
        if prev_value is not None:
            ImageFile.LOAD_TRUNCATED_IMAGES = prev_value
25
        return img