Merge branch 'main' into gabby-dev

Reviewed-on: #1

src/py/bot_aux.py

@@ -6,7 +6,6 @@ This automation will be run on a daily basis, through a cron job + docker.
 @author: Gabriella 'contrastellar' Agathon
 """
 import argparse
-import os
 import discord
 import helper.db_helper
 
@@ -37,11 +36,22 @@ async def on_ready():
 
     print(f'{client.user} has connected.')
     print(args.guild_id)
+
     guild: discord.Guild = client.get_guild(args.guild_id)
     channel: discord.TextChannel = guild.get_channel(args.channel_id)
+
     callouts = DATABASE_CONN.query_callouts(NUMBER_OF_DAYS)
     formatted_callouts = DATABASE_CONN.formatted_list_of_callouts(callouts)
     output = f'Callouts for the next {NUMBER_OF_DAYS} days:\n' + formatted_callouts
+    output += '\n\n'
+    
+    output += f'Breaks for the next {NUMBER_OF_DAYS} days:\n'
+
+    breaks: str = DATABASE_CONN.query_breaks(NUMBER_OF_DAYS)
+    formatted_breaks: str = DATABASE_CONN.formatted_list_of_breaks(breaks)
+
+    output += formatted_breaks
+
     await channel.send(output)
     await client.close() # Another way to exit, a little bit cleaner than exit(0)
     return

src/py/bot_core.py

@@ -160,7 +160,8 @@ async def ping(interaction: discord.Interaction) -> None:
     delete_invalidate()
     cleanup_invalidate()
     user_id = interaction.user.id
-    await interaction.response.send_message(f'Pong! {user_id} -- the bot is active, please message contrastellar with issues!')
+    charname = DATABASE_CONN.return_char_name(uid=user_id)
+    await interaction.response.send_message(f'Pong! {charname} -- the bot is active, please message contrastellar with issues!')
     return
 
 
@@ -287,6 +288,24 @@ async def remove_callout(interaction: discord.Interaction, day: int, month: int,
         await interaction.response.send_message(f'{user_char_name} removed a callout for {callout_date}')
 
 
+@client.tree.command()
+async def remove_break(interaction: discord.Interaction, day: int, month: int, year: int) -> None:
+    delete_invalidate()
+    cleanup_invalidate()
+    user_id = interaction.user.id
+    user_char_name = DATABASE_CONN.return_char_name(user_id)
+    break_date: datetime.date = datetime.date(year=year, month=month, day=day)
+
+    try:
+        DATABASE_CONN.remove_break(user_id=user_id, break_date=break_date)
+    except psycopg2.Error:
+        await interaction.response.send_message(f'{user_char_name} -- no break was added for {break_date}')
+    else:
+        await interaction.response.send_message(f'{user_char_name} -- you removed the break starting on {break_date}')
+
+    return
+
+
 @client.tree.command()
 async def schedule(interaction: discord.Interaction, days: int = DAYS_FOR_CALLOUTS) -> None:
     delete_invalidate()
@@ -297,6 +316,18 @@ async def schedule(interaction: discord.Interaction, days: int = DAYS_FOR_CALLOU
     await interaction.followup.send(f'Callouts for the next {days} days:\n{callouts}')
     return
 
+
+@client.tree.command()
+async def breaks(interaction: discord.Interaction, days: int = 365) -> None:
+    delete_invalidate()
+    cleanup_invalidate()
+    await interaction.response.defer(thinking=True)
+    breaks: list = DATABASE_CONN.query_breaks(days=days)
+    break_output: str = DATABASE_CONN.format_list_of_breaks(breaks=breaks)
+    await interaction.followup.send(f'Breaks for the next {days} days:\n{break_output}')
+    return
+
+
 @client.tree.command()
 async def self_callouts(interaction: discord.Interaction, days: int = 365) -> None:
     delete_invalidate()

src/py/db_helper_tests.py

@@ -58,4 +58,12 @@ class TestClass():
         formatted_callouts = self.DATABASE_CONN.format_list_of_callouts(callouts=callouts)
         assert formatted_callouts.__class__ is str
 
-    # TODO write tests for the break addition/removal
+    def test_add_break(self) -> None:
+        today: datetime.date = datetime.date.today()
+        new_break = self.DATABASE_CONN.add_break(user_id=1, break_start=today, break_end=today)
+        assert new_break is None
+
+    def test_remove_break(self) -> None:
+        today: datetime.date = datetime.date.today()
+        remove_break = self.DATABASE_CONN.remove_break(user_id=1, start_date=today)
+        assert remove_break is None

src/py/helper/db_helper.py

@@ -100,15 +100,17 @@ class DBHelper():
 
         return cursor.fetchall()
     
-    def query_breaks(self) -> list:
+
+    def query_breaks(self, days) -> list:
         self.__CONN = connect_config(self._config)
         self.__CONN.autocommit = True
         cursor = self.__CONN.cursor()
-        cursor.execute(f"SELECT * FROM breaks")
+        cursor.execute(f"SELECT * FROM breaks WHERE open_range >= NOW() - INTERVAL '1 day' and open_range <= NOW() + interval '{days} days' ORDER BY open_range ASC;")
         self.__CONN.commit()
 
         return cursor.fetchall()
     
+
     def query_self_callouts(self, user_id: int, days: int = 365):
         self.__CONN = connect_config(self._config)
         self.__CONN.autocommit = True
@@ -137,16 +139,18 @@ class DBHelper():
 
         return
 
+
     def add_break(self, user_id: int, break_start: datetime.date, break_end: datetime.date) -> None:
 
         self.__CONN = connect_config(self._config)
         self.__CONN.autocommit = True
         cursor = self.__CONN.cursor()
-        is_range: int = 1
+        is_range: bool = True
         if break_start == break_end:
-            is_range = 0
+            is_range = False
         cursor.execute("INSERT INTO breaks (created_user, is_range, open_range, close_range) VALUES (%s, %s, %s, %s)", (user_id, is_range, break_start, break_end))
 
+        return
 
 
     def remove_callout(self, user_id: int, callout: datetime.date) -> None:
@@ -166,6 +170,76 @@ class DBHelper():
         return
     
 
+    def remove_break(self, user_id: int, start_date: datetime.date) -> None:
+        """Remove a callout based on user + date, which form the primary key in the db
+
+        Args:
+            user_id (int): The Discord UUID of the user removing something from the db
+            callout (datetime.datetime): The date of the callout
+        """
+        self.__CONN = connect_config(self._config)
+        self.__CONN.autocommit = True
+        cursor = self.__CONN.cursor()
+
+        cursor.execute("DELETE FROM breaks WHERE created_user = %s AND open_range = %s", (user_id, start_date))
+        self.__CONN.commit()
+
+        return
+
+
+    def formatted_list_of_breaks(self, breaks: list) -> str:
+        """Format the python list of breaks.
+
+        Args:
+            breaks (list): The list that needs to be formatted
+
+        Returns:
+            str: the list as an outputtable string
+        """
+        length = len(breaks)
+        output: str = ''
+
+        if length == 0:
+            return 'No breaks have been scheduled!'
+        
+        for entry in breaks:
+
+            for item in range(4):
+                if item == 0:
+                    output += f"Break submitted by: {self.return_char_name(entry[0])}"
+
+                if item == 1:
+                    # is_range boolean
+                    if entry[1] == True:
+                        # if it is a range, format as a range
+                        output += f' • Break from: {entry[2]} until... '
+
+                    else:
+                        # if it is not a range, format as a single date
+                        output += f' • We are taking a break on: {entry[2]}'
+                
+                if item == 2:
+                    if entry[1] == True:
+                        output += f'{entry[3]}.\n'
+                    else:
+                        output += '\n'
+
+        output += "\nEND OF MESSAGE"
+        return output
+    
+
+    def format_list_of_breaks(self, breaks: list) -> str:
+        """Format the list of breaks
+
+        Args:
+            breaks (list): The list that needs to be formatted
+
+        Returns:
+            str: The formatted list to the bot_core or bot_aux
+        """
+        return self.formatted_list_of_breaks(breaks=breaks)
+
+
     def formatted_list_of_callouts(self, callouts: list) -> str:
         """Format the python list of callouts.