phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

sanitize inputs

Browse source
This commit is contained in:
phixxy 2024-01-21 02:49:52 -08:00
parent 052df7d273
commit 6ef592823f
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
extensions/phixxycom.py
View file

@ -2,7 +2,7 @@ import os
import io import io
import base64 import base64
import time import time
import json import html
import asyncssh import asyncssh
from PIL import Image, PngImagePlugin from PIL import Image, PngImagePlugin
from discord.ext import commands, tasks from discord.ext import commands, tasks
@ -36,7 +36,7 @@ class PhixxyCom(commands.Cog):
if filename in line: if filename in line:
prompt = line[line.index("Prompt: ") + 7:line.index("Filename: ")] prompt = line[line.index("Prompt: ") + 7:line.index("Filename: ")]
prompt = ''.join(prompt.rsplit(',', 1)) # Remove the last comma prompt = ''.join(prompt.rsplit(',', 1)) # Remove the last comma
return prompt return html.escape(prompt)
return "Unknown Prompt" return "Unknown Prompt"
async def upload_sftp(self, local_filename, server_folder, server_filename): async def upload_sftp(self, local_filename, server_folder, server_filename):