diff options
| -rw-r--r-- | bot/exts/moderation/stream.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/bot/exts/moderation/stream.py b/bot/exts/moderation/stream.py index 008ee8afe..4c86219f5 100644 --- a/bot/exts/moderation/stream.py +++ b/bot/exts/moderation/stream.py @@ -1,7 +1,9 @@ -import datetime import logging +from datetime import timedelta +import arrow import discord +from arrow import Arrow from async_rediscache import RedisCache from discord.ext import commands @@ -56,7 +58,7 @@ class Stream(commands.Cog): except discord.HTTPException as e: log.exception(f"Exception while trying to retrieve member {key} from discord\n{e}") continue - revoke_time = datetime.datetime.utcfromtimestamp(value) + revoke_time = Arrow.utcfromtimestamp(value) log.debug(f"Scheduling {member} ({member.id}) to have streaming permission revoked at {revoke_time}") self.scheduler.schedule_at( revoke_time, @@ -85,7 +87,7 @@ class Stream(commands.Cog): log.trace(f"Attempting to give temporary streaming permission to {member} ({member.id}).") # If duration is none then calculate default duration if duration is None: - duration = datetime.datetime.utcnow() + datetime.timedelta( + duration = arrow.utcnow().naive + timedelta( minutes=VideoPermission.default_permission_duration ) @@ -98,7 +100,7 @@ class Stream(commands.Cog): # Schedule task to remove streaming permission from Member and add it to task cache self.scheduler.schedule_at(duration, member.id, self._revoke_streaming_permission(member)) - await self.task_cache.set(member.id, duration.timestamp()) + await self.task_cache.set(member.id, Arrow.fromdatetime(duration).timestamp()) await member.add_roles(discord.Object(Roles.video), reason="Temporary streaming access granted") revoke_time = format_infraction_with_duration(str(duration)) await ctx.send(f"{Emojis.check_mark} {member.mention} can now stream until {revoke_time}.") |