python类打印bash颜色

Question

我有可以打印到屏幕上的 python 脚本，有时我希望它们是粗体或绿色，有时是粗体和绿色。但我似乎无法弄清楚如何做到这两点。

class colortext():
    def __init__(self, text:str):
        self.text = text
        self.ending = '\033[0m'

    def bold(self):
        return '\033[1m' + self.text + self.ending

    def green(self):
        return '\033[92m' + self.text + self.ending


print(colortext('hello').bold().green())

AttributeError: 'str' object has no attribute 'green'

Answer 1

我认为更好的策略是让您的方法将控制序列添加到用于在 __str__ 方法中构造最终字符串的列表中：

class ColorText():
    def __init__(self, text:str):
        self.text = text
        self.ending = '\033[0m'
        self.sequences = []
    def bold(self):
        self.sequences.append('\033[1m')
        return self
    def green(self):
        self.sequences.append('\033[92m')
        return self
    def __str__(self):
        return ''.join(self.sequences) + self.text + self.ending  # ''.join(chain would probably work better

Answer 2

因为您在 bold 和 green 方法中返回一个字符串。我认为您实际上想要返回对 colortext 对象本身的引用。所以修改这些方法中的文本并返回self。此外，如果您希望它在print 中调用它时打印字符串，请定义__str__ 函数。试试：

class colortext():
    def __init__(self, text:str):
        self.text = text
        self.ending = '\033[0m'

    def __str__(self):
        return self.text

    def bold(self):
        self.text = '\033[1m' + self.text + self.ending
        return self

    def green(self):
        self.text = '\033[92m' + self.text + self.ending
        return self

print(colortext('hello').bold().green())