diff options
| -rwxr-xr-x | bin/frest | 11 | ||||
| -rw-r--r-- | frest/manage/utils.py | 22 |
2 files changed, 32 insertions, 1 deletions
@@ -1,14 +1,16 @@ #!/usr/bin/env python import argparse -from frest.manage.utils import logo, create_app +from frest.manage.utils import logo, create_app, init_project from frest.manage.utils import logging, logging_arg import os +import sys def main(): parser = argparse.ArgumentParser() parser.add_argument("--startapp", "-s", type=str, help="create new app") + parser.add_argument("--init", "-", type=str, help="create new project") args = parser.parse_args() if args.startapp: @@ -18,6 +20,13 @@ def main(): os.mkdir("scheme") create_app(args.startapp) + elif args.init: + appname = args.init + if os.path.exists(appname): + logging("App already exists", 2, "\n") + sys.exit(1) + + init_project(appname) if __name__ == "__main__": diff --git a/frest/manage/utils.py b/frest/manage/utils.py index 347d3f8..a364ae9 100644 --- a/frest/manage/utils.py +++ b/frest/manage/utils.py @@ -226,3 +226,25 @@ def create_app(name): logging_arg("Create {}... ", f"scheme/{name}/routes.py") create_routes(name, fields) logging("OK", 3, "\n") + + +def init_project(name): + logging_arg("Create {}... ", f"{name}/") + os.mkdir(f"{name}") + logging("OK", 3, "\n") + + path = f"{name}/__init__.py" + logging_arg("Create {}... ", path) + open(path, "w").close() + logging("OK", 3, "\n") + + paths = ["app", "database", "mail", "wsgi"] + for path in paths: + logging_arg("Create {}... ", f"{name}/{path}.py") + with open(os.path.join(TEMPLATE_PATH, f"{path}.txt")) as f: + text = "".join(f.readlines()) + + with open(f"{name}/{path}.py", "w") as f: + f.write(text) + + logging("OK", 3, "\n") |