mirror of
https://github.com/majmongoose/PageBot.git
synced 2024-11-03 01:01:16 -04:00
Update pagebot.py
This commit is contained in:
parent
addc49269d
commit
d15fd53c86
62
pagebot.py
62
pagebot.py
@ -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}')
|
||||
|
Loading…
Reference in New Issue
Block a user