def __init__(self, **options): Formatter.__init__(self, **options) if self.encoding: raise OptionError("the raw formatter does not support the " "encoding option") self.encoding = "ascii" # let plushcms.syntaxhighlight.pygments.format() do the right thing self.compress = get_choice_opt(options, "compress", ["", "none", "gz", "bz2"], "") self.error_color = options.get("error_color", None) if self.error_color is True: self.error_color = "red" if self.error_color is not None: try: colorize(self.error_color, "") except KeyError: raise ValueError("Invalid color %r specified" % self.error_color)
def __init__(self, **options): Formatter.__init__(self, **options) if self.encoding: raise OptionError('the raw formatter does not support the ' 'encoding option') self.encoding = 'ascii' # let plushcms.syntaxhighlight.pygments.format() do the right thing self.compress = get_choice_opt(options, 'compress', ['', 'none', 'gz', 'bz2'], '') self.error_color = options.get('error_color', None) if self.error_color is True: self.error_color = 'red' if self.error_color is not None: try: colorize(self.error_color, '') except KeyError: raise ValueError("Invalid color %r specified" % self.error_color)
def format(self, tokensource, outfile): try: outfile.write(b("")) except TypeError: raise TypeError("The raw tokens formatter needs a binary " "output file") if self.compress == "gz": import gzip outfile = gzip.GzipFile("", "wb", 9, outfile) def write(text): outfile.write(text.encode()) flush = outfile.flush elif self.compress == "bz2": import bz2 compressor = bz2.BZ2Compressor(9) def write(text): outfile.write(compressor.compress(text.encode())) def flush(): outfile.write(compressor.flush()) outfile.flush() else: def write(text): outfile.write(text.encode()) flush = outfile.flush lasttype = None lastval = u"" if self.error_color: for ttype, value in tokensource: line = "%s\t%r\n" % (ttype, value) if ttype is Token.Error: write(colorize(self.error_color, line)) else: write(line) else: for ttype, value in tokensource: write("%s\t%r\n" % (ttype, value)) flush()
def format(self, tokensource, outfile): try: outfile.write(b('')) except TypeError: raise TypeError('The raw tokens formatter needs a binary ' 'output file') if self.compress == 'gz': import gzip outfile = gzip.GzipFile('', 'wb', 9, outfile) def write(text): outfile.write(text.encode()) flush = outfile.flush elif self.compress == 'bz2': import bz2 compressor = bz2.BZ2Compressor(9) def write(text): outfile.write(compressor.compress(text.encode())) def flush(): outfile.write(compressor.flush()) outfile.flush() else: def write(text): outfile.write(text.encode()) flush = outfile.flush lasttype = None lastval = u'' if self.error_color: for ttype, value in tokensource: line = "%s\t%r\n" % (ttype, value) if ttype is Token.Error: write(colorize(self.error_color, line)) else: write(line) else: for ttype, value in tokensource: write("%s\t%r\n" % (ttype, value)) flush()