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

Changes by yrrad8

Browse Source
This commit is contained in:
GaMeNu 2023-10-22 23:41:19 +03:00
parent dfd454d120
commit 989d749615
1 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
cog_notificator.py
View File

@ -138,9 +138,10 @@ class Notificator(commands.Cog):
def in_registered_channel(self, intr: discord.Interaction) -> bool | None: def in_registered_channel(self, intr: discord.Interaction) -> bool | None:
""" """
an info about current channel
:param intr: discord command interaction :param intr: Command interaction from discord
:return: True - is a registered server channel, False - is a registered DM, None - was not found (is not registered) :return: Boolean:
True - is a registered server channel, False - is a registered DM, None - was not found (may not be registered)
""" """
# OPTIONS: # OPTIONS:
@ -285,6 +286,12 @@ class Notificator(commands.Cog):
return view return view
async def send_new_alert(self, alert_data: dict, new_districts: list[str]): async def send_new_alert(self, alert_data: dict, new_districts: list[str]):
"""
Push an alert to all registered channels
:param alert_data: Alert data dict (see test_alert for format)
:param new_districts: Currently active districts (districts that were not already active)
:return:
"""
try: try:
alert_history = AlertReqs.request_history_json()[0:100] alert_history = AlertReqs.request_history_json()[0:100]
except requests.exceptions.Timeout as error: except requests.exceptions.Timeout as error:
@ -420,10 +427,6 @@ class Notificator(commands.Cog):
await intr.response.send_message(f'Invalid confirmation string!') await intr.response.send_message(f'Invalid confirmation string!')
return return
channel = self.db.get_channel(channel_id)
if channel is None:
channel = self.db.get_channel(intr.user.id)
await self.attempt_unregistration(intr, channel) await self.attempt_unregistration(intr, channel)
async def attempt_unregistration(self, intr, channel: db_access.Channel): async def attempt_unregistration(self, intr, channel: db_access.Channel):
@ -604,6 +607,15 @@ class Notificator(commands.Cog):
desc: str = 'התעלמו מהתראה זו', desc: str = 'התעלמו מהתראה זו',
districts: str = 'בדיקה', districts: str = 'בדיקה',
cat: int = 99): cat: int = 99):
"""
A function to send a test alert
:param intr: Command interaction from discord
:param title: Title of the alert
:param desc: Description of the alert
:param districts: Districts of the alert
:param cat: Category of the alert
:return:
"""
if intr.user.id != AUTHOR_ID: if intr.user.id != AUTHOR_ID:
await intr.response.send_message('No access.') await intr.response.send_message('No access.')
return return