cpython/Tools/patchcheck/untabify.py

56 lines
1.3 KiB
Python
Raw Normal View History

#! /usr/bin/env python3
1998-09-14 12:43:38 -03:00
"Replace tabs with spaces in argument files. Print names of changed files."
import os
import sys
import getopt
import tokenize
2010-08-09 09:24:20 -03:00
1998-09-14 12:43:38 -03:00
def main():
tabsize = 8
try:
opts, args = getopt.getopt(sys.argv[1:], "t:")
if not args:
2007-08-22 20:05:06 -03:00
raise getopt.error("At least one file argument required")
except getopt.error as msg:
print(msg)
print("usage:", sys.argv[0], "[-t tabwidth] file ...")
1998-09-14 12:43:38 -03:00
return
for optname, optvalue in opts:
if optname == '-t':
tabsize = int(optvalue)
1998-09-14 12:43:38 -03:00
for filename in args:
process(filename, tabsize)
1998-09-14 12:43:38 -03:00
2010-08-09 09:24:20 -03:00
def process(filename, tabsize, verbose=True):
1998-09-14 12:43:38 -03:00
try:
with tokenize.open(filename) as f:
2010-08-09 09:24:20 -03:00
text = f.read()
encoding = f.encoding
except IOError as msg:
print("%r: I/O error: %s" % (filename, msg))
1998-09-14 12:43:38 -03:00
return
newtext = text.expandtabs(tabsize)
1998-09-14 12:43:38 -03:00
if newtext == text:
return
backup = filename + "~"
1998-09-14 12:43:38 -03:00
try:
os.unlink(backup)
except OSError:
1998-09-14 12:43:38 -03:00
pass
try:
os.rename(filename, backup)
except OSError:
1998-09-14 12:43:38 -03:00
pass
with open(filename, "w", encoding=encoding) as f:
2010-08-09 09:24:20 -03:00
f.write(newtext)
if verbose:
print(filename)
1998-09-14 12:43:38 -03:00
2010-08-09 09:24:20 -03:00
1998-09-14 12:43:38 -03:00
if __name__ == '__main__':
main()