Move code into main()

2016-03-29 20:11:37 -07:00
parent 5f04505675
commit 19de0f1038
3 changed files with 26 additions and 12 deletions
--- a/wrap_file.py
+++ b/wrap_file.py
@@ -49,7 +49,12 @@ class Wrapper(object):
    }


-wrapper = Wrapper(FLAGS.key, FLAGS.cert, FLAGS.other_certs)
-wrapped = wrapper.Wrap(sys.stdin.read())
-json.dump(wrapped, sys.stdout, sort_keys=True, indent=4)
-sys.stdout.write('\n')
+def main():
+  wrapper = Wrapper(FLAGS.key, FLAGS.cert, FLAGS.other_certs)
+  wrapped = wrapper.Wrap(sys.stdin.read())
+  json.dump(wrapped, sys.stdout, sort_keys=True, indent=4)
+  sys.stdout.write('\n')
+
+
+if __name__ == '__main__':
+  main()