diff --git a/image_creator/main.py b/image_creator/main.py index 319fbf3eee2ff4a493c67e6ef4a47df20122f18b..8857eb84fd599deb14f427ac62fd07d72c1b5fd2 100644 --- a/image_creator/main.py +++ b/image_creator/main.py @@ -39,7 +39,7 @@ from image_creator.disk import Disk from image_creator.util import get_command, FatalError, MD5 from image_creator.output.cli import SilentOutput, SimpleOutput, \ OutputWthProgress -from image_creator.os_type import get_os_class +from image_creator.os_type import os_cls from image_creator.kamaki_wrapper import Kamaki import sys import os @@ -187,8 +187,8 @@ def image_creator(): dev = disk.get_device(snapshot) dev.mount() - osclass = get_os_class(dev.distro, dev.ostype) - image_os = osclass(dev.root, dev.g, out) + cls = os_cls(dev.distro, dev.ostype) + image_os = cls(dev.root, dev.g, out) out.output() for sysprep in options.disabled_syspreps: @@ -283,10 +283,8 @@ def main(): ret = image_creator() sys.exit(ret) except FatalError as e: - if sys.stdout.isatty(): - error(e) - else: - error(e, True, False) + colored = sys.stderr.isatty() + SimpleOutput(colored).error(e) sys.exit(1) diff --git a/image_creator/os_type/__init__.py b/image_creator/os_type/__init__.py index 98d570d0727982a17e0497f8481a43e499564ae3..6fe13366fea535fbdd651c6a2797ed4ada0c14e1 100644 --- a/image_creator/os_type/__init__.py +++ b/image_creator/os_type/__init__.py @@ -37,7 +37,7 @@ import textwrap import re -def get_os_class(distro, osfamily): +def os_cls(distro, osfamily): module = None classname = None try: