mirror of
https://github.com/phin05/discord-rich-presence-plex
synced 2024-11-21 17:13:04 +00:00
31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
from .config import config
|
|
from PIL import Image
|
|
from typing import Optional
|
|
from utils.logging import logger
|
|
import io
|
|
import models.imgur
|
|
import requests
|
|
|
|
def uploadToImgur(url: str) -> Optional[str]:
|
|
try:
|
|
originalImageBytesIO = io.BytesIO(requests.get(url).content)
|
|
originalImage = Image.open(originalImageBytesIO).convert("RGB")
|
|
newImage = Image.new("RGB", originalImage.size)
|
|
newImage.putdata(originalImage.getdata()) # pyright: ignore[reportArgumentType]
|
|
maxSize = config["display"]["posters"]["maxSize"]
|
|
if maxSize:
|
|
newImage.thumbnail((maxSize, maxSize))
|
|
newImageBytesIO = io.BytesIO()
|
|
newImage.save(newImageBytesIO, subsampling = 0, quality = 90, format = "JPEG")
|
|
response = requests.post(
|
|
"https://api.imgur.com/3/image",
|
|
headers = { "Authorization": f"Client-ID {config['display']['posters']['imgurClientID']}" },
|
|
files = { "image": newImageBytesIO.getvalue() }
|
|
)
|
|
logger.debug("HTTP %d, %s, %s", response.status_code, response.headers, response.text.strip())
|
|
data: models.imgur.UploadResponse = response.json()
|
|
if not data["success"]:
|
|
raise Exception(data["data"]["error"])
|
|
return data["data"]["link"]
|
|
except:
|
|
logger.exception("An unexpected error occured while uploading an image to Imgur")
|