mirror of
https://github.com/chatmail/relay.git
synced 2026-05-10 16:04:37 +00:00
cmdeploy: remove colors from cmdeploy init again, hard to test
This commit is contained in:
@@ -25,9 +25,6 @@ from .sshexec import SSHExec
|
|||||||
# cmdeploy sub commands and options
|
# cmdeploy sub commands and options
|
||||||
#
|
#
|
||||||
|
|
||||||
def is_pytest():
|
|
||||||
return "PYTEST_CURRENT_TEST" in os.environ
|
|
||||||
|
|
||||||
|
|
||||||
def init_cmd_options(parser):
|
def init_cmd_options(parser):
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
@@ -49,15 +46,18 @@ def init_cmd(args, out):
|
|||||||
inipath = args.inipath
|
inipath = args.inipath
|
||||||
if args.inipath.exists():
|
if args.inipath.exists():
|
||||||
if not args.recreate_ini:
|
if not args.recreate_ini:
|
||||||
out.green(f"[WARNING] Path exists, not modifying: {inipath}")
|
print(f"[WARNING] Path exists, not modifying: {inipath}")
|
||||||
return 1
|
return 1
|
||||||
else:
|
else:
|
||||||
out.yellow(f"[WARNING] Force argument was provided, deleting config file: {inipath}")
|
print(
|
||||||
|
f"[WARNING] Force argument was provided, deleting config file: {inipath}"
|
||||||
|
)
|
||||||
inipath.unlink()
|
inipath.unlink()
|
||||||
|
|
||||||
write_initial_config(inipath, mail_domain, overrides={})
|
write_initial_config(inipath, mail_domain, overrides={})
|
||||||
out.green(f"created config file for {mail_domain} in {inipath}")
|
out.green(f"created config file for {mail_domain} in {inipath}")
|
||||||
|
|
||||||
|
|
||||||
def run_cmd_options(parser):
|
def run_cmd_options(parser):
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--dry-run",
|
"--dry-run",
|
||||||
@@ -276,20 +276,11 @@ class Out:
|
|||||||
def red(self, msg, file=sys.stderr):
|
def red(self, msg, file=sys.stderr):
|
||||||
print(colored(msg, "red"), file=file)
|
print(colored(msg, "red"), file=file)
|
||||||
|
|
||||||
def green(self, msg, file=sys.stdout):
|
def green(self, msg, file=sys.stderr):
|
||||||
print(colored(msg, "green"), file=file)
|
print(colored(msg, "green"), file=file)
|
||||||
|
|
||||||
def yellow(self, msg, file=sys.stdout):
|
def __call__(self, msg, red=False, green=False, file=sys.stdout):
|
||||||
print(colored(msg, "yellow"), file=file)
|
color = "red" if red else ("green" if green else None)
|
||||||
|
|
||||||
def __call__(self, msg, red=False, green=False, yellow=False, file=sys.stdout):
|
|
||||||
color = None
|
|
||||||
if red:
|
|
||||||
color = "red"
|
|
||||||
elif green:
|
|
||||||
color = "green"
|
|
||||||
elif yellow:
|
|
||||||
color = "yellow"
|
|
||||||
print(colored(msg, color), file=file)
|
print(colored(msg, color), file=file)
|
||||||
|
|
||||||
def check_call(self, arg, env=None, quiet=False):
|
def check_call(self, arg, env=None, quiet=False):
|
||||||
@@ -374,12 +365,6 @@ def main(args=None):
|
|||||||
args.get_sshexec = get_sshexec
|
args.get_sshexec = get_sshexec
|
||||||
|
|
||||||
out = Out()
|
out = Out()
|
||||||
if is_pytest(): ## issue: https://github.com/chatmail/relay/issues/622
|
|
||||||
out.green = print
|
|
||||||
out.red = print
|
|
||||||
out.yellow = print
|
|
||||||
out.__call__ = print
|
|
||||||
|
|
||||||
kwargs = {}
|
kwargs = {}
|
||||||
if args.func.__name__ not in ("init_cmd", "fmt_cmd"):
|
if args.func.__name__ not in ("init_cmd", "fmt_cmd"):
|
||||||
if not args.inipath.exists():
|
if not args.inipath.exists():
|
||||||
|
|||||||
@@ -25,8 +25,7 @@ class TestCmdline:
|
|||||||
|
|
||||||
def test_init_not_overwrite(self, capsys):
|
def test_init_not_overwrite(self, capsys):
|
||||||
assert main(["init", "chat.example.org"]) == 0
|
assert main(["init", "chat.example.org"]) == 0
|
||||||
out, err = capsys.readouterr()
|
capsys.readouterr()
|
||||||
assert "created config file" in out.lower()
|
|
||||||
|
|
||||||
assert main(["init", "chat.example.org"]) == 1
|
assert main(["init", "chat.example.org"]) == 1
|
||||||
out, err = capsys.readouterr()
|
out, err = capsys.readouterr()
|
||||||
|
|||||||
Reference in New Issue
Block a user