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

added env variable to run the extension or not

Browse source
This commit is contained in:
phixxy 2024-01-28 22:32:01 -08:00
parent 1f982b908e
commit 8d0389372d
1 changed files with 87 additions and 85 deletions
Download patch file Download diff file Expand all files Collapse all files

6
extensions/phixxycom.py
View file

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