File tree Expand file tree Collapse file tree 1 file changed +11
-8
lines changed Expand file tree Collapse file tree 1 file changed +11
-8
lines changed Original file line number Diff line number Diff line change 3
3
import argparse
4
4
import logging
5
5
import sys
6
+ from typing import Set
6
7
7
8
from . import options , render
8
9
9
10
10
- def _parse (tags ) :
11
+ def _parse (tags : Set [ str ]) -> argparse . Namespace :
11
12
parser = argparse .ArgumentParser ()
12
13
for opt in options .get (tags ):
13
14
opt .add_to (parser )
14
- ret = parser .parse_args ()
15
- log_level = logging .DEBUG if ret .verbose else logging .INFO
16
- logging .basicConfig (format = "{levelname} {message}" , style = '{' , level = log_level )
17
- return ret
15
+ return parser .parse_args ()
18
16
19
17
20
- def run (* modules ):
18
+ def run (* modules , ** kwargs ):
21
19
""" run generation functions in specified in `modules`, or in current module by default
22
20
"""
23
21
if modules :
24
- opts = _parse ({t for m in modules for t in m .tags })
22
+ if kwargs :
23
+ opts = argparse .Namespace (** kwargs )
24
+ else :
25
+ opts = _parse ({t for m in modules for t in m .tags })
26
+ log_level = logging .DEBUG if opts .verbose else logging .INFO
27
+ logging .basicConfig (format = "{levelname} {message}" , style = '{' , level = log_level )
25
28
for m in modules :
26
29
m .generate (opts , render .Renderer ())
27
30
else :
28
- run (sys .modules ["__main__" ])
31
+ run (sys .modules ["__main__" ], ** kwargs )
You can’t perform that action at this time.
0 commit comments