eryn/ubot2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Switch to random.sample for lore instead of random.choice

Browse source
This commit is contained in:
Eryn Wells 2018-08-27 01:05:23 +00:00
parent 966e0fba61
commit 599dff39a8
1 changed files with 4 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
plugins/lore.py
View file

@ -166,13 +166,10 @@ def _lore(channel, count):
pins = channel.saved_pins
if not pins:
return None
if len(pins) < count:
return [_extract_lore(p) for p in pins]
out_lore = set()
while len(out_lore) < count:
random_lore = random.choice(pins)
lore = _extract_lore(random_lore)
out_lore.add(lore)
try:
out_lore = set(_extract_lore(l) for l in random.sample(pins, count))
except ValueError:
out_lore = [_extract_lore(p) for p in pins]
return out_lore
def _scribe():