cli/ytrssil/config.py

34 lines
855 B
Python

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)