Make the asn1ate command work again.

[third_party/asn1ate.git] / asn1ate / pyasn1gen.py

diff --git a/asn1ate/pyasn1gen.py b/asn1ate/pyasn1gen.py
index 0040c4c80b58f3bcfe29b434994ddd258e3854f3..6d640ba87d01327636a9db3d57757b536b0baa54 100755 (executable)
--- a/asn1ate/pyasn1gen.py
+++ b/asn1ate/pyasn1gen.py
@@ -667,7 +667,7 @@ def main(args):
     return 0
 
 
-if __name__ == '__main__':
+def main_cli():
     arg_parser = argparse.ArgumentParser(
         description=('Generate Python classes from an ASN.1 definition file. '
                      'Output to stdout by default.'))
@@ -675,4 +675,8 @@ if __name__ == '__main__':
     arg_parser.add_argument('--split', action='store_true',
                             help='output multiple modules to separate files')
     args = arg_parser.parse_args()
-    sys.exit(main(args))
+    return main(args)
+
+
+if __name__ == '__main__':
+    sys.exit(main_cli())