added env variable to run the extension or not

This commit is contained in:
phixxy 2024-01-28 22:32:01 -08:00
parent 1f982b908e
commit 8d0389372d

View file

@ -11,6 +11,7 @@ class PhixxyCom(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.upload_enabled = os.getenv('upload_phixxy')
self.SERVER = os.getenv('ftp_server')
self.USERNAME = os.getenv('ftp_username')
self.PASSWORD = os.getenv('ftp_password')
@ -41,6 +42,7 @@ class PhixxyCom(commands.Cog):
return "Unknown Prompt"
async def upload_sftp(self, local_filename, server_folder, server_filename):
if self.upload_enabled.lower() == "true":
remotepath = server_folder + server_filename
async with asyncssh.connect(self.SERVER, username=self.USERNAME, password=self.PASSWORD) as conn:
async with conn.start_sftp_client() as sftp:
@ -152,6 +154,7 @@ class PhixxyCom(commands.Cog):
async def upload_ftp_ai_images(self, ai_dict):
try:
if self.upload_enabled.lower() == "true":
for folder in ai_dict:
for filename in os.listdir(folder):
if filename[-4:] == '.png':
@ -220,6 +223,7 @@ class PhixxyCom(commands.Cog):
await ctx.send("Saved suggestion!")
async def generate_blog(self):
if self.upload_enabled.lower() == "true":
start_time = time.time()
topic = ''
filename = f"{self.data_dir}ai-blog/index.html"
@ -244,8 +248,6 @@ class PhixxyCom(commands.Cog):
else:
self.bot.logger.info("No topic given for blogpost, generating one.")
topic = await self.answer_question("Give me one topic for an absurd blogpost.")
post_div = '''<!--replace this with a post-->
<div class="post">
<h2 class="post-title"><!--POST_TITLE--></h2>