Python: Define Function to contain list of attributes as variable

Question

I am using the following class to format integers and string values into a desired format as shown below:

class bcolors:
    HEADER = '\033[95m'
    OKBLUE = '\033[94m'
    OKGREEN = '\033[92m'
    WARNING = '\033[93m'
    FAIL = '\033[91m'
    ENDC = '\033[0m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'

print(bcolors.WARNING + "Sample Warning Text. Continue?" + bcolors.ENDC)

However, my goal is to have a function such as:

color_format(input,format)

which would essentially make the use of formatting easier.

Is there an option to define an array of possible option in a function? E.g.

def color_format(input, format)
         format = [
                   HEADER = '\033[95m',
                   OKBLUE = '\033[94m',
                   OKGREEN = '\033[92m',
                   WARNING = '\033[93m',
                   FAIL = '\033[91m',
                   ENDC = '\033[0m',
                   BOLD = '\033[1m',
                   UNDERLINE = '\033[4m'
                   ]
          print(format + input + ENDC='\033[0m')

color_format("This is a bold text",BOLD)

~This is a bold text

I'd appreciate your help alot!

Fedor Soldatkin · Accepted Answer · 2020-10-06 14:22:28Z

3

Yes, you can do this using dict with mapping from names to values.

FORMATS = {
    'HEADER': '\033[95m',
    'OKBLUE': '\033[94m',
    'OKGREEN': '\033[92m',
    'WARNING': '\033[93m',
    'FAIL': '\033[91m',
    'ENDC': '\033[0m',
    'BOLD': '\033[1m',
    'UNDERLINE': '\033[4m'
}

def color_format(text, format):
    try:
        formatted = FORMATS[format] + text + FORMATS['ENDC']
    except KeyError:
        print('Invalid format! Possible values are: ', FORMATS.keys())
    else:
        print(formatted)

answered Oct 6, 2020 at 14:22

Fedor Soldatkin

1,20311 silver badges24 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Python: Define Function to contain list of attributes as variable

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related