1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""The code base had inconsistent usage of tabs/spaces for indenting in Lua
files. Spaces were more prominent - and I prefer them over tabs. So I wrote
this small script to fix leading tabs in Lua files to spaces.
It also saves files in unix file endings ("\r\n") and strips empty lines at the
end of files and whitespace characters at the end of lines.
"""
import argparse
import os
import re
import sys
LEADING_TABS = re.compile(r'^\s*\t+\s*')
PYTHON3 = sys.version_info >= (3, 0)
SPACES_PER_TAB = 3
def parse_args():
p = argparse.ArgumentParser(description=
"Fix common whitespace errors in Lua files. Recurses over all Lua files."
)
return p.parse_args()
def read_text_file(filename):
"""Reads the contens of a text file."""
if PYTHON3:
return open(filename, 'r', encoding='utf-8').read()
else:
return open(filename, 'r').read().decode('utf-8')
def write_text_file(filename, content):
"""Writes 'content' into a text file."""
if PYTHON3:
open(filename, 'w', encoding='utf-8').write(content)
else:
open(filename, 'w').write(content.encode('utf-8'))
def find_lua_files():
for (dirpath, _, filenames) in os.walk("."):
for filename in filenames:
if os.path.splitext(filename)[-1].lower() == ".lua":
yield os.path.join(dirpath, filename)
def main():
parse_args()
if not os.path.isdir("src") or not os.path.isdir("utils"):
print("CWD is not the root of the repository.")
return 1
for filename in find_lua_files():
print "Fixing %r" % filename
lines = read_text_file(filename).strip().split("\n")
new_lines = []
for line in lines:
m = LEADING_TABS.match(line)
if m is not None:
line = line[m.start():m.end()].expandtabs(SPACES_PER_TAB) + line[m.end():]
new_lines.append(line.rstrip() + "\n")
write_text_file(filename, "".join(new_lines))
return 0
if __name__ == "__main__":
sys.exit(main())
|
