提交 998ef837 编写于 作者: W wjmcat

update args_config

上级 378e16fb
...@@ -29,7 +29,7 @@ from .loader import TemplateLoader, PluginLoader ...@@ -29,7 +29,7 @@ from .loader import TemplateLoader, PluginLoader
# pylint: disable=invalid-name # pylint: disable=invalid-name
log = get_logger() log = get_logger()
# pylint: disable=invalid-name # pylint: disable=invalid-name
args: dict = {} args_config: dict = {}
def run_server(): def run_server():
...@@ -46,7 +46,9 @@ def _save_to_file(template_dir: str, file_name: str, content: str): ...@@ -46,7 +46,9 @@ def _save_to_file(template_dir: str, file_name: str, content: str):
def _read_to_json(file: str) -> dict: def _read_to_json(file: str) -> dict:
"""read json file to json data""" """
read json file to json data
"""
file_path = os.path.join(os.getcwd(), file) file_path = os.path.join(os.getcwd(), file)
print(file_path) print(file_path)
if not os.path.exists(file_path): if not os.path.exists(file_path):
...@@ -55,16 +57,17 @@ def _read_to_json(file: str) -> dict: ...@@ -55,16 +57,17 @@ def _read_to_json(file: str) -> dict:
return json.load(f) return json.load(f)
def render_by_config(): def render_by_config(params: dict):
"""render the template with config file mode""" """render the template with config file mode"""
global args_config
args = args_config.update(params)
config = _read_to_json(args['config']) config = _read_to_json(args['config'])
plugins = PluginLoader(args['plugins']).load_to_file() plugins = PluginLoader(args['plugins']).load_to_file()
log.info(plugins)
templates = TemplateLoader(args['templates']).load_templates(plugins) templates = TemplateLoader(args['templates']).load_templates(plugins)
for name, template in templates.items(): for name, template in templates.items():
result = template.render(**config) result = template.render(**config)
base_name = os.path.basename(args['templates']) + '_result' dir_name = os.path.basename(args['templates']) + '_result'
_save_to_file(base_name, name, result) _save_to_file(dir_name, name, result)
def main(): def main():
...@@ -84,8 +87,8 @@ def main(): ...@@ -84,8 +87,8 @@ def main():
config_parser.set_defaults(func=render_by_config) config_parser.set_defaults(func=render_by_config)
local_args = parser.parse_args() local_args = parser.parse_args()
log.info(args) log.info(args_config)
args.update(local_args.__dict__) args_config.update(local_args.__dict__)
local_args.func() local_args.func()
......
...@@ -4,3 +4,4 @@ import pytest ...@@ -4,3 +4,4 @@ import pytest
def test_loader(): def test_loader():
"""tests template loader""" """tests template loader"""
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册