progress_logger.py

# Copyright Google

# BSD License

import copy
import wash

# from http://stackoverflow.com/questions/8924173/how-do-i-print-bold-text-in-python
class color:
  PURPLE = '\033[95m'
  CYAN = '\033[96m'
  DARKCYAN = '\033[36m'
  BLUE = '\033[94m'
  GREEN = '\033[92m'
  YELLOW = '\033[93m'
  RED = '\033[91m'
  BOLD = '\033[1m'
  UNDERLINE = '\033[4m'
  END = '\033[0m'

class TermLogger(object):
  def print_progress(self, lots, text, red_lots):
    lots = copy.copy(lots)  # so I can re-sort non-destructively
    print text
    lots.sort(cmp=wash.cmp_by_buy_date)
    red_ids = [id(lot) for lot in red_lots]
    for lot in lots:
      header = ''
      footer = ''
      if id(lot) in red_ids:
        header = color.RED
        footer = color.END
      print header + str(lot) + footer
    raw_input('hit enter>')

class NullLogger(object):
  def print_progress(self, lots, text, red_lots):
    pass