1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
try:
import tomllib
except ImportError:
import tomli as tomllib
from pathlib import Path
from typing import Any, Dict, MutableMapping, Optional, Union
from taskipy.task import Task
from taskipy.variable import Variable
from taskipy.exceptions import (
InvalidRunnerTypeError,
InvalidVariableError,
MalformedPyProjectError,
MissingPyProjectFileError,
MissingTaskipyTasksSectionError,
)
class PyProject:
def __init__(self, base_dir: Path):
self.__pyproject_path = PyProject.__find_pyproject_path(base_dir)
self.__items = PyProject.__load_toml_file(self.__pyproject_path)
@property
def tasks(self) -> Dict[str, Task]:
try:
toml_tasks = self.__items['tool']['taskipy']['tasks'].items()
except KeyError:
raise MissingTaskipyTasksSectionError()
tasks = {}
for name, toml_contents in toml_tasks:
tasks[name] = Task(name, toml_contents)
return tasks
@property
def dirpath(self) -> Path:
return self.__pyproject_path.parent
@property
def variables(self) -> Dict[str, Variable]:
try:
toml_vars = self.__items['tool']['taskipy'].get('variables', {})
except KeyError:
return {}
vars_dict: Dict[str, Variable] = {}
for name, toml_contents in toml_vars.items():
if isinstance(toml_contents, str):
vars_dict[name] = Variable(name, toml_contents, recursive=False)
elif (
isinstance(toml_contents, dict)
and isinstance(toml_contents.get('var'), str)
):
vars_dict[name] = Variable(
name,
toml_contents['var'],
toml_contents.get('recursive', False),
)
else:
raise InvalidVariableError(
name,
f'expected variable to contain a string or be a table '
'with a key "var" that contains a string value, got '
f'{toml_contents}.'
)
return vars_dict
@property
def settings(self) -> dict:
try:
return self.__items['tool']['taskipy']['settings']
except KeyError:
return {}
@property
def runner(self) -> Optional[str]:
try:
runner = self.settings['runner']
if not isinstance(runner, str):
raise InvalidRunnerTypeError()
return runner.strip()
except KeyError:
return None
@staticmethod
def __load_toml_file(file_path: Union[str, Path]) -> MutableMapping[str, Any]:
try:
if isinstance(file_path, str):
file_path = Path(file_path).resolve()
with open(file_path, 'rb') as file:
return tomllib.load(file)
except FileNotFoundError:
raise MissingPyProjectFileError()
except tomllib.TOMLDecodeError as e:
raise MalformedPyProjectError(reason=str(e))
@staticmethod
def __find_pyproject_path(base_dir: Path) -> Path:
def candidate_dirs(base: Path):
yield base
for parent in base.parents:
yield parent
for candidate_dir in candidate_dirs(base_dir):
pyproject = candidate_dir / 'pyproject.toml'
if pyproject.exists():
return pyproject
raise MissingPyProjectFileError()
|
