From 37178026258a56b50910d42196a0d50c13092a74 Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Thu, 3 Feb 2011 18:53:57 +0000 Subject: Add fix-shebang. --- fix-shebang | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 52 insertions(+) create mode 100755 fix-shebang (limited to 'fix-shebang') diff --git a/fix-shebang b/fix-shebang new file mode 100755 index 0000000..6d78bb0 --- /dev/null +++ b/fix-shebang @@ -0,0 +1,52 @@ +#!/usr/bin/python +# Copyright 2011 Lars Wirzenius +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . + + +import cliapp +import os +import tempfile + + +class FixShebang(cliapp.Application): + + def add_settings(self): + self.add_string_setting(['shebang'], 'interpreter path and args') + + def process_input(self, arg): + name = self.create_tempfile_for(arg) + temp = open(name, 'w') + shebang = self['shebang'] + if not shebang.startswith('#!'): + shebang = '#!%s' % shebang + temp.write('%s\n' % shebang) + f = open(arg) + line1 = f.readline() + if line1 and not line1.startswith('#!'): + temp.write(line1) + for line in f: + temp.write(line) + f.close() + temp.close() + os.rename(name, arg) + + def create_tempfile_for(self, filename): + fd, name = tempfile.mkstemp(dir=os.path.dirname(filename)) + os.close(fd) + return name + + +if __name__ == '__main__': + FixShebang().run() -- cgit v1.2.1