blob: 861c83ced90f241e301fa45ab2febc39c41e7232 [file] [log] [blame]
Benjamin Peterson90f5ba52010-03-11 22:53:45 +00001#! /usr/bin/env python3
Guido van Rossumce858271998-09-14 15:43:38 +00002
3"Replace tabs with spaces in argument files. Print names of changed files."
4
5import os
6import sys
Guido van Rossumce858271998-09-14 15:43:38 +00007import getopt
Alexander Belopolsky73f382d2010-11-30 17:30:43 +00008import tokenize
Florent Xiclunae4a33802010-08-09 12:24:20 +00009
Guido van Rossumce858271998-09-14 15:43:38 +000010def main():
11 tabsize = 8
12 try:
13 opts, args = getopt.getopt(sys.argv[1:], "t:")
14 if not args:
Collin Wintera817e582007-08-22 23:05:06 +000015 raise getopt.error("At least one file argument required")
Guido van Rossumb940e112007-01-10 16:19:56 +000016 except getopt.error as msg:
Collin Winter6afaeb72007-08-03 17:06:41 +000017 print(msg)
18 print("usage:", sys.argv[0], "[-t tabwidth] file ...")
Guido van Rossumce858271998-09-14 15:43:38 +000019 return
Guido van Rossum8fd0f141998-10-20 15:33:33 +000020 for optname, optvalue in opts:
21 if optname == '-t':
22 tabsize = int(optvalue)
Guido van Rossumce858271998-09-14 15:43:38 +000023
Andrew M. Kuchlingac6df952003-05-13 18:14:25 +000024 for filename in args:
25 process(filename, tabsize)
Guido van Rossumce858271998-09-14 15:43:38 +000026
Florent Xiclunae4a33802010-08-09 12:24:20 +000027
Georg Brandla9afb682010-10-21 12:49:28 +000028def process(filename, tabsize, verbose=True):
Guido van Rossumce858271998-09-14 15:43:38 +000029 try:
Alexander Belopolsky73f382d2010-11-30 17:30:43 +000030 with tokenize.open(filename) as f:
Florent Xiclunae4a33802010-08-09 12:24:20 +000031 text = f.read()
Alexander Belopolsky73f382d2010-11-30 17:30:43 +000032 encoding = f.encoding
Guido van Rossumb940e112007-01-10 16:19:56 +000033 except IOError as msg:
Collin Winter6afaeb72007-08-03 17:06:41 +000034 print("%r: I/O error: %s" % (filename, msg))
Guido van Rossumce858271998-09-14 15:43:38 +000035 return
Walter Dörwaldaaab30e2002-09-11 20:36:02 +000036 newtext = text.expandtabs(tabsize)
Guido van Rossumce858271998-09-14 15:43:38 +000037 if newtext == text:
38 return
Andrew M. Kuchlingac6df952003-05-13 18:14:25 +000039 backup = filename + "~"
Guido van Rossumce858271998-09-14 15:43:38 +000040 try:
41 os.unlink(backup)
Andrew Svetlov8b33dd82012-12-24 19:58:48 +020042 except OSError:
Guido van Rossumce858271998-09-14 15:43:38 +000043 pass
44 try:
Andrew M. Kuchlingac6df952003-05-13 18:14:25 +000045 os.rename(filename, backup)
Andrew Svetlov8b33dd82012-12-24 19:58:48 +020046 except OSError:
Guido van Rossumce858271998-09-14 15:43:38 +000047 pass
Alexander Belopolsky73f382d2010-11-30 17:30:43 +000048 with open(filename, "w", encoding=encoding) as f:
Florent Xiclunae4a33802010-08-09 12:24:20 +000049 f.write(newtext)
Georg Brandla9afb682010-10-21 12:49:28 +000050 if verbose:
51 print(filename)
Guido van Rossumce858271998-09-14 15:43:38 +000052
Florent Xiclunae4a33802010-08-09 12:24:20 +000053
Guido van Rossumce858271998-09-14 15:43:38 +000054if __name__ == '__main__':
55 main()