copy default template

Author: benjaminhon

bin/modelmaker

@@ -10,7 +10,7 @@ import subprocess
 from importlib import import_module
 from modelmaker import folders_in_folder
 import shutil
-from jinja2 import Template
+from jinja2 import Template, Environment, PackageLoader, FileSystemLoader
 
 module_folder = os.path.dirname(sys.modules['modelmaker'].__file__)
 current_folder = os.path.abspath(os.curdir)
@@ -32,40 +32,43 @@ args = parser.parse_args()
 ############################################################################################
 
 if args.subcommand == 'new':
+
+    template = 'default'
+    template_folder = os.path.join(module_folder, 'resources', 'templates', template)
     project_path = os.path.join(current_folder, args.name)
+
     print(f"Creating project in {project_path} ...")
 
     # copy project template
     shutil.copytree(
-        os.path.join(module_folder, 'resources', 'project_template'),
-        project_path
+        template_folder,
+        project_path,
+        ignore=shutil.ignore_patterns('*.pyc', '__pycache__')
     )
 
     # rename package folder
     os.rename(
         os.path.join(project_path, 'package'),
-        os.path.join(project_path, args.name),
+        os.path.join(project_path, args.name)
     )
 
-    # template setup.py
-    setup_file = os.path.join(project_path, 'setup.py')
-    template = Template(open(setup_file).read())
-    with open(setup_file, 'w') as f:
-        print(f"Generating template {setup_file} ...")
-        f.write(template.render(package_name=args.name))
+    env = Environment(
+        loader=FileSystemLoader(project_path)
+    )
 
-    # template README.md
-    readme_file = os.path.join(project_path, 'README.md')
-    template = Template(open(readme_file).read())
-    with open(readme_file, 'w') as f:
-        print(f"Generating template {readme_file} ...")
-        f.write(template.render(package_name=args.name))
+    # replace template strings
+    for file in glob(os.path.join(project_path, '**/*'), recursive=True):
+        if os.path.isfile(file):
+            template = env.get_template(os.path.relpath(file, start=project_path))
+            with open(file, 'w') as f:
+                print(f"Generating template {file} ...")
+                f.write(template.render(package_name=args.name))
 
 ############################################################################################
 ## list templates
 ############################################################################################
 
-if args.subcommand == 'templates':
+elif args.subcommand == 'templates':
     print("\nTemplates available:")
     for t in templates:
         print(f"    {t}")

modelmaker/resources/project_template/README.md

@@ -0,0 +1,16 @@
+# {{ package_name }}
+
+## Install
+
+```bash
+cd {{ package_name }}
+pip3 install ./
+```
+
+## Using The Model
+
+Import the python module
+
+```python
+import {{ package_name }}
+```