jarvis/jarvis-tasks
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add reddit stream count

Browse source
This commit is contained in:
Zeva Rose 2022-04-21 10:29:21 -06:00
parent 4157989040
commit 2cd67aceaf
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
jarvis_tasks/tasks/reddit.py
View file

@ -174,7 +174,7 @@ async def reddit(bot: Snake) -> None:
guild = await bot.fetch_guild(follow.guild) guild = await bot.fetch_guild(follow.guild)
channel = await bot.fetch_channel(follow.channel) channel = await bot.fetch_channel(follow.channel)
if not guild or not channel: if not guild or not channel:
logger.debug(f"Follow {follow.id} no longer valid, deleting") logger.debug(f"Follow {follow.id} invalid, deleting")
await follow.delete() await follow.delete()
count -= 1 count -= 1
continue continue
@ -183,7 +183,9 @@ async def reddit(bot: Snake) -> None:
logger.debug(f"Subreddit {sub.display_name} has no followers, removing") logger.debug(f"Subreddit {sub.display_name} has no followers, removing")
await sub.delete() await sub.delete()
old_count = 0
while True: while True:
count = len(running)
subs = Subreddit.find(q(display_name__nin=running)) subs = Subreddit.find(q(display_name__nin=running))
# Go through all actively followed subreddits # Go through all actively followed subreddits
@ -214,6 +216,10 @@ async def reddit(bot: Snake) -> None:
# Create and run stream # Create and run stream
coro = _stream(sub, bot) coro = _stream(sub, bot)
asyncio.create_task(coro) asyncio.create_task(coro)
count += 1
if old_count != count:
logger.debug(f"Now streaming {count} subreddits")
old_count = count
# Check every 60 seconds # Check every 60 seconds
await asyncio.sleep(60) await asyncio.sleep(60)