summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/frest/manage.py2
-rw-r--r--src/frest/manage/utils.py110
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")