diff options
author | Lars Wirzenius <liw@liw.fi> | 2011-02-03 18:53:57 +0000 |
---|---|---|
committer | Lars Wirzenius <liw@liw.fi> | 2011-02-03 18:53:57 +0000 |
commit | 37178026258a56b50910d42196a0d50c13092a74 (patch) | |
tree | bd186cf812fb10fac01e5ece1c984f307edce19d /fix-shebang | |
parent | e5bd4da735e18e824d923720b7fa5d8591917c47 (diff) | |
download | extrautils-37178026258a56b50910d42196a0d50c13092a74.tar.gz |
Add fix-shebang.
Diffstat (limited to 'fix-shebang')
-rwxr-xr-x | fix-shebang | 52 |
1 files changed, 52 insertions, 0 deletions
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 <http://www.gnu.org/licenses/>. + + +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() |