94 lines
4.0 KiB
Python
94 lines
4.0 KiB
Python
import random
|
|
|
|
class Generators():
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
# Returns a list of all letter pairs in the text
|
|
def make_pairs(self, corpus):
|
|
for i in range(len(corpus)-1):
|
|
yield (corpus[i], corpus[i+1])
|
|
|
|
# Returns a list of all letter triplets in the text
|
|
def make_triplets(self, corpus):
|
|
for i in range(len(corpus)-2):
|
|
yield (corpus[i], corpus[i+1], corpus[i+2])
|
|
|
|
# Generates a random name
|
|
async def nameGen(self, ctx):
|
|
# Makes a list of all names from "names.txt"
|
|
names = open('resources/names.txt', encoding='utf8').read()
|
|
corpus = list(names)
|
|
|
|
# Makes a list of pairs
|
|
pairs = self.make_pairs(corpus)
|
|
triplets = self.make_triplets(corpus)
|
|
|
|
letter_dict = {}
|
|
|
|
# Makes a dictionary of all letters that come after all other letters
|
|
for letter_1, letter_2 in pairs:
|
|
if letter_1 in letter_dict.keys():
|
|
letter_dict[letter_1].append(letter_2)
|
|
else:
|
|
letter_dict[letter_1] = [letter_2]
|
|
|
|
for letter_1, letter_2, letter_3 in triplets:
|
|
if letter_1+letter_2 in letter_dict.keys():
|
|
letter_dict[letter_1+letter_2].append(letter_3)
|
|
else:
|
|
letter_dict[letter_1+letter_2] = [letter_3]
|
|
|
|
# Choses a random first letter
|
|
first_letter = random.choice(corpus)
|
|
|
|
# Makes sure the first letter is not something a name can't start with.
|
|
while first_letter.islower() or first_letter == " " or first_letter == "-" or first_letter == "\n":
|
|
first_letter = random.choice(corpus)
|
|
|
|
# Starts the name
|
|
chain = [first_letter]
|
|
|
|
# Picks second letter
|
|
second_letter = random.choice(letter_dict[chain[-1]])
|
|
|
|
while second_letter == "\n":
|
|
second_letter = random.choice(letter_dict[chain[-1]])
|
|
|
|
chain.append(second_letter)
|
|
|
|
done = False
|
|
|
|
# Creates the name one letter at a time
|
|
while done == False:
|
|
if random.randint(1,10) > 1:
|
|
try:
|
|
new_letter = random.choice(letter_dict[chain[-2]+chain[-1]])
|
|
except KeyError():
|
|
new_letter = random.choice(letter_dict[chain[-1]])
|
|
else:
|
|
new_letter = random.choice(letter_dict[chain[-1]])
|
|
chain.append(new_letter)
|
|
# Ends name if the name ends
|
|
if new_letter == "\n":
|
|
done = True
|
|
genName = "".join(chain)
|
|
self.bot.log("Generated "+genName[:-1])
|
|
|
|
# Returns the name
|
|
await ctx.send(genName)
|
|
|
|
# Generates a random tavern name
|
|
async def tavernGen(self, ctx):
|
|
# Lists first parts, second parts and third parts of tavern names
|
|
fp = ["The Silver","The Golden","The Staggering","The Laughing","The Prancing","The Gilded","The Running","The Howling","The Slaughtered","The Leering","The Drunken","The Leaping","The Roaring","The Frowning","The Lonely","The Wandering","The Mysterious","The Barking","The Black","The Gleaming","The Tap-Dancing","The Sad","The Sexy","The Artificial","The Groovy","The Merciful","The Confused","The Pouting","The Horny","The Okay","The Friendly","The Hungry","The Handicapped","The Fire-breathing","The One-Eyed","The Psychotic","The Mad","The Evil","The Idiotic","The Trusty","The Busty"]
|
|
sp = ["Eel","Dolphin","Dwarf","Pegasus","Pony","Rose","Stag","Wolf","Lamb","Demon","Goat","Spirit","Horde","Jester","Mountain","Eagle","Satyr","Dog","Spider","Star","Dad","Rat","Jeremy","Mouse","Unicorn","Pearl","Ant","Crab","Penguin","Octopus","Lawyer","Ghost","Toad","Handjob","Immigrant","SJW","Dragon","Bard","Sphinx","Soldier","Salmon","Owlbear","Kite","Frost Giant","Arsonist"]
|
|
tp = [" Tavern"," Inn","","","","","","","","",""]
|
|
|
|
# Picks one of each
|
|
genTav = random.choice(fp)+" "+random.choice(sp)+random.choice(tp)
|
|
self.bot.log("Generated "+genTav)
|
|
|
|
# Return the name
|
|
await ctx.send(genTav)
|