Uploading to show the Python Discord server my buggy code
This commit is contained in:
289
DiscordBot/madlibsDiscord.py
Executable file
289
DiscordBot/madlibsDiscord.py
Executable file
@@ -0,0 +1,289 @@
|
||||
#!/usr/bin/python
|
||||
#General purpose Discord Bot developed by Caleb Fontenot (CCF_100)
|
||||
#Written in python.
|
||||
# Toggle me for debugging
|
||||
debug = 1
|
||||
# Displays warning message
|
||||
testing = 1
|
||||
# Import the libraries we will use
|
||||
#from mega import Mega
|
||||
import tracemalloc
|
||||
from concurrent.futures import ProcessPoolExecutor
|
||||
from datetime import datetime
|
||||
from gtts import gTTS
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
import re
|
||||
import sys
|
||||
import random
|
||||
import os
|
||||
import asyncio, threading
|
||||
import ffmpeg
|
||||
#import talkey
|
||||
|
||||
tracemalloc.start()
|
||||
|
||||
async def madlibsLoop():
|
||||
# In the future we will detect what channel we were summoned in, but for now:
|
||||
activeChannel = 656233549837631508
|
||||
#Init tts, and connect voice to channel, reinit connection if broken
|
||||
voiceChannel = client.get_channel(682688245964079127)
|
||||
voice = await voiceChannel.connect()
|
||||
if voice.is_connected() == True:
|
||||
await voice.disconnect()
|
||||
voice = await voiceChannel.connect()
|
||||
# Set bot presence
|
||||
await client.change_presence(activity=discord.Game(name='madlibs.py'))
|
||||
# Introduce yourself
|
||||
channel = client.get_channel(656233549837631508)
|
||||
await channel.send("**<<madlibsDiscord.py <:python:656239601723113472> - Written by CCF_100>>**")
|
||||
await channel.send("Initial project started on **July 13, 2019**")
|
||||
await channel.send("Discord Bot started on **December 16, 2019**")
|
||||
# Notify if verbose
|
||||
if debug == 1:
|
||||
await channel.send("Debug mode is enabled! Being verbose!")
|
||||
# Now on to business!
|
||||
# Load files
|
||||
async with channel.typing():
|
||||
f = open('storyCount.txt', 'r')
|
||||
StoryCount = f.read()
|
||||
IntStoryCount = int(StoryCount)
|
||||
await channel.send("Detected "+str(IntStoryCount)+" stories")
|
||||
# Randomly pick what story we will use
|
||||
story = random.randint(1, IntStoryCount)
|
||||
|
||||
#Declare vars
|
||||
storyContentStr = []
|
||||
storyNameStr = []
|
||||
# Alright, let's get the data from stories.txt
|
||||
i = 1
|
||||
f = open('stories.txt', 'r')
|
||||
for line in f.readlines():
|
||||
if i % 2 == 0 :
|
||||
storyContent = line
|
||||
storyContentStr.append(storyContent)
|
||||
else:
|
||||
storyName = line
|
||||
storyNameStr.append(storyName)
|
||||
i+=1
|
||||
f.close()
|
||||
await channel.send(storyNameStr)
|
||||
# Print current story title, but remove the brackets first
|
||||
filteredTitle = re.findall(r'<(.*?)>', storyNameStr[story-1])
|
||||
|
||||
# print the first result
|
||||
await channel.send("Current story title is "+'"'+str(filteredTitle[0])+'"'+'\n')
|
||||
# Alright, now onto the tricky part. We need to filter out all of the bracketed words in stories.txt, putting them into a list, replacing them with incremental strings. We also need to count how many there are for later.
|
||||
# Pull all of the items with the <> brackets
|
||||
filtered = re.findall(r'<(.*?)>', storyContentStr[story-1])
|
||||
# We got them!
|
||||
if debug == 1:
|
||||
await channel.send(str(filtered))
|
||||
# Now we need to count them
|
||||
replacedNumber = len(filtered)
|
||||
|
||||
# Run a loop to get the words
|
||||
|
||||
replaceList = []
|
||||
#replaceList =['', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20', '21', '22', '23', '24']
|
||||
replaceList.append("")
|
||||
await channel.send(str("Type a noun, verb, adjective, or adverb depending on what it asks you, followed by enter."))
|
||||
|
||||
for loopCount in range(replacedNumber):
|
||||
print("Times looped: "+str(loopCount))
|
||||
#Wait for user to reply
|
||||
await channel.send("Give me a(n) "+"**"+str(filtered[loopCount])+"**"+": ")
|
||||
# Push text to gTTS and save it to a file
|
||||
tts = gTTS(text="Give me a(n) "+str(filtered[loopCount])+": ")
|
||||
os.system("rm badCurrentTTS.mp3")
|
||||
os.system("rm currentTTS.wav")
|
||||
tts.save("badCurrentTTS.mp3")
|
||||
# gTTS is stupid and gives us a file that discord.py doesn't understand, so we have to convert it
|
||||
(
|
||||
ffmpeg
|
||||
.input('badCurrentTTS.mp3')
|
||||
.output('currentTTS.wav', audio_bitrate=48000, format='wav', sample_fmt='s16', ac='2')
|
||||
.run()
|
||||
)
|
||||
if voice.is_playing() == True:
|
||||
print("Audio is playing! Stopping playback!"+'\n')
|
||||
voice.stop()
|
||||
print("Attempting to play audio"+'\n')
|
||||
voice.play(discord.FFmpegPCMAudio("currentTTS.wav"))
|
||||
message = await self.bot.wait_for('message', check=lambda messages: message.author.id == ctx.author.id and ctx.channel.id == message == ctx.message.id, timeout=30.0)
|
||||
replaceVar = message.content
|
||||
print("You gave me: "+replaceVar)
|
||||
replaceList.append(replaceVar)
|
||||
print(replaceList)
|
||||
# Run a loop to replace the words
|
||||
|
||||
await channel.send("Replacing Words...")
|
||||
|
||||
# Split the Story Content into a list
|
||||
storyContentList = re.split(r'<.*?>', storyContentStr[story-1])
|
||||
# Count the items in the list
|
||||
storyContentCount = len(storyContentList)
|
||||
x = 0
|
||||
for loopCount in range(storyContentCount):
|
||||
#print(storyContentList[loopCount])
|
||||
storyContentList.insert(x, replaceList[loopCount])
|
||||
x = x+2
|
||||
x = 0
|
||||
|
||||
# Merge lists into a string
|
||||
generatedStory = ""
|
||||
generatedStory = generatedStory.join(storyContentList)
|
||||
# Determine file name for file output
|
||||
now = datetime.now()
|
||||
currentDate = now.strftime("%d-%m-%Y-%H:%M:%S")
|
||||
saveFile = 'saved stories/generatedStory-'+currentDate
|
||||
# Send Story to Discord
|
||||
await channel.send(generatedStory)
|
||||
await channel.send("Processing TTS, please wait!")
|
||||
async with channel.typing():
|
||||
tts = gTTS(text=generatedStory+"This story was generated by CCF_100's MadLibs.py", lang='en')
|
||||
os.system("rm badCurrentStory.mp3")
|
||||
tts.save("badCurrentStory.mp3")
|
||||
# gTTS is stupid and gives us a file that discord.py doesn't understand, so we have to convert it
|
||||
(
|
||||
ffmpeg
|
||||
.input('badCurrentStory.mp3')
|
||||
.output(saveFile+'.mp3', audio_bitrate=48000, format='wav', sample_fmt='s16', ac='2')
|
||||
.run()
|
||||
)
|
||||
if voice.is_playing() == True:
|
||||
print("Audio is playing! Stopping playback!"+'\n')
|
||||
voice.stop()
|
||||
print("Attempting to play audio"+'\n')
|
||||
voice.play(discord.FFmpegPCMAudio(saveFile+".mp3"))
|
||||
#exit()
|
||||
#Alright! We're done! Let's save the story to a file
|
||||
if os.path.exists("saved stories"):
|
||||
pass
|
||||
else:
|
||||
os.system("mkdir \"saved stories\"")
|
||||
print("Saving story to .txt file")
|
||||
await channel.send("Saving story to .txt file")
|
||||
async with channel.typing():
|
||||
file = open(saveFile+'.txt', 'w+')
|
||||
line_offset = []
|
||||
offset = 0
|
||||
for line in file:
|
||||
line_offset.append(offset)
|
||||
offset += len(line)
|
||||
file.seek(0)
|
||||
file.write(filteredTitle[0]+'\n'+'\n')
|
||||
file.write(generatedStory)
|
||||
file.write('\n'+"Generated by Caleb Fontenot\'s madlibs.py")
|
||||
file.close()
|
||||
#Send generated .txt file to Discord
|
||||
await channel.send("Sending .txt file...")
|
||||
discordFile = discord.File(saveFile+'.txt', filename="generatedStory.txt")
|
||||
await channel.send(file=discordFile)
|
||||
#Send generated .mp3 file to Discord
|
||||
mp3File = saveFile+'.mp3'
|
||||
def file_size(fname):
|
||||
statinfo = os.stat(fname)
|
||||
return statinfo.st_size
|
||||
print("MP3 is "+str(file_size(mp3File))+" bytes.")
|
||||
await channel.send("Converted WAV is "+str(file_size(mp3File))+" bytes.")
|
||||
if int(file_size(mp3File)) <= int(8389999):
|
||||
# File is over 8 MiB! This will fail if we send the file corrected for transmission via Discord's voice chat. Let's send the original instead.
|
||||
discordFile = discord.File("badCurrentStory.mp3", filename=saveFile+'.mp3')
|
||||
await channel.send(file=discordFile)
|
||||
else:
|
||||
discordFile = discord.File(saveFile+'.wav', filename=saveFile+'.wav')
|
||||
await channel.send(file=discordFile)
|
||||
|
||||
async def messageListening():
|
||||
global messageslist, messageAuthorList, logMessages
|
||||
print("Now Listening for messages...")
|
||||
messagesList= []
|
||||
messageAuthorList = []
|
||||
print("Loop start!")
|
||||
x = 0
|
||||
while True:
|
||||
async def on_message(message):
|
||||
raw_message = await client.wait_for('message')
|
||||
if logMessages == True:
|
||||
print("Loop Count: "+str(x))
|
||||
messagesList.append(raw_message.content)
|
||||
messageAuthorList.append(raw_message.author.nick)
|
||||
if logMessages == True:
|
||||
print("Message from "+messageAuthorList[x]+": "+messagesList[x])
|
||||
x += 1
|
||||
#await client.process_commands(message)
|
||||
#Setup Discord functions and announce on discord that we are ready
|
||||
|
||||
class MyClient(discord.Client):
|
||||
async def on_ready(self):
|
||||
#print(loop)
|
||||
print('Logged on as', self.user)
|
||||
channel = client.get_channel(656233549837631508)
|
||||
await channel.send("(Yet to be named) Discord bot, successfully connected!")
|
||||
await channel.send("Developed by CCF_100")
|
||||
if testing == 1:
|
||||
await channel.send("This bot is currently being worked on! Please don't start a game!")
|
||||
await channel.send("Run `mad!madlibs` to start madlibs")
|
||||
print("Ready!")
|
||||
|
||||
async def on_message(self, message, pass_context=True):
|
||||
if message.content == 'mad!madlibs':
|
||||
channel = client.get_channel(656233549837631508)
|
||||
await madlibsLoop()
|
||||
await channel.send("Done!")
|
||||
#Turn on message logging
|
||||
async def on_message(self, message, pass_context=True):
|
||||
if message.content == 'mad!logMessagesOn':
|
||||
pass
|
||||
if message.author.id == 294976590658666497:
|
||||
channel = client.get_channel(656233549837631508)
|
||||
await channel.send("Logging messages to console.")
|
||||
global logMessages
|
||||
logMessages = True
|
||||
else:
|
||||
channel = client.get_channel(656233549837631508)
|
||||
await channel.send("You are not authorized to use this command! Only @CCF_100#1050 may use this command!")
|
||||
#Turn off message logging
|
||||
async def on_message(self, message, pass_context=True):
|
||||
if message.content == 'mad!logMessagesOff':
|
||||
pass
|
||||
if message.author.id == 294976590658666497:
|
||||
channel = client.get_channel(656233549837631508)
|
||||
await channel.send("Stopping message logging!")
|
||||
global logMessages
|
||||
logMessages = False
|
||||
else:
|
||||
channel = client.get_channel(656233549837631508)
|
||||
await channel.send("You are not authorized to use this command! Only @CCF_100#1050 may use this command!")
|
||||
#Calls message listening function.
|
||||
logMessages = False
|
||||
loop = asyncio.get_event_loop()
|
||||
asyncio.create_task(messageListening())
|
||||
print("lol3214")
|
||||
#Disconnect Voice
|
||||
#await asyncio.sleep(60)
|
||||
#voiceChannel = client.get_channel(682688245964079127)
|
||||
#await voice.disconnect()
|
||||
|
||||
#Run main loop
|
||||
# The Discord bot ID isn't stored in this script for security reasons, so we have to go get it
|
||||
f = open('botID.txt', 'r')
|
||||
BotID = f.read()
|
||||
#Cleanup from previous session
|
||||
#os.system("rm badCurrentTTS.mp3")
|
||||
#os.system("rm currentTTS.wav")
|
||||
# Connect Bot To Discord and start running
|
||||
|
||||
client = MyClient()
|
||||
client.run(BotID)
|
||||
exit()
|
||||
|
||||
|
||||
# say the tts
|
||||
#print('\n'+"Processing Text-To-Speech, please wait..."+'\n')
|
||||
#tts = gTTS(text=generatedStory+"This story was generated by Caleb Fontenot's MadLibs.py", lang='en')
|
||||
#tts.save("TTS.mp3")
|
||||
##os.system("play TTS.mp3")
|
||||
#os.system("mv TTS.mp3 "+"\""+saveFile+".mp3"+"\"")
|
||||
#Start Discord Bot loop
|
||||
Reference in New Issue
Block a user