cli func to return exit status code

mattrobenolt · Jan 3, 2017 · 4b6c048 · 4b6c048
1 parent 9276bff
commit 4b6c048
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/jinja2cli/cli.py b/jinja2cli/cli.py
@@ -252,14 +252,14 @@ def cli(opts, args):
             data = data[section]
         else:
             sys.stderr.write('ERROR: unknown section. Exiting.')
-            sys.exit(1)
+            return 1
 
     output = render(template_path, data, extensions, opts.strict)
 
     if isinstance(output, binary_type):
         output = output.decode('utf-8')
     sys.stdout.write(output)
-    sys.exit(0)
+    return 0
 
 
 def parse_kv_string(pairs):
@@ -332,8 +332,7 @@ def main():
     if opts.format not in formats and opts.format != 'auto':
         raise InvalidDataFormat(opts.format)
 
-    cli(opts, args)
-    sys.exit(0)
+    sys.exit(cli(opts, args))
 
 
 if __name__ == '__main__':