-
Notifications
You must be signed in to change notification settings - Fork 0
/
dcss_challenge_generator.py
263 lines (188 loc) · 7.88 KB
/
dcss_challenge_generator.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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
#This program is under GNU General Public License v3.0
#Made by Krzysztof Kiełczewski, 2017
from random import choice
import sys
SHORTCUT_LENGTH = 2
MAX_COMBOS = 100
TIERS_PER_CHALLENGE = 3
def open_file(filename):
"""Opens file and check if file exists in generator's directory."""
try:
file = open(filename,"r")
except FileNotFoundError:
print("Failure!\nFile not found! Check if name of file is %s"
% filename)
end_program(1)
return file
def end_program(exit_code = 0):
"""Here program ends both successfully or abnormally."""
input("Press Enter to exit.")
sys.exit(exit_code)
def import_to_dictionary(filename):
"""Imports database from file to dictionary."""
print("Importing %s..." % filename[:-13],end="")
file = open_file(filename)
dictionary = {}
iteration = 0
for line in file:
iteration += 1
line = line.strip().split(" ", maxsplit=1)
if len(line[0]) != SHORTCUT_LENGTH:
print("Failure!\n%s (line %d) is not %d-char! Correct it in %s file!"
% (line[0],iteration,SHORTCUT_LENGTH,filename))
end_program(2)
dictionary[line[0]] = line[1]
file.close()
print("%d...Done!" % iteration)
return dictionary
def import_non_available_combos(filename):
"""Imports bad combos database from file to list."""
print("Importing non available combos...",end="")
file = open_file(filename)
list = []
iteration = 0
for line in file:
iteration += 1
line = line.strip()
if len(line) != SHORTCUT_LENGTH*2:
print("Failure!\n%s (line %d) is not %d-char! Correct it in %s file!"
% (line,iteration,SHORTCUT_LENGTH*2,filename))
end_program(2)
list.append(line)
file.close()
print("%d...Done!" % iteration)
return list
def import_challenges(filename):
"""Imports challenges from file to mixed dictionary with lists"""
print("Importing challenges...",end="")
file = open_file(filename)
dictionary = {}
iteration = 0
while True:
#Importing all challenges line by line
challenge_name = file.readline()
if challenge_name[:5] != "Name:":
break
iteration += 1
challenge_name = challenge_name[6:].strip()
challenge_tiers = []
for i in range(TIERS_PER_CHALLENGE):
tier = file.readline()
if tier[:6] != ("Tier%s:" % str(i+1)):
print("Failure!\nExpected 'Tier%s:' data format in %s, challenge No.%d. Correct it!"
% (str(i+1), filename, iteration))
end_program(2)
tier = tier[6:].strip()
challenge_tiers.append(tier)
banned_species = file.readline()
if banned_species[:11] != "BanSpecies:":
print("Failure!\nExpected 'BanSpecies:' data format in %s, challenge No.%d. Correct it!"
% (filename, iteration))
end_program(2)
banned_species = banned_species[11:].strip().split(" ")
banned_backgrounds = file.readline()
if banned_backgrounds[:15] != "BanBackgrounds:":
print("Failure!\nExpected 'BanBackgrounds:' data format in %s, challenge No.%d. Correct it!"
% (filename, iteration))
end_program(2)
banned_backgrounds = banned_backgrounds[15:].strip().split(" ")
dictionary[challenge_name] = [challenge_tiers,banned_species,
banned_backgrounds]
#Skip blank line
file.readline()
file.close()
print("%d...Done!" % iteration)
return dictionary
def ask_number_of_challenges():
"""Here program asks for number of challenge to generate"""
print("\n\nHow many challenges you want to generate?")
print("Enter number from 1 to %d or enter 0 to exit..." % MAX_COMBOS)
while True:
#Checking for valid input
try:
amount = int(input("Your choice: "))
if amount < 0 or amount > MAX_COMBOS:
raise ValueError
else:
break
except ValueError:
print("Invalid value! Try again!")
return amount
def ask_name_of_export_file():
"""Program asks for name of exported file."""
while True:
try:
filename = input("Enter name of txt file for exported challenges: ")
if (filename == "background_database"
or filename == "challenge_database"
or filename == "nocombo_database"
or filename == "species_database"
or filename.lstrip() == ""):
raise ValueError
else:
break
except ValueError:
print("You can't use that file name! Try another one!")
filename += ".txt"
return filename
def generate_challenge(species_db,backgrounds_db,no_combo_db,challenges_db):
"""Here program composes challenge with all required databases."""
chosen_challenge = choice(list(challenges_db.keys()))
if challenges_db[chosen_challenge][2] != [""]:
for banned_background in challenges_db[chosen_challenge][2]:
if banned_background in backgrounds_db:
backgrounds_db.pop(banned_background)
chosen_background = choice(list(backgrounds_db))
if challenges_db[chosen_challenge][1] != [""]:
for banned_species in challenges_db[chosen_challenge][1]:
if banned_species in species_db:
species_db.pop(banned_species)
chosen_species = choice(list(species_db))
while chosen_species+chosen_background in no_combo_db:
species_db.pop(chosen_species)
chosen_species = choice(list(species_db))
chosen_combo = species_db[chosen_species] + " " + backgrounds_db[chosen_background]
new_challenge = [chosen_species+chosen_background,chosen_combo,chosen_challenge]
new_challenge.append(challenges_db[chosen_challenge][0])
return new_challenge
def export_challenges(file,generated_challenges):
"""Here all elements of generated list are exported to output file."""
iteration = 0
for challenge in generated_challenges:
iteration += 1
output = "[Challenge No.%d]\n\n" % iteration
output += "%s (%s) - %s\n\n" % (challenge[1],challenge[0],challenge[2])
for tier in range(TIERS_PER_CHALLENGE):
output += "* Tier %d: %s\n" % (tier+1, challenge[3][tier])
output += "\n"
file.write(output)
#Creating underline for end of export file
line = ""
for i in range(100):
line += "-"
file.write(line)
def main():
"""This is where program starts."""
species_database = import_to_dictionary("species_database.txt")
backgrounds_database = import_to_dictionary("background_database.txt")
no_combo_database = import_non_available_combos("nocombo_database.txt")
challenges_database = import_challenges("challenge_database.txt")
print("\n--WELCOME TO DCSS CHALLENGE GENERATOR!--")
number_of_challenges = ask_number_of_challenges()
if number_of_challenges == 0:
print("You decided not to generate any challenges.")
print("Thank you for using this software!")
end_program(0)
else:
output_challenges = []
for i in range(number_of_challenges):
generated = generate_challenge(species_database, backgrounds_database,
no_combo_database, challenges_database)
output_challenges.append(generated)
output_file = open(ask_name_of_export_file(),"w")
export_challenges(output_file,output_challenges)
output_file.close()
print("\nExport ended successfully!")
print("Thank you for using this software!")
end_program(0)
main()