pferd/PFERD/config.py

160 lines
4.6 KiB
Python
Raw Normal View History

2021-04-27 12:41:49 +02:00
import os
2021-05-05 23:45:10 +02:00
from configparser import ConfigParser, SectionProxy
from dataclasses import dataclass
2021-04-27 12:41:49 +02:00
from pathlib import Path
2021-05-05 23:45:10 +02:00
from typing import Any, List, NoReturn, Optional, Tuple
2021-04-27 12:41:49 +02:00
from .utils import prompt_yes_no
class ConfigLoadException(Exception):
pass
class ConfigDumpException(Exception):
pass
2021-05-05 23:45:10 +02:00
@dataclass
class ConfigFormatException(Exception):
section: str
key: str
desc: str
class Section:
"""
Base class for the crawler and auth section classes.
"""
2021-05-05 23:45:10 +02:00
def __init__(self, section: SectionProxy):
self.s = section
def error(self, key: str, desc: str) -> NoReturn:
raise ConfigFormatException(self.s.name, key, desc)
def invalid_value(
self,
key: str,
value: Any,
reason: Optional[str],
) -> NoReturn:
if reason is None:
self.error(key, f"Invalid value {value!r}")
else:
self.error(key, f"Invalid value {value!r}: {reason}")
2021-05-05 23:45:10 +02:00
def missing_value(self, key: str) -> NoReturn:
self.error(key, "Missing value")
2021-04-27 12:41:49 +02:00
class Config:
@staticmethod
def _default_path() -> Path:
if os.name == "posix":
return Path("~/.config/PFERD/pferd.cfg").expanduser()
elif os.name == "nt":
return Path("~/AppData/Roaming/PFERD/pferd.cfg").expanduser()
else:
return Path("~/.pferd.cfg").expanduser()
2021-05-05 23:45:10 +02:00
def __init__(self, parser: ConfigParser):
2021-04-27 12:41:49 +02:00
self._parser = parser
@staticmethod
def _fail_load(path: Path, reason: str) -> None:
print(f"Failed to load config file at {path}")
print(f"Reason: {reason}")
raise ConfigLoadException()
@staticmethod
2021-05-05 23:45:10 +02:00
def load_parser(path: Optional[Path] = None) -> ConfigParser:
2021-04-27 12:41:49 +02:00
"""
May throw a ConfigLoadException.
"""
if not path:
path = Config._default_path()
2021-05-05 23:45:10 +02:00
parser = ConfigParser()
2021-04-27 12:41:49 +02:00
# Using config.read_file instead of config.read because config.read
# would just ignore a missing file and carry on.
try:
with open(path) as f:
parser.read_file(f, source=str(path))
except FileNotFoundError:
Config._fail_load(path, "File does not exist")
except IsADirectoryError:
Config._fail_load(path, "That's a directory, not a file")
except PermissionError:
Config._fail_load(path, "Insufficient permissions")
return parser
@staticmethod
def _fail_dump(path: Path, reason: str) -> None:
print(f"Failed to dump config file to {path}")
print(f"Reason: {reason}")
raise ConfigDumpException()
def dump(self, path: Optional[Path] = None) -> None:
"""
May throw a ConfigDumpException.
"""
if not path:
path = self._default_path()
print(f"Dumping config to {path}")
try:
path.parent.mkdir(parents=True, exist_ok=True)
except PermissionError:
self._fail_dump(path, "Could not create parent directory")
try:
# Ensuring we don't accidentally overwrite any existing files by
# always asking before overwriting a file.
try:
# x = open for exclusive creation, failing if the file already
# exists
with open(path, "x") as f:
self._parser.write(f)
except FileExistsError:
print("That file already exists.")
if prompt_yes_no("Overwrite it?", default=False):
with open(path, "w") as f:
self._parser.write(f)
else:
self._fail_dump(path, "File already exists")
except IsADirectoryError:
self._fail_dump(path, "That's a directory, not a file")
except PermissionError:
self._fail_dump(path, "Insufficient permissions")
2021-04-30 16:22:14 +02:00
@property
2021-05-05 23:45:10 +02:00
def default_section(self) -> SectionProxy:
return self._parser[self._parser.default_section]
2021-04-30 16:22:14 +02:00
2021-05-05 23:45:10 +02:00
def crawler_sections(self) -> List[Tuple[str, SectionProxy]]:
2021-04-30 16:22:14 +02:00
result = []
for name, proxy in self._parser.items():
if name.startswith("crawler:"):
result.append((name, proxy))
2021-04-30 16:22:14 +02:00
return result
def authenticator_sections(self) -> List[Tuple[str, SectionProxy]]:
result = []
for name, proxy in self._parser.items():
if name.startswith("auth:"):
result.append((name, proxy))
return result
2021-04-30 16:22:14 +02:00
@property
def working_dir(self) -> Path:
pathstr = self.default_section.get("working_dir", ".")
return Path(pathstr).expanduser()