fix bug with sources that only have lists

This commit is contained in:
evilchili
2023-12-22 23:22:17 -08:00
parent 15904a51d2
commit e579d11cfa
2 changed files with 13 additions and 8 deletions
+2 -4
View File
@@ -128,10 +128,7 @@ class DataSource:
]
else:
# If the option is either a list or a string, just select it.
if rand:
choices = [random.choice(self.data[option])]
else:
choices = self.data[option]
choices = self.data[option]
for choice in choices:
# If the randomly-selected choice is a dict, choose a random item and return a list consisting
@@ -153,6 +150,7 @@ class DataSource:
flattened.append(choice)
return flattened
class RollTable:
"""
Generate a roll table using weighted distributions of random options.