add loader&config to code_generator

src/code_generator/config.py

+"""
+Code Generator - https://github.com/wj-Mcat/code-generator
+
+Authors:    Jingjing WU (吴京京) <https://github.com/wj-Mcat>
+
+2020-now @ Copyright wj-Mcat
+
+Licensed under the Apache License, Version 2.0 (the 'License');
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an 'AS IS' BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+from __future__ import annotations
+
+import logging
+
+
+def get_logger():
+    """get the logger"""
+    log_formatter = logging.Formatter(
+        fmt='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+
+    logger = logging.getLogger("CodeGenerator")
+    logger.handlers = []
+    logger.setLevel(logging.DEBUG)
+    logger.propagate = False
+
+    # create console handler and set level to info
+    console_handler = logging.StreamHandler()
+    console_handler.setLevel(logging.INFO)
+    console_handler.setFormatter(log_formatter)
+    logger.addHandler(console_handler)
+
+    return logger

src/code_generator/loader.py

+"""
+Code Generator - https://github.com/wj-Mcat/code-generator
+
+Authors:    Jingjing WU (吴京京) <https://github.com/wj-Mcat>
+
+2020-now @ Copyright wj-Mcat
+
+Licensed under the Apache License, Version 2.0 (the 'License');
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an 'AS IS' BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+from __future__ import annotations
+import os
+from typing import Dict
+from abc import ABCMeta
+
+from jinja2 import Template
+from jinja2.runtime import Macro
+
+from .config import get_logger
+
+
+log = get_logger()
+
+
+class Loader(metaclass=ABCMeta):
+    """load the plugins from local or default path"""
+    def __init__(self, file_or_dir: str):
+        """
+        init the plugin files path, not load the file content,
+        lazy load plugins
+        """
+        self.files: Dict[str, str] = {}
+        log.info('load the plugins from : %s', file_or_dir)
+        if not os.path.exists(file_or_dir):
+            log.warning('plugin file/dir doesn"t exist, there are '
+                        'no plugins to load')
+        self.file_or_dir = file_or_dir
+
+    def _load_files(self):
+        """load the plugins from file or dir"""
+        if os.path.isdir(self.file_or_dir):
+            files = os.listdir(self.file_or_dir)
+
+            for file in files:
+                file_path = os.path.join(self.file_or_dir, file)
+
+                with open(file_path, 'r+', encoding='utf-8') as f:
+                    code = f.read()
+                    file_name = file
+
+                    if '.' in file_name:
+                        file_name = file_name[:file_name.rindex('.')]
+
+                    if file_name in self.files:
+                        log.warning('there are the same plugin file name : %s'
+                                    , file_name)
+
+                        # add format for file_name
+                        if '.' in file_name:
+                            r_index = file_name.rindex('.')
+                            file_type = file_name[r_index:]
+                            file_name = file_name[:r_index] + '%s' + file_type
+                        else:
+                            file_name = file_name + '%s'
+
+                        file_name = file_name % '-more'
+
+                    self.files[file_name] = code
+
+
+class TemplateLoader(Loader):
+    """template loader"""
+    def __init__(self, file_or_dir: str = 'templates'):
+        super(TemplateLoader, self).__init__(file_or_dir)
+
+    def load_templates(self) -> Dict[str, Template]:
+        """load template"""
+        self._load_files()
+        templates: Dict[str, Template] = {}
+        for key, template in self.files.items():
+            templates[key] = Template(template)
+        return templates
+
+
+class PluginLoader(Loader):
+    """plugin loader"""
+    def __init__(self, file_or_dir: str = 'templates'):
+        super(PluginLoader, self).__init__(file_or_dir)
+
+    def _load_plugins(self) -> Dict[str, str]:
+        self._load_files()
+        return self.files
+
+    def load_to_file(self) -> str:
+        files = self._load_plugins()
+        return '\n'.join(files)
+
+    def load_plugin_descriptions(self) -> Dict[str, str]:
+        descriptions: Dict[str, str] = {}
+        plugins = self._load_plugins()
+
+        for name, plugin in plugins.items():
+            module = Template(plugin).module
+            attrs = dir(module)
+            macros = [(attr, getattr(module, attr)) for attr in attrs
+                      if isinstance(getattr(module, attr), Macro)]
+
+            # find macros description from html node description using bs4
+            
+
+
+        return descriptions
+
+
+

src/code_generator/main.py

+"""
+Code Generator - https://github.com/wj-Mcat/code-generator
+
+Authors:    Jingjing WU (吴京京) <https://github.com/wj-Mcat>
+
+2020-now @ Copyright wj-Mcat
+
+Licensed under the Apache License, Version 2.0 (the 'License');
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an 'AS IS' BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+from __future__ import annotations
+from argparse import ArgumentParser
+
+from .config import get_logger
+from .loader import TemplateLoader, PluginLoader
+
+
+log = get_logger()
+args = {}
+
+
+def run_server():
+    pass
+
+
+def render_by_config():
+    file = args['file']
+    templates = TemplateLoader(file).load_templates()
+    plugin_file = PluginLoader(file).load_to_file()
+
+
+
+
+def main():
+    parser = ArgumentParser()
+    subparsers = parser.add_subparsers(help='sub command')
+    serve_parser = subparsers.add_parser(name='serve')
+    serve_parser.add_argument('--model', default='models', type=str)
+    serve_parser.add_argument('--port', default=5002, type=int)
+    serve_parser.add_argument('--file', default='config.json', type=str)
+    serve_parser.set_defaults(func=run_server)
+
+    config_parser = subparsers.add_parser(name='config')
+    config_parser.add_argument('--file', default='config.json', type=str)
+    config_parser.add_argument('--template', default='flask,vue', type=str)
+    config_parser.set_defaults(func=render_by_config)
+
+    local_args = parser.parse_args()
+    log.info(args)
+    args.update(local_args.__dict__)
+    local_args.func(args)
+
+
+if __name__ == '__main__':
+    main()