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
|
import argparse
import json
from dataclasses import dataclass
from typing import Optional
@dataclass
class LinterArgs:
source_root: str
project_path: str
output_path: str
lint_name: str
depends: dict[str, str]
global_resources: dict[str, str]
configs: list[str]
extra_params: dict[str, str]
report_file: str
files: list[str]
def get_params(raw_args: Optional[list[str]] = None) -> LinterArgs:
parser = argparse.ArgumentParser()
parser.add_argument("--params")
parser.add_argument("--source-root")
parser.add_argument("--project-path")
parser.add_argument("--output-path")
parser.add_argument("--lint-name", default="")
parser.add_argument("--depends", action="append")
parser.add_argument("--global-resource", action="append", dest="global_resources")
parser.add_argument("--config", action="append", dest="configs")
parser.add_argument("--extra-param", action="append", dest="extra_params")
parser.add_argument("--report-file", default="-")
parser.add_argument("files", nargs="*")
args = parser.parse_args(raw_args)
if args.params:
with open(args.params) as f:
params = json.load(f)
source_root = params["source_root"]
project_path = params["project_path"]
output_path = params["output_path"]
lint_name = params.get("lint_name", "")
depends = params.get("depends", {})
global_resources = params.get("global_resources", {})
configs = params.get("configs", [])
extra_params = params.get("extra_params", {})
report_file = params["report_file"]
files = params["files"]
else:
source_root = args.source_root
project_path = args.project_path
output_path = args.output_path
lint_name = args.lint_name
depends = _parse_kv_arg(args.depends, ":")
global_resources = _parse_kv_arg(args.global_resources, ":")
configs = args.configs if args.configs else []
extra_params = _parse_kv_arg(args.extra_params, "=")
report_file = args.report_file
files = args.files
return LinterArgs(
source_root=source_root,
project_path=project_path,
output_path=output_path,
lint_name=lint_name,
depends=depends,
global_resources=global_resources,
configs=configs,
extra_params=extra_params,
report_file=report_file,
files=files,
)
def _parse_kv_arg(arg, sep):
result = {}
if arg:
for item in arg:
var, val = item.split(sep, 1)
result[var] = val
return result
|
