-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathargparser.py
46 lines (35 loc) · 1.49 KB
/
argparser.py
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
class ArgumentsParser:
def __init__(self, params, arguments, method):
if not isinstance(arguments, dict):
raise Exception("Command arguments is not a dict")
self.params = params
self.arguments = arguments
self.method = method
self._parse()
def _check_required(self, required, key):
if required == False:
return
if required == True and key in self.params:
return
if required == True and key not in self.params:
raise KeyError(f"{key} is required and was not given")
def _parse(self):
for key, val in self.arguments.items():
if isinstance(val, dict):
if "required" in val:
required = val["required"]
if isinstance(val["required"], list) and self.method in required:
self._check_required(True, key)
else:
self._check_required(required, key)
if "type" in val:
type_ = val["type"]
if key in self.params and not isinstance(self.params[key], type_):
raise TypeError(f"'{key}' is not of type '{type_}'")
if "default" in val and key not in self.params:
self.params[key] = val["required"]
else:
if key not in self.params:
self.params[key] = val
def get_params(self):
return self.params