-
Notifications
You must be signed in to change notification settings - Fork 0
/
build_default_list.py
46 lines (41 loc) · 1.47 KB
/
build_default_list.py
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
# This script generates the default_name_list.lua file from the text files in the list subdirectory
import os
# Makes a "[[string]]""[=[string]=]"/ style quote where the amount of equal signs is equal to the depth given
def makequotedepth(c, depth):
s = c
for x in range(0, depth):
s += "="
s += c
return s
# Finds out the needed quote depth in order to encapsulate the given string s without escaping anything inside
def getquotedepth(s):
depth = 0
while True:
opener = makequotedepth("[", depth)
closer = makequotedepth("]", depth)
if opener in s or closer in s:
depth += 1
continue
else:
break
return depth
# Joins all text files in the list subdirectory
directory = "lists"
joinedText = ""
for filename in os.listdir(directory):
if filename.endswith(".txt"):
fullname = os.path.join(directory, filename)
print(filename)
f = open(fullname, "r")
joinedText += (f.read() + "\n")
continue
else:
continue
# Generate the .lua file
quoteDepth = getquotedepth(joinedText)
opener = makequotedepth("[", quoteDepth) + "\n"
closer = "\n" + makequotedepth("]", quoteDepth)
joinedText = ("return --[[This list is autogenerated by build_default_list.py. If you wish to edit it, please make changes to the files in the list subdirectory.]] " + opener + joinedText + closer)
f = open("scripts/default_name_list.lua","w+")
f.write(joinedText)
f.close()