the start of the webui stuff

This commit is contained in:
phixxy 2024-06-18 10:00:00 -07:00
parent 995bb08a0a
commit d94163e246
6 changed files with 149 additions and 3 deletions

View file

@ -10,7 +10,6 @@ intents.message_content = True
bot = commands.Bot(command_prefix='!', intents=intents)
logger = src.logger.logger_setup()
async def load_cogs(bot: commands.Bot, cog_path: str) -> None:
for cog_file in os.listdir(cog_path):
if cog_file[-3:] == '.py':
@ -27,6 +26,7 @@ async def on_ready():
await utils.delete_all_files("tmp/")
await load_cogs(bot, 'cogs/')
logger.info('We have logged in as {0.user}'.format(bot))
print("Visit http://localhost:5000 to change config!")
except:
logger.warning(f"Error in on_ready")
@ -43,4 +43,4 @@ async def on_message(ctx):
except discord.ext.commands.errors.CommandNotFound:
logger.info("Command not found.")
except Exception as e:
logger.warning(f"Error processing commands: {e}")
logger.warning(f"Error processing commands: {e}")

45
src/webui.py Normal file
View file

@ -0,0 +1,45 @@
import logging
import os
import subprocess
from flask import Flask, render_template, request
logger = logging.getLogger("bot")
flask_app = Flask(__name__, template_folder='../flask_templates')
def read_env(filename):
if os.path.exists(filename):
with open(filename, 'r') as file:
key_value_pairs = {}
for line in file:
try:
key, value = line.strip().split('=')
key = key.strip()
value = value.strip()[1:-1]
key_value_pairs[key] = value
except:
print("This line isnt a kv pair")
return key_value_pairs
else:
return None
@flask_app.route('/', methods=['GET', 'POST'])
async def index():
key_value_pairs = read_env('.env')
if not key_value_pairs:
logger.warn("No .env file found! Copying defaults.")
key_value_pairs = read_env('.env_default')
form_dict = {}
if request.method == 'POST':
if key_value_pairs:
for form_name in key_value_pairs.keys():
form_dict[form_name] = request.form[form_name]
with open('.env', 'w') as file:
for key, value in form_dict.items():
file.write(f"{key}='{value}'\n")
print(form_dict)
flask_app.bot_process.terminate()
flask_app.bot_process = subprocess.Popen(["python", "sparkytron3000.py"])
return 'Your input has been saved! The bot must be restarted for the changes to take effect.'
return render_template('index.html', key_value_pairs = key_value_pairs)