1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
from PIL import Image, UnidentifiedImageError
def get_image_background_color(img, img_format: str):
has_alpha = img_format in {"hexa", "rgba"}
img = img.convert("RGBA" if has_alpha else "RGB")
pixel_color = img.getpixel((1, 1))
if img_format in {"hex", "hexa"}:
color_format = "#" + "%02x" * len(pixel_color)
color = color_format % pixel_color
color = color.upper()
elif img_format in {"rgb", "rgba"}:
if has_alpha:
# Normalize alpha channel to be between 0 and 1
pixel_color = (
*pixel_color[:3],
round(pixel_color[3] / 255, 2),
)
# Should look like `rgb(1, 2, 3) or rgba(1, 2, 3, 1.0)
color = f"{img_format}{pixel_color}"
else: # pragma: no cover
raise NotImplementedError(f"Unsupported color format: {img_format}")
return color
def get_image_file_background_color(img_file, img_format: str):
color = ""
try:
with Image.open(img_file) as image:
color = get_image_background_color(image, img_format)
except UnidentifiedImageError:
pass
return color
|
