(python)在函数中使用装饰器进行彩色打印

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了(python)在函数中使用装饰器进行彩色打印，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1635字，纯文字阅读大概需要3分钟。

内容图文

我如何装饰一个函数,以便它打印到stdout的任何东西都是绿色的,打印到stderr的任何东西都是红色的？我有termcolor模块可用.

奖金业力：如何将参数传递给装饰器以指定颜色,将它们默认为红色和绿色？

解决方法:

一个有趣的问题.最简单的解决方案与Pete建议的类似.只需在将函数运行到每个stderr和stdout之前打印转义码.但是,如果stderr和stdout都是同一个终端,通常就是这种情况,它们会干扰.

因此,另一种解决方案是使用一个微小的包装器修补stdout和stderr,在每次写入的持续时间内启用颜色,只有当我们在终端(而不是管道)时才注意这一点.

#!/usr/bin/python2

import sys

def colorize(stdoutColor, stderrColor):
  defaultColor = '\033[0;0m'

  def applyColorize(f):
    class colorWrapper(object):
      def __init__(self, wrapee, color):
        self.wrapee = wrapee
        self.color = color
      def __getattr__(self, attr):
        if attr == 'write' and self.wrapee.isatty():
          return lambda x: self.wrapee.write(self.color + x + defaultColor)
        else:
          return getattr(self.wrapee, attr)

    def wrapper(*args, **kwds):
      oldStdout = sys.stdout
      oldStderr = sys.stderr
      sys.stdout = colorWrapper(oldStdout, stdoutColor)
      sys.stderr = colorWrapper(oldStderr, stderrColor)
      try:
        f(*args, **kwds)
      finally:
        sys.stdout = oldStdout
        sys.stderr = oldStderr

    return wrapper

  return applyColorize


greenColor = '\033[01;32m'
redColor = '\033[01;31m'

def foo():
  print "I'm ordinary and boring!"
  print >> sys.stderr, 'Writing to stderr!'

@colorize(greenColor, redColor)
def colorFoo():
  print "I'm colorful and exciting!"
  print >> sys.stderr, 'Writing to stderr!'

if __name__ == '__main__':
  foo()
  colorFoo()
  foo()

这仍然可以稍微改进一下,但它应该在大多数情况下完成工作并在适当后清理.当然,请记住我使用的是特定于shell的转义码.如果您想要可移植性,则必须使用对便携式终端控制模块的调用替换转义码写入.

内容总结

以上是互联网集市为您收集整理的(python)在函数中使用装饰器进行彩色打印全部内容，希望文章能够帮你解决(python)在函数中使用装饰器进行彩色打印所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端