mirror of
https://github.com/ChristianLight/tutor.git
synced 2024-05-29 20:30:48 +00:00
0a670d7ead
Annotations were generated with pyannotate: https://github.com/dropbox/pyannotate We are running in strict mode, which is awesome! This affects a large part of the code base, which might be an issue for people running a fork of Tutor. Nonetheless, the behavior should not be affected. If anything, this process has helped find and resolve a few type-related bugs. Thus, this is not considered as a breaking change.
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import re
|
|
from typing import cast, Any, Dict, IO, Iterator, Tuple, Union
|
|
|
|
import yaml
|
|
from _io import TextIOWrapper
|
|
from yaml.parser import ParserError
|
|
from yaml.scanner import ScannerError
|
|
|
|
import click
|
|
|
|
|
|
def load(stream: Union[str, IO[str]]) -> Dict[str, str]:
|
|
return cast(Dict[str, str], yaml.load(stream, Loader=yaml.SafeLoader))
|
|
|
|
|
|
def load_all(stream: str) -> Iterator[Any]:
|
|
return yaml.load_all(stream, Loader=yaml.SafeLoader)
|
|
|
|
|
|
def dump(content: Dict[str, str], fileobj: TextIOWrapper) -> None:
|
|
yaml.dump(content, stream=fileobj, default_flow_style=False)
|
|
|
|
|
|
def parse(v: Union[str, IO[str]]) -> Any:
|
|
"""
|
|
Parse a yaml-formatted string.
|
|
"""
|
|
try:
|
|
return load(v)
|
|
except (ParserError, ScannerError):
|
|
pass
|
|
return v
|
|
|
|
|
|
class YamlParamType(click.ParamType):
|
|
name = "yaml"
|
|
PARAM_REGEXP = r"(?P<key>[a-zA-Z0-9_-]+)=(?P<value>.*)"
|
|
|
|
def convert(self, value: str, param: Any, ctx: Any) -> Tuple[str, Any]:
|
|
match = re.match(self.PARAM_REGEXP, value)
|
|
if not match:
|
|
self.fail("'{}' is not of the form 'key=value'.".format(value), param, ctx)
|
|
key = match.groupdict()["key"]
|
|
value = match.groupdict()["value"]
|
|
if not value:
|
|
# Empty strings are interpreted as null values, which is incorrect.
|
|
value = "''"
|
|
return key, parse(value)
|