cli/ytrssil/config.py

import json
import os
from dataclasses import dataclass
from typing import Literal


@dataclass
class Configuration:
    username: str
    password: str
    api_url: str = 'https://ytrssil.theedgeofrage.com'
    max_res: Literal['480', '720', '1080', '1440', '2160'] = '1440'

    @property
    def mpv_options(self) -> list[str]:
        return [
            '--no-terminal',
            f'--ytdl-format=bestvideo[height<=?{self.max_res}]+bestaudio/best',
        ]


def load_config() -> Configuration:
    config_prefix: str
    try:
        config_prefix = os.environ['XDG_CONFIG_HOME']
    except KeyError:
        config_prefix = os.path.expanduser('~/.config')

    config_path: str = os.path.join(config_prefix, 'ytrssil', 'config.json')
    with open(config_path) as f:
        config_data = json.load(f)

    return Configuration(**config_data)