parker/cordarr
parker/cordarr
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Test type
Some checks are pending
Create and publish a Docker image / build-and-push-image (push) Waiting to run
Details

Browse Source
This commit is contained in:
Parker M. 2025-01-20 22:39:14 -06:00
parent 87a0bec85d
commit 076da15485
Signed by: parker
GPG Key ID: 505ED36FC12B5D5E
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
code/cogs/status.py
View File

@ -90,18 +90,18 @@ class Status(commands.Cog):
sonarr_content_info = {} sonarr_content_info = {}
for content in requested_content: for content in requested_content:
title, release_year, local_id, tmdbid, tvdbid = content title, (release_year), local_id, tmdbid, tvdbid = content
if tmdbid is not None: if tmdbid is not None:
radarr_content_info[local_id] = { radarr_content_info[int(local_id)] = {
"title": title, "title": title,
"release_year": release_year, "release_year": int(release_year),
"tmdbid": tmdbid, "tmdbid": int(tmdbid),
} }
else: else:
sonarr_content_info[local_id] = { sonarr_content_info[int(local_id)] = {
"title": title, "title": title,
"release_year": release_year, "release_year": int(release_year),
"tvdbid": tvdbid, "tvdbid": int(tvdbid),
} }
return radarr_content_info, sonarr_content_info return radarr_content_info, sonarr_content_info
@ -130,18 +130,18 @@ class Status(commands.Cog):
id_str = "movieId" if service == "radarr" else "seriesId" id_str = "movieId" if service == "radarr" else "seriesId"
# If the content was requested by the user # If the content was requested by the user
if ( if (
download[id_str] in content_info.keys() download[int(id_str)] in content_info.keys()
and download[id_str] not in added_ids and download[int(id_str)] not in added_ids
): ):
# Append local ID # Append local ID
added_ids.append(download[id_str]) added_ids.append(download[int(id_str)])
# Add the download to the embed # Add the download to the embed
try: try:
time_left = self.process_time(download["timeleft"]) time_left = self.process_time(download["timeleft"])
except KeyError: except KeyError:
time_left = "Unknown" time_left = "Unknown"
description += ( description += (
f"\n**{content_info[download[id_str]]['title']} ({content_info[download[id_str]]['release_year']})**" f"\n**{content_info[download[int(id_str)]]['title']} ({content_info[download[int(id_str)]]['release_year']})**"
f" - Time Left: `{time_left}`" f" - Time Left: `{time_left}`"
) )
@ -167,7 +167,7 @@ class Status(commands.Cog):
for content in requested_content: for content in requested_content:
title, release_year, local_id, tmdbid, _ = content title, release_year, local_id, tmdbid, _ = content
# If not in queue # If not in queue
if local_id not in added_ids: if int(local_id) not in added_ids:
# Pull the movie data from the service # Pull the movie data from the service
if tmdbid is not None: if tmdbid is not None:
data = requests.get( data = requests.get(
@ -188,7 +188,7 @@ class Status(commands.Cog):
cursor.execute( cursor.execute(
"DELETE FROM requests WHERE user_id = ? AND" "DELETE FROM requests WHERE user_id = ? AND"
" local_id = ?", " local_id = ?",
(user_id, local_id), (user_id, int(local_id)),
) )
db.commit() db.commit()
db.close() db.close()