Update pagebot.py

This commit is contained in:
ceb515 2024-03-25 14:23:03 -04:00
parent addc49269d
commit d15fd53c86

View File

@ -1,5 +1,5 @@
#!/usr/bin/python3
import discord
from discord.ext import commands
import os
import subprocess
import time
@ -9,6 +9,9 @@ from watchdog.observers import Observer
from watchdog.events import FileSystemEventHandler
import speech_recognition as sr
## Initialize the bot with command prefix
bot = commands.Bot(command_prefix='!')
## New File Handler
class MyHandler(FileSystemEventHandler):
def on_created(self, event):
@ -29,7 +32,7 @@ class MyHandler(FileSystemEventHandler):
print("Converting to MP4")
mp4_file = convert_to_mp4(filepath)
print("Sending to Discord")
client.loop.create_task(upload_to_discord(mp4_file,text))
bot.loop.create_task(upload_to_discord(mp4_file,text))
## Convert MP3 to MP4
def convert_to_mp4(mp3_file):
@ -62,8 +65,6 @@ def convert_to_text(mp3_path,mp3_name):
print(f"Error during conversion: {e}")
return ""
## upload to discord
async def upload_to_discord(mp4_file,text):
## Check to make sure conversion worked.
@ -71,7 +72,7 @@ async def upload_to_discord(mp4_file,text):
print("Conversion failed. Skipping upload.")
return
## Get channel to send in
channel = client.get_channel(secrets_file.channel_id)
channel = bot.get_channel(secrets_file.channel_id)
if channel:
filename = os.path.basename(mp4_file)
## Send Video with name
@ -92,39 +93,52 @@ async def upload_to_discord(mp4_file,text):
else:
print(f"Could not find channel with ID {channel}")
def launch_and_watch(program_path):
## Command to restart TTD software manually
@bot.command()
async def restart_ttd(ctx):
if secrets_file.ttd_path:
await ctx.send("Restarting TTD software...")
restart_ttd_process(secrets_file.ttd_path)
else:
await ctx.send("TTD path not configured.")
## Function to restart TTD software
def restart_ttd_process(program_path):
program_directory = os.path.dirname(program_path)
process = subprocess.Popen(program_path, cwd=program_directory)
process.wait()
if process.returncode != 0:
print("TTD has crashed. Relaunching...")
else:
print("TTD has exited normally.")
## Launch and watch TTD
def launch_and_watch_ttd():
while True:
process = subprocess.Popen(program_path,cwd=program_directory)
process = subprocess.Popen(secrets_file.ttd_path)
process.wait()
if process.returncode != 0:
print("TTD has crashed. Relaunching...")
else:
print("TTD has exited normally.")
break
# Wait for a few seconds before relaunching
time.sleep(2)
if __name__ == "__main__":
## initialize watchdogs
## Initialize watchdogs
event_handler = MyHandler()
observer = Observer()
observer.schedule(event_handler, secrets_file.watch_folder, recursive=False)
observer.start()
## Launch TTD
if (secrets_file.ttd_path != ""):
watchdog_thread = threading.Thread(target=launch_and_watch, args=(secrets_file.ttd_path,))
watchdog_thread.start()
## Start watching TTD
if secrets_file.ttd_path:
ttd_thread = threading.Thread(target=launch_and_watch_ttd)
ttd_thread.start()
## initialize discord
intents = discord.Intents.default()
intents.message_content = True
client = discord.Client(intents=intents)
client.run(secrets_file.key)
## Initialize Discord bot
bot.run(secrets_file.key)
try:
while True:
@ -133,9 +147,7 @@ if __name__ == "__main__":
observer.stop()
observer.join()
##discord stuffs
@client.event
## Discord event
@bot.event
async def on_ready():
print(f'We have logged in as {client.user}')
print(f'We have logged in as {bot.user}')