Staging
v0.5.1
https://github.com/python/cpython
Raw File
Tip revision: dcd4c4f9c9f9fb75ad4c1f138a285fbe19a9cb61 authored by Ɓukasz Langa on 23 March 2020, 16:19:13 UTC
Python 3.9.0a5
Tip revision: dcd4c4f
fixps.py
#!/usr/bin/env python3

# Fix Python script(s) to reference the interpreter via /usr/bin/env python.
# Warning: this overwrites the file without making a backup.

import sys
import re


def main():
    for filename in sys.argv[1:]:
        try:
            f = open(filename, 'r')
        except IOError as msg:
            print(filename, ': can\'t open :', msg)
            continue
        with f:
            line = f.readline()
            if not re.match('^#! */usr/local/bin/python', line):
                print(filename, ': not a /usr/local/bin/python script')
                continue
            rest = f.read()
        line = re.sub('/usr/local/bin/python',
                      '/usr/bin/env python', line)
        print(filename, ':', repr(line))
        with open(filename, "w") as f:
            f.write(line)
            f.write(rest)

if __name__ == '__main__':
    main()
back to top