feat: create model scheme with CLI

2 files changed, 109 insertions, 3 deletions

diff --git a/src/frest/manage.py b/src/frest/manage.py
index 079719e..1164d70 100644
--- a/src/frest/manage.py
+++ b/src/frest/manage.py
@@ -13,7 +13,7 @@ def main():
         if not os.path.exists("scheme"):
             logging_arg("Create {}... ", "scheme/")
             logging("OK", 3, "\n")
-            os.makedirs("scheme")
+            os.mkdir("scheme")
 
         create_app(args.startapp)
 
diff --git a/src/frest/manage/utils.py b/src/frest/manage/utils.py
index 5aae18d..4f35937 100644
--- a/src/frest/manage/utils.py
+++ b/src/frest/manage/utils.py
@@ -1,9 +1,10 @@
 import os
-from .bcolors import COLORS 
+from .bcolors import COLORS
 
 
 ENDC = len(COLORS) - 1
 
+
 def logging(text, _type=ENDC, end=""):
     print(f"{COLORS[_type]}{text}{COLORS[ENDC]}", end=end)
 
@@ -12,6 +13,7 @@ def logging_arg(text, *args):
     args = [f"{COLORS[0]}{i}{COLORS[ENDC]}" for i in args]
     print(text.format(*args), end="")
 
+
 def logo():
     print(
         """  
@@ -25,5 +27,109 @@ def logo():
     )
 
 
+def inputsr():
+    return input().strip().replace(" ", "_").lower()
+
+
+def create_field(fields):
+    field = {"name": "", "type": "", "nullable": True}
+    logging("Choose field name: ")
+    field_name = inputsr()
+    while (len(field_name) < 1 or field_name[0].isdigit()) or (
+        field_name in fields or field_name in ["id", "created_at", "updated_at"]
+    ):
+        if len(field_name) < 1 or field_name[0].isdigit():
+            logging("Field name must not be empty or starts with a number", 2, "\n")
+        else:
+            logging("Field name already exists", 2, "\n")
+
+        logging("Choose field name: ")
+        field_name = inputsr()
+
+    field["name"] = field_name
+
+    logging("Choose field type: ")
+    logging("int, str, text, datetime, float, bool", 0, " ")
+    field_type = inputsr()
+    if field_type not in ["int", "str", "text", "datetime", "float", "bool"]:
+        logging("Field type must be one of the supported type", 2, "\n")
+        logging("Choose field type: ")
+        logging("int, str, text, datetime, float, bool", 0, " ")
+        field_type = inputsr()
+
+    if field_type == "str":
+        logging("Choose string size: ")
+        stringsize = inputsr()
+        while not stringsize.isdigit():
+            logging("String size must be an integer number", 2, "\n")
+            logging("Choose string size: ")
+            stringsize = inputsr()
+
+        stringsize = int(stringsize)
+        if stringsize < 1:
+            logging("You inserted 0, so it will be 1 by default", 4, "\n")
+            stringsize = 1
+
+        field["size"] = stringsize
+
+    field["type"] = field_type
+
+    logging("Field is nullable? (Y/n): ")
+    field_nullable = inputsr()
+
+    if len(field_nullable) > 0:
+        while True:
+            if field_nullable[0] not in ["y", "n"]:
+                logging("Field is nullable? (Y/n): ")
+                field_nullable = inputsr()
+            else:
+                break
+
+        if field_nullable[0] == "n":
+            field["nullable"] = False
+
+    return field
+
+
+def create_model_cli(name):
+    logging(
+        "Fields id, created_at, updated_at are default on every new model, you can delete it from model file",
+        4,
+        "\n\n",
+    )
+    fields = []
+    logging("Create field: (Y/n)")
+    answer = inputsr()
+
+    while answer in ["y", "", "yes"]:
+        field = create_field(fields)
+        fields.append(field)
+
+        logging("Create new field: (Y/n)")
+        answer = inputsr()
+
+    return fields
+
+
 def create_app(name):
-    pass
+    name = name.lower()
+    if len(name) < 2:
+        logging("Name of app must be minimun 2 characters long", 2)
+        return
+
+    if os.path.exists(f"scheme/{name}"):
+        logging("App already exists", 2)
+        return
+
+    logging_arg("Create {}... ", f"scheme/{name}")
+    os.mkdir(f"scheme/{name}")
+    logging("OK", 3, "\n")
+
+    logging_arg("Create {}... ", f"scheme/{name}/__init__.py")
+    open(f"scheme/{name}/__init__.py", "w").close()
+    logging("OK", 3, "\n")
+
+    logging_arg("Create model for {}...\n", name)
+    fields = create_model_cli(name)
+    print(fields)
+    logging("OK", 3, "\n")