50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
import logging
|
|
import requests
|
|
import io
|
|
import os
|
|
|
|
from PIL import Image
|
|
|
|
import ycast.generic as generic
|
|
from ycast import __version__
|
|
|
|
MAX_SIZE = 290
|
|
CACHE_NAME = 'icons'
|
|
|
|
|
|
def get_icon(station):
|
|
cache_path = generic.get_cache_path(CACHE_NAME)
|
|
if not cache_path:
|
|
return None
|
|
station_icon_file = cache_path + '/' + station.id
|
|
if not os.path.exists(station_icon_file):
|
|
logging.debug("Station icon cache miss. Fetching and converting station icon for station id '%s'", station.id)
|
|
headers = {'User-Agent': generic.USER_AGENT + '/' + __version__}
|
|
try:
|
|
response = requests.get(station.icon, headers=headers)
|
|
except requests.exceptions.ConnectionError as err:
|
|
logging.error("Connection to station icon URL failed (%s)", err)
|
|
return None
|
|
if response.status_code != 200:
|
|
logging.error("Could not get station icon data from %s (HTML status %s)", station.icon, response.status_code)
|
|
return None
|
|
try:
|
|
image = Image.open(io.BytesIO(response.content))
|
|
image = image.convert("RGB")
|
|
if image.size[0] > image.size[1]:
|
|
ratio = MAX_SIZE / image.size[0]
|
|
else:
|
|
ratio = MAX_SIZE / image.size[1]
|
|
image = image.resize((int(image.size[0] * ratio), int(image.size[1] * ratio)), Image.ANTIALIAS)
|
|
image.save(station_icon_file, format="JPEG")
|
|
except Exception as e:
|
|
logging.error("Station icon conversion error (%s)", e)
|
|
return None
|
|
try:
|
|
with open(station_icon_file, 'rb') as file:
|
|
image_conv = file.read()
|
|
except PermissionError:
|
|
logging.error("Could not access station icon file in cache (%s) because of access permissions",
|
|
station_icon_file)
|
|
return None
|
|
return image_conv
|