Source code for c4dynamics.utils.cprint

import sys 
sys.path.append('.')

TXTCOLORS =     { 'k': '30', 'black':   '30'
                , 'r': '31', 'red':     '31'
                , 'g': '32', 'green':   '32'
                , 'y': '33', 'yellow':  '33'
                , 'b': '34', 'blue':    '34'
                , 'm': '35', 'magenta': '35'
                , 'c': '36', 'cyan':    '36'
                , 'w': '37', 'white':   '37'
                    }



def cprint(txt = '', color = 'white', end = '\n'):
  '''
    Printing colored text in the console.

    Parameters
    ----------

    txt : str
        The text to be printed.
    
    color : str, optional
        The color of the text. Default is 'white'.
    
    Example
    -------

    .. code:: 
    
      >>> carr = ['y', 'w', 'r', 'm', 'c', 'g', 'k', 'b']
      >>> for c in carr:
      ...   c4d.cprint('C4DYNAMICS', c)

    .. raw:: html

      <span style="color:yellow">C4DYNAMICS</span><br>
      <span style="color:white">C4DYNAMICS</span><br>
      <span style="color:red">C4DYNAMICS</span><br>
      <span style="color:magenta">C4DYNAMICS</span><br>
      <span style="color:cyan">C4DYNAMICS</span><br>
      <span style="color:green">C4DYNAMICS</span><br>
      <span style="color:black">C4DYNAMICS</span><br>
      <span style="color:blue">C4DYNAMICS</span><br>
 
  '''
  settxt = '\033['


  settxt += TXTCOLORS[color] 
      
  print(settxt + 'm' + str(txt) + '\033[0m', end = end)



if __name__ == "__main__":

  import os 
  import doctest, contextlib
  from c4dynamics import IgnoreOutputChecker, cprint
  
  # Register the custom OutputChecker
  doctest.OutputChecker = IgnoreOutputChecker

  tofile = False 
  optionflags = doctest.FAIL_FAST

  if tofile: 
    with open(os.path.join('tests', '_out', 'output.txt'), 'w') as f:
      with contextlib.redirect_stdout(f), contextlib.redirect_stderr(f):
        result = doctest.testmod(optionflags = optionflags) 
  else: 
    result = doctest.testmod(optionflags = optionflags)

  if result.failed == 0:
    cprint("All tests passed!", 'g')
  else:
    print(f"{result.failed}")