diff options
Diffstat (limited to 'pygments/formatters/other.py')
| -rw-r--r-- | pygments/formatters/other.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/pygments/formatters/other.py b/pygments/formatters/other.py index 9422dd23..bd375f32 100644 --- a/pygments/formatters/other.py +++ b/pygments/formatters/other.py @@ -10,6 +10,7 @@ """ from pygments.formatter import Formatter +from pygments.util import get_choice_opt __all__ = ['NullFormatter', 'RawTokenFormatter'] @@ -54,7 +55,8 @@ class RawTokenFormatter(Formatter): def __init__(self, **options): Formatter.__init__(self, **options) - self.compress = options.get('compress', '') + self.compress = get_choice_opt(options, 'compress', + ['', 'none', 'gz', 'bz2'], '') def format(self, tokensource, outfile): if self.compress == 'gz': |