mirror of
https://github.com/DragoonAethis/itch-dl.git
synced 2024-12-30 15:01:50 +01:00
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
import os
|
|
import json
|
|
import logging
|
|
import platform
|
|
from typing import Optional
|
|
|
|
import requests
|
|
from pydantic import BaseModel
|
|
|
|
from . import __version__
|
|
|
|
|
|
class Settings(BaseModel):
|
|
"""Available settings for itch-dl. Make sure all of them
|
|
have default values, as the config file may not exist."""
|
|
api_key: Optional[str] = None
|
|
user_agent: str = f"python-requests/{requests.__version__} itch-dl/{__version__}"
|
|
|
|
|
|
def create_and_get_config_path() -> str:
|
|
"""Returns the configuration directory in the appropriate
|
|
location for the current OS. The directory may not exist."""
|
|
system = platform.system()
|
|
if system == "Linux":
|
|
base_path = os.environ.get('XDG_CONFIG_HOME') or os.path.expanduser('~/.config/')
|
|
elif system == "Darwin":
|
|
base_path = os.path.expanduser('~/Library/Application Support/')
|
|
elif system == "Windows":
|
|
base_path = os.environ.get('APPDATA') or os.path.expanduser('~/AppData/Roaming/')
|
|
else:
|
|
raise NotImplementedError(f"Unknown platform: {system}")
|
|
|
|
return os.path.join(base_path, "itch-dl")
|
|
|
|
|
|
def load_config(profile: Optional[str] = None) -> Settings:
|
|
"""Loads the configuration from the file system if it exists,
|
|
the returns a Settings object."""
|
|
config_path = create_and_get_config_path()
|
|
config_file_path = os.path.join(config_path, "config.json")
|
|
profile_file_path = os.path.join(config_path, "profiles", profile or "")
|
|
|
|
if os.path.isfile(config_file_path):
|
|
logging.debug(f"Found config file: {config_file_path}")
|
|
with open(config_file_path) as f:
|
|
config_data = json.load(f)
|
|
else:
|
|
config_data = {}
|
|
|
|
if os.path.isfile(profile_file_path):
|
|
logging.debug(f"Found profile: {profile_file_path}")
|
|
with open(config_file_path) as f:
|
|
profile_data = json.load(f)
|
|
|
|
config_data.update(profile_data)
|
|
|
|
return Settings(**config_data)
|