From 538aeee2b213c5cd2882579599f732e78a0be5d2 Mon Sep 17 00:00:00 2001
From: Charley Peng <cpeng92@gmail.com>
Date: Thu, 29 May 2014 09:40:58 +1000
Subject: [PATCH] adding creation of batch file

---
 setup.py | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 50 insertions(+), 4 deletions(-)

diff --git a/setup.py b/setup.py
index 72eec5d..7e1a232 100644
--- a/setup.py
+++ b/setup.py
@@ -1,10 +1,54 @@
 from distutils.core import setup
 import os
+from os.path import join as pjoin, splitext, split as psplit
+from distutils.command.install_scripts import install_scripts
+from distutils import log
+
+BAT_TEMPLATE = \
+r"""@echo off
+REM wrapper to use shebang first line of {FNAME}
+set mypath=%~dp0
+set pyscript="%mypath%{FNAME}"
+set /p line1=<%pyscript%
+if "%line1:~0,2%" == "#!" (goto :goodstart)
+echo First line of %pyscript% does not start with "#!"
+exit /b 1
+:goodstart
+set py_exe=%line1:~2%
+call %py_exe% %pyscript% %*
+"""
+
+
+class my_install_scripts(install_scripts):
+    def run(self):
+        install_scripts.run(self)
+        if not os.name == "nt":
+            return
+        for filepath in self.get_outputs():
+            # If we can find an executable name in the #! top line of the script
+            # file, make .bat wrapper for script.
+            with open(filepath, 'rt') as fobj:
+                first_line = fobj.readline()
+            if not (first_line.startswith('#!') and
+                    'python' in first_line.lower()):
+                log.info("No #!python executable found, skipping .bat "
+                         "wrapper")
+                continue
+            pth, fname = psplit(filepath)
+            froot, ext = splitext(fname)
+            bat_file = pjoin(pth, froot + '.bat')
+            bat_contents = BAT_TEMPLATE.replace('{FNAME}', fname)
+            log.info("Making %s wrapper for %s" % (bat_file, filepath))
+            if self.dry_run:
+                continue
+            with open(bat_file, 'wt') as fobj:
+                fobj.write(bat_contents)
+
 
 data = [
-       ('/usr/share/zsh/site-functions', ['cheat/autocompletion/_cheat.zsh']),
-       ('/etc/bash_completion.d'       , ['cheat/autocompletion/cheat.bash']),
-       ('/usr/share/fish/completions'  , ['cheat/autocompletion/cheat.fish'])
+         ('/usr/share/zsh/site-functions', ['cheat/autocompletion/_cheat.zsh']),
+         ('/etc/bash_completion.d'       , ['cheat/autocompletion/cheat.bash']),
+         ('/usr/share/fish/completions'  , ['cheat/autocompletion/cheat.fish'])
        ]
 
 if os.name == 'nt':
@@ -34,5 +78,7 @@ setup(
     install_requires = [
         'docopt >= 0.6.1',
         'pygments >= 1.6.0',
-    ]
+    ],
+    cmdclass = {'install_scripts': my_install_scripts}
+
 )