Simplify extension test for XML format

DarkWizarD24 · Oct 7, 2015 · 1f8225f · 1f8225f
1 parent d33c9af
commit 1f8225f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jinja2cli/cli.py b/jinja2cli/cli.py
@@ -211,7 +211,7 @@ def cli(opts, args):
             else:
                 if ext in ('yml', 'yaml'):
                     raise InvalidDataFormat('%s: install pyyaml to fix' % ext)
-                if ext in ('xml'):
+                if ext == 'xml':
                     raise InvalidDataFormat('%s: install xmltodict' % ext)
                 raise InvalidDataFormat(ext)
         data = open(path).read()
@@ -301,7 +301,7 @@ def main():
     if opts.format not in formats and opts.format != 'auto':
         if opts.format in ('yml', 'yaml'):
             raise InvalidDataFormat('%s: install pyyaml to fix' % opts.format)
-        if opts.format in ('xml'):
+        if opts.format == 'xml':
             raise InvalidDataFormat('%s: install xmltodict' % opts.format)
         raise InvalidDataFormat(opts.format)