added env variable to run the extension or not
This commit is contained in:
parent
1f982b908e
commit
8d0389372d
1 changed files with 87 additions and 85 deletions
|
|
@ -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>
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue