morgan/Grinch-AP
1
0
Fork 0
mirror of https://github.com/MarioSpore/Grinch-AP.git synced 2025-10-21 12:11:33 -06:00
Code Issues Packages Projects Releases Wiki Activity

User-specified random range (#203)

Browse Source 
* Add user-specified random range for yaml options
This commit is contained in:
Alchav
2022-01-15 19:59:40 -05:00
committed by GitHub
parent 0dc714f947
commit 6a7e1d920a
3 changed files with 43 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
Options.py
View File

@@ -210,6 +210,23 @@ class Range(Option, int):
return cls(int(round(random.triangular(cls.range_start, cls.range_end, cls.range_end), 0)))
elif text == "random-middle":
return cls(int(round(random.triangular(cls.range_start, cls.range_end), 0)))
elif text.startswith("random-range-"):
textsplit = text.split("-")
try:
randomrange = [int(textsplit[len(textsplit)-2]), int(textsplit[len(textsplit)-1])]
except ValueError:
raise ValueError(f"Invalid random range {text} for option {cls.__name__}")
randomrange.sort()
if randomrange[0] < cls.range_start or randomrange[1] > cls.range_end:
raise Exception(f"{randomrange[0]}-{randomrange[1]} is outside allowed range {cls.range_start}-{cls.range_end} for option {cls.__name__}")
if text.startswith("random-range-low"):
return cls(int(round(random.triangular(randomrange[0], randomrange[1], randomrange[0]))))
elif text.startswith("random-range-middle"):
return cls(int(round(random.triangular(randomrange[0], randomrange[1]))))
elif text.startswith("random-range-high"):
return cls(int(round(random.triangular(randomrange[0], randomrange[1], randomrange[1]))))
else:
return cls(int(round(random.randint(randomrange[0], randomrange[1]))))
else:
return cls(random.randint(cls.range_start, cls.range_end))
return cls(int(text))