Skip to content

Instantly share code, notes, and snippets.

@dnmellen
Created May 15, 2013 13:32
Show Gist options
  • Save dnmellen/5584007 to your computer and use it in GitHub Desktop.
Save dnmellen/5584007 to your computer and use it in GitHub Desktop.
Auxiliary Python module to get styled terminal outputs in a pythonic way
# encoding: utf-8
# Copyright 2013 Diego Navarro Mellén. All rights reserved.
#
# Redistribution and use in source and binary forms, with or without modification, are
# permitted provided that the following conditions are met:
#
# 1. Redistributions of source code must retain the above copyright notice, this list of
# conditions and the following disclaimer.
#
# 2. Redistributions in binary form must reproduce the above copyright notice, this list
# of conditions and the following disclaimer in the documentation and/or other materials
# provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY DIEGO NAVARRO MELLÉN ''AS IS'' AND ANY EXPRESS OR IMPLIED
# WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
# FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL DIEGO NAVARRO MELLÉN OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
# ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
# ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
# The views and conclusions contained in the software and documentation are those of the
# authors and should not be interpreted as representing official policies, either expressed
# or implied, of Diego Navarro Mellén.
from __future__ import print_function
# Special END separator
END = '0e8ed89a-47ba-4cdb-938e-b8af8e084d5c'
# Text attributes
ALL_OFF = '\033[0m'
BOLD = '\033[1m'
UNDERSCORE = '\033[4m'
BLINK = '\033[5m'
REVERSE = '\033[7m'
CONCEALED = '\033[7m'
# Foreground colors
FG_BLACK = '\033[30m'
FG_RED = '\033[31m'
FG_GREEN = '\033[32m'
FG_YELLOW = '\033[33m'
FG_BLUE = '\033[34m'
FG_MAGENTA = '\033[35m'
FG_CYAN = '\033[36m'
FG_WHITE = '\033[37m'
# Background colors
BG_BLACK = '\033[40m'
BG_RED = '\033[41m'
BG_GREEN = '\033[42m'
BG_YELLOW = '\033[43m'
BG_BLUE = '\033[44m'
BG_MAGENTA = '\033[45m'
BG_CYAN = '\033[46m'
BG_WHITE = '\033[47m'
class pretty_output():
'''
Context manager for pretty terminal prints
'''
def __init__(self, *attr):
self.attributes = attr
def __enter__(self):
return self
def __exit__(self, type, value, traceback):
pass
def write(self, msg):
style = ''.join(self.attributes)
print('{}{}{}'.format(style, msg.replace(END, ALL_OFF + style), ALL_OFF))
if __name__ == '__main__':
with pretty_output(FG_RED) as out:
out.write('This is a test in RED')
with pretty_output(FG_BLUE) as out:
out.write('This is a test in BLUE')
with pretty_output(BOLD, FG_GREEN) as out:
out.write('This is a bold text in green')
with pretty_output(BOLD, BG_GREEN) as out:
out.write('This is a text with green background')
with pretty_output(FG_GREEN) as out:
out.write('This is a green text with ' + BOLD + 'bold' + END + ' text included')
with pretty_output() as out:
out.write(BOLD + 'Use this' + END + ' even with ' + BOLD + FG_RED + 'no parameters' + END + ' in the with statement')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment