Have setuptools generate .exe wrappers for scripts on Windows

1 files changed, 10 insertions, 0 deletions

diff --git a/cheetah/CheetahWrapper.py b/cheetah/CheetahWrapper.py
index 96f57d5..9af8f1e 100644
--- a/cheetah/CheetahWrapper.py
+++ b/cheetah/CheetahWrapper.py
@@ -614,6 +614,16 @@ be named according to the same rules as Python modules.""" % tup)
             f.close()
             
 
+# Called when invoked as `cheetah`
+def _cheetah():
+    CheetahWrapper().main()
+
+# Called when invoked as `cheetah-compile`
+def _cheetah_compile():
+    sys.argv.insert(1, "compile")
+    CheetahWrapper().main()
+
+
 ##################################################
 ## if run from the command line
 if __name__ == '__main__':  CheetahWrapper().main()