diff --git a/DiscordBot.py b/DiscordBot.py index 0d2a7a4..946179e 100644 --- a/DiscordBot.py +++ b/DiscordBot.py @@ -89,6 +89,11 @@ def files_to_upload(dir: str) -> list: files = os.listdir(dir) return [i for i in files if os.path.splitext(i)[1] == ".torrent" in i] +def update_rss_feed(): + df = rss_gen.create_df() + rss_gen.create_48h_latest(df) + rss_gen.create_25_recent_feed(df) + @bot.event async def on_ready(): logger.info(f'{bot.user} has connected to Discord!') @@ -97,9 +102,7 @@ async def on_ready(): @bot.slash_command() async def update_rss(ctx): if ctx.user.id == 345153972765720577 or ctx.user.id == 468866369782415360: - df = rss_gen.create_df() - rss_gen.create_48h_latest(df) - rss_gen.create_25_recent_feed(df) + update_rss_feed() await ctx.respond(f"Sucessfully updated RSS feed") @bot.slash_command() @@ -145,9 +148,7 @@ async def on_message(message): os.remove(i) # noqa: E501 logger.info(f"Posted link for {i}") - df = rss_gen.create_df() - rss_gen.create_48h_latest(df) - rss_gen.create_25_recent_feed(df) + update_rss_feed() return def main():