BonHowi/discordPogmareBot

View on GitHub
modules/utils.py

Summary

Maintainability
A
0 mins
Test Coverage
from io import BytesIO

import requests
from PIL import Image


# Get dominant color from image
def get_dominant_color(url) -> int:
    response = requests.get(url)
    pil_img = Image.open(BytesIO(response.content))
    img = pil_img.copy()
    img.convert("RGB")
    img.resize((1, 1), resample=0)
    dominant_color = img.getpixel((0, 0))
    try:
        return_color = int('%02x%02x%02x' % (dominant_color[0], dominant_color[1], dominant_color[2]), 16)
    except TypeError:
        return_color = 0
    return return_color