Mihit/upload-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added function for updating rss feed

Browse Source
This commit is contained in:
Mihit Mittal
2025-04-09 15:54:42 +10:00
parent 189b2cd36b
commit cbe46eaa0a
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
DiscordBot.py
View File

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