combining print and sys.exit statements

Author: dantownsend

piccolo/apps/app/commands/new.py

@@ -22,8 +22,7 @@ def new_app(app_name: str, root: str = "."):
     app_root = os.path.join(root, app_name)
 
     if os.path.exists(app_root):
-        print("Folder already exists - exiting.")
-        sys.exit(1)
+        sys.exit("Folder already exists - exiting.")
     os.mkdir(app_root)
 
     with open(os.path.join(app_root, "__init__.py"), "w"):

piccolo/apps/migrations/commands/new.py

@@ -75,8 +75,7 @@ def _create_new_migration(app_config: AppConfig, auto=False) -> None:
         )
 
         if sum([len(i.statements) for i in alter_statements]) == 0:
-            print("No changes detected - exiting.")
-            sys.exit(1)
+            sys.exit("No changes detected - exiting.")
 
         file_contents = render_template(
             migration_id=_id,
@@ -151,8 +150,7 @@ def new(app_name: str, auto: bool = False):
 
     engine = Finder().get_engine()
     if auto and isinstance(engine, SQLiteEngine):
-        print("Auto migrations aren't currently supported by SQLite.")
-        sys.exit(1)
+        sys.exit("Auto migrations aren't currently supported by SQLite.")
 
     app_config = Finder().get_app_config(app_name=app_name)
 

piccolo/apps/playground/commands/run.py

@@ -131,10 +131,9 @@ def run(
     try:
         import IPython  # type: ignore
     except ImportError:
-        print(
+        sys.exit(
             "Install iPython using `pip install ipython` to use this feature."
         )
-        sys.exit(1)
 
     if engine.upper() == "POSTGRES":
         db: Engine = PostgresEngine(

piccolo/apps/project/commands/new.py

@@ -23,8 +23,7 @@ def new_piccolo_conf(engine_name: str, force: bool = False, root: str = "."):
     file_path = os.path.join(root, "piccolo_conf.py")
 
     if os.path.exists(file_path) and not force:
-        print("The file already exists - exiting.")
-        sys.exit(1)
+        sys.exit("The file already exists - exiting.")
 
     with open(file_path, "w") as f:
         template = JINJA_ENV.get_template("piccolo_conf.py.jinja")

piccolo/apps/shell/commands/run.py

@@ -17,10 +17,9 @@ def start_ipython_shell(
     **tables: t.Dict[str, t.Type[Table]]
 ):  # pragma: no cover
     if not IPYTHON:
-        print(
+        sys.exit(
             "Install iPython using `pip install ipython` to use this feature."
         )
-        sys.exit(1)
 
     existing_global_names = globals().keys()
     for table_class_name, table_class in tables.items():

piccolo/apps/user/commands/change_password.py

@@ -17,8 +17,7 @@ def change_password():
     confirmed_password = get_confirmed_password()
 
     if not password == confirmed_password:
-        print("Passwords don't match!")
-        sys.exit(1)
+        sys.exit("Passwords don't match!")
 
     BaseUser.update_password_sync(user=username, password=password)
 

piccolo/apps/user/commands/create.py

@@ -43,12 +43,10 @@ def create():
     confirmed_password = get_confirmed_password()
 
     if not password == confirmed_password:
-        print("Passwords don't match!")
-        sys.exit(1)
+        sys.exit("Passwords don't match!")
 
     if len(password) < 4:
-        print("The password is too short")
-        sys.exit(1)
+        sys.exit("The password is too short")
 
     is_admin = get_is_admin()