-
Notifications
You must be signed in to change notification settings - Fork 0
/
generator.py
92 lines (71 loc) · 2.44 KB
/
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
from os import system
import json
class Option:
def __init__(self, name, chance):
self.name = name
self.chance = chance
self.enfants = []
self.parent = None
def __str__(self):
return f"{self.name} -- {self.chance}"
def __repr__(self):
return f"{self.name} -- {self.chance}"
def main():
system("clear")
tree = {}
other_tree = []
nb_epreuves = convertInt(input("Nombre d'epreuves: "))
for i in range(nb_epreuves):
other_tree.append([])
if i == 0: #premiere fois
issues = convertInt(input(f"Nombre d'issues pour l'epreuve #{i+1}: "))
for j in range(issues):
issue = input(f"Issue #{j+1}: ")
chance = convertFloat(input(f"Chance de l'issue #{j+1}: "))
add = Option(issue, chance)
other_tree[i].append(add)
else:
for count, j in enumerate(other_tree[i-1]):
issues = convertInt(input(f"Nombre d'issues pour l'epreuve #{i+1}, dependant de '{other_tree[i-1][count]}': "))
for k in range(issues):
issue = input(f"Issue #{k+1}, dependant de '{other_tree[i-1][count]}': ")
chance = convertFloat(input(f"Chance de l'issue #{k+1}, dependant de '{other_tree[i-1][count]}': "))
add = Option(issue, chance)
add.parent = other_tree[i-1][count]
other_tree[i-1][count].enfants.append(add)
other_tree[i].append(add)
print(other_tree)
for i in other_tree[0]:
tree[i.name] = {}
tree[i.name]["chance"] = i.chance
tree[i.name]["enfants"] = createDictFromChilds(i)
printDict(tree)
def createDictFromChilds(d):
temp = {}
if not len(d.enfants) == 0:
for i in d.enfants:
temp[i.name] = {}
temp[i.name]["chance"] = i.chance
temp[i.name]["enfants"] = createDictFromChilds(i)
return temp
else:
return {}
def convertInt(val):
try:
return int(val)
except ValueError:
print("Erreur")
exit(1)
def convertFloat(val):
try:
return float(val)
except ValueError:
print("Erreur")
exit(1)
def printDict(d):
nice = json.dumps(d, indent=4)
print(nice)
with open("tree.json", "w+") as f:
f.write(nice)
if __name__ == "__main__":
main()