NgTech-LTD/HFCNotificator
2
0
Fork 0
mirror of https://github.com/GaMeNu/HFCNotificator.git synced 2024-11-16 15:24:51 +02:00
Code Issues Packages Projects Releases Wiki Activity

v2.1.1

Browse Source 
Attempted update_loop error fix
issue #004
This commit is contained in:
GaMeNu 2023-10-12 12:55:26 +03:00
parent 3c43fad859
commit 36ac574155
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
cog_notificator.py
View File

@ -81,8 +81,6 @@ class Notificator(commands.Cog):
self.active_districts = []
if not self.check_for_updates.is_running():
self.check_for_updates.start()
@ -128,6 +126,12 @@ class Notificator(commands.Cog):
await self.send_new_alert(current_alert, new_districts)
self.active_districts = data
@check_for_updates.error
async def update_loop_error(self):
self.db.connection.close()
self.db = DBAccess()
@staticmethod
def generate_alert_embed(alert_object: Alert, district: str, arrival_time: int | None, time: str,
lang: str) -> discord.Embed:
@ -218,7 +222,6 @@ class Notificator(commands.Cog):
self.log.warning(f'Failed to send alert in channel id={channel.id}:\n'
f'{e}')
@app_commands.command(name='register',
description='Register a channel to receive HFC alerts (Requires Manage Channels)')
@app_commands.checks.has_permissions(manage_channels=True)
@ -251,14 +254,13 @@ class Notificator(commands.Cog):
ch = self.bot.get_channel(channel_id)
perms = ch.overwrites_for(self.bot.user)
perms.update(send_messages=True)
await ch.set_permissions(target=ch.guild.me, overwrite=perms, reason='Update perms to allow bot to send messages in channel.')
await ch.set_permissions(target=ch.guild.me, overwrite=perms,
reason='Update perms to allow bot to send messages in channel.')
except discord.errors.Forbidden as e:
await intr.followup.send(f'Could not allow bot to send messages to this channel! Please add the bot to this channel and allow it to send messages.\n'
await intr.followup.send(
f'Could not allow bot to send messages to this channel! Please add the bot to this channel and allow it to send messages.\n'
f'Error info: {e.__str__()}')
@register_channel.error
async def register_channel_error(self, intr: discord.Interaction, error):
if isinstance(error, app_commands.MissingPermissions):
@ -398,7 +400,7 @@ class Notificator(commands.Cog):
if page_number < 0:
raise ValueError("Page number is too low.")
page_info = f'Page {page_number + 1}/{alert_counter//alerts_in_page + 1}\n\n'
page_info = f'Page {page_number + 1}/{alert_counter // alerts_in_page + 1}\n\n'
ret_str = ''
@ -417,7 +419,6 @@ class Notificator(commands.Cog):
else:
ret_str = page_info + ret_str
return ret_str
@app_commands.command(name='about', description='Info about the bot')