Create CordArr

1 files changed, 111 insertions, 0 deletions

diff --git a/code/cogs/status.py b/code/cogs/status.py
new file mode 100644
index 0000000..7b6b463
--- /dev/null
+++ b/code/cogs/status.py
@@ -0,0 +1,111 @@
+import discord
+from discord import app_commands
+from discord.ext import commands
+import requests
+import sqlite3
+import datetime
+import humanize
+
+from global_variables import RADARR_HOST_URL, RADARR_HEADERS
+
+
+class Status(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @app_commands.command()
+    async def status(self, interaction: discord.Interaction):
+        "Get the status of the movies you have requested"
+        # Get all the movie_ids that were requested by the user
+        db = sqlite3.connect("cordarr.db")
+        cursor = db.cursor()
+        cursor.execute(
+            "SELECT movie_id, movie_title FROM movies WHERE user_id = ?",
+            (interaction.user.id,),
+        )
+        requested_movies = cursor.fetchall()
+
+        users_movies = {}  # Dictionary to store the movies that the user has requested
+        for movie_id, movie_title in requested_movies:
+            users_movies[movie_id] = movie_title
+        # If theres no movies, return a message saying so
+        if not users_movies:
+            embed = discord.Embed(
+                title="No Movies Requested",
+                description="You have no movies being downloaded at the moment. If you previously added a movie, it is likely that it has finished downloading. If you believe this is an error, please contact an administrator.",
+                color=0xD01B86
+            )
+            return await interaction.response.send_message(embed=embed, ephemeral=True)
+        # Otherwise, create the default embed to display the movies being downloaded
+        embed = discord.Embed(
+            title="Movies Requested",
+            description="Here are the movies you have requested that are currently being downloaded:\n",
+            color=0xD01B86
+        )
+
+        # Now, we get the download status of all movies from the Radarr queue
+        response = requests.get(
+            f"{RADARR_HOST_URL}/api/v3/queue/", headers=RADARR_HEADERS
+        ).json()
+
+        count = 0
+        added_movie_ids = []
+        for movie in response["records"]:
+            movie_id = movie["movieId"]
+            # If the movie is user requested and is being downloaded
+            if movie_id in users_movies.keys():
+                count += 1
+                added_movie_ids.append(movie_id)
+                if movie["status"] == "downloading":
+                    # Humanize the download time left, or result to 'Unknown
+                    try:
+                        time_left = humanize.precisedelta(
+                            datetime.datetime.strptime(movie["timeleft"], "%H:%M:%S")
+                            - datetime.datetime.strptime("00:00:00", "%H:%M:%S"),
+                            minimum_unit="seconds",
+                        )
+                    except ValueError:
+                        # Sometimes movies will download extremely show and therefore might
+                        # show 'days' in the time left, so strptime appropriately
+                        time_left = humanize.precisedelta(
+                            datetime.datetime.strptime(movie["timeleft"], "%d.%H:%M:%S")
+                            - datetime.datetime.strptime("00:00:00", "%H:%M:%S"),
+                            minimum_unit="seconds",
+                        )
+                    except KeyError or ValueError:
+                        time_left = "Unknown"
+
+                    # Add all the information
+                    embed.description += f"\n{count}. **{users_movies[movie_id]}** - Time Left: ` {time_left} `"
+                else:
+                    embed.description += f"\n{count}. **{users_movies[movie_id]}** - Status: `{str(movie['status']).upper()}`"
+
+        # If a movie wasn't found in the Radarr queue, then it has either finished downloading
+        # or the movie was never found for download
+        if len(added_movie_ids) != len(users_movies.keys()):
+            # Grab all of the "missing" movies to see if a movie is missing or finished downloading
+            response = requests.get(
+                f"{RADARR_HOST_URL}/api/v3/wanted/missing", headers=RADARR_HEADERS
+            ).json()
+            for movie in response["records"]:
+                movie_id = movie["id"]
+                if movie_id in users_movies.keys() and movie_id not in added_movie_ids:
+                    count += 1
+                    added_movie_ids.append(movie_id)
+                    embed.description += f"\n{count}. **{users_movies[movie_id]}** - Status: ` NOT FOUND `"
+        # If there are still movies that haven't been added to the embed, then they
+        # have finished downloading and can be removed from the database
+        for movie_id in users_movies.keys():
+            if movie_id not in added_movie_ids:
+                cursor.execute(
+                    "DELETE FROM movies WHERE user_id = ? AND movie_id = ?",
+                    (interaction.user.id, movie_id),
+                )
+        db.commit()
+        db.close()
+
+        await interaction.response.send_message(embed=embed, ephemeral=True)
+
+
+async def setup(bot):
+    await bot.add_cog(Status(bot))