YCast/ycast/my_stations.py

import logging
import hashlib

import yaml

import ycast.vtuner as vtuner
import ycast.generic as generic

ID_PREFIX = "MY"

config_file = 'my_stations.yml'


class Station:
    def __init__(self, uid, name, url, category):
        self.id = generic.generate_stationid_with_prefix(uid, ID_PREFIX)
        self.name = name
        self.url = url
        self.tag = category

    def to_vtuner(self):
        return vtuner.Station(self.id, self.name, self.tag, self.url, None, self.tag, None, None, None, None)


def set_config(config):
    global config_file
    if config:
        config_file = config
    if get_stations_yaml():
        return True
    else:
        return False


def get_station_by_id(uid):
    my_stations_yaml = get_stations_yaml()
    if my_stations_yaml:
        for category in my_stations_yaml:
            for station in get_stations_by_category(category):
                if uid == generic.get_stationid_without_prefix(station.id):
                    return station
    return None


def get_stations_yaml():
    try:
        with open(config_file, 'r') as f:
            my_stations = yaml.safe_load(f)
    except FileNotFoundError:
        logging.error("Station configuration '%s' not found", config_file)
        return None
    except yaml.YAMLError as e:
        logging.error("Station configuration format error: %s", e)
        return None
    return my_stations


def get_category_directories():
    my_stations_yaml = get_stations_yaml()
    categories = []
    if my_stations_yaml:
        for category in my_stations_yaml:
            categories.append(generic.Directory(category, len(get_stations_by_category(category))))
    return categories


def get_stations_by_category(category):
    my_stations_yaml = get_stations_yaml()
    stations = []
    if my_stations_yaml and category in my_stations_yaml:
        for station_name in my_stations_yaml[category]:
            station_url = my_stations_yaml[category][station_name]
            station_id = str(get_checksum(station_name + station_url)).upper()
            stations.append(Station(station_id, station_name, station_url, category))
    return stations


def get_checksum(feed):
    hash_feed = feed.encode()
    hash_object = hashlib.sha256(hash_feed)
    return hash_object.hexdigest()