starred/MediaManager-maxdorninger-1
1
0
Fork 0
mirror of https://github.com/maxdorninger/MediaManager.git synced 2026-04-17 15:43:28 +02:00
Code Issues Packages Projects Releases Wiki Activity

refactor search queries to remove special characters and parentheses

Browse Source
This commit is contained in:
maxid
2025-12-20 20:05:02 +01:00
parent cfe34358a0
commit d5c649d5bf
2 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
media_manager/metadataProvider/tmdb.py
View File

@@ -8,6 +8,7 @@ from media_manager.metadataProvider.abstractMetaDataProvider import (
AbstractMetadataProvider,
)
from media_manager.metadataProvider.schemas import MetaDataProviderSearchResult
from media_manager.torrent.utils import remove_special_chars_and_parentheses
from media_manager.tv.schemas import Episode, Season, Show, SeasonNumber, EpisodeNumber
from media_manager.movies.schemas import Movie
from media_manager.notification.manager import notification_manager
@@ -60,7 +61,11 @@ class TmdbMetadataProvider(AbstractMetadataProvider):
def __search_tv(self, query: str, page: int) -> dict:
try:
response = requests.get(
url=f"{self.url}/tv/search", params={"query": query, "page": page}
url=f"{self.url}/tv/search",
params={
"query": remove_special_chars_and_parentheses(query),
"page": page,
},
)
response.raise_for_status()
return response.json()
@@ -104,7 +109,11 @@ class TmdbMetadataProvider(AbstractMetadataProvider):
def __search_movie(self, query: str, page: int) -> dict:
try:
response = requests.get(
url=f"{self.url}/movies/search", params={"query": query, "page": page}
url=f"{self.url}/movies/search",
params={
"query": remove_special_chars_and_parentheses(query),
"page": page,
},
)
response.raise_for_status()
return response.json()

11
media_manager/metadataProvider/tvdb.py
View File

@@ -8,6 +8,7 @@ from media_manager.metadataProvider.abstractMetaDataProvider import (
AbstractMetadataProvider,
)
from media_manager.metadataProvider.schemas import MetaDataProviderSearchResult
from media_manager.torrent.utils import remove_special_chars_and_parentheses
from media_manager.tv.schemas import Episode, Season, Show, SeasonNumber
from media_manager.movies.schemas import Movie
@@ -29,7 +30,10 @@ class TvdbMetadataProvider(AbstractMetadataProvider):
return requests.get(f"{self.url}/tv/seasons/{id}").json()
def __search_tv(self, query: str) -> dict:
return requests.get(f"{self.url}/tv/search", params={"query": query}).json()
return requests.get(
f"{self.url}/tv/search",
params={"query": remove_special_chars_and_parentheses(query)},
).json()
def __get_trending_tv(self) -> dict:
return requests.get(f"{self.url}/tv/trending").json()
@@ -38,7 +42,10 @@ class TvdbMetadataProvider(AbstractMetadataProvider):
return requests.get(f"{self.url}/movies/{id}").json()
def __search_movie(self, query: str) -> dict:
return requests.get(f"{self.url}/movies/search", params={"query": query}).json()
return requests.get(
f"{self.url}/movies/search",
params={"query": remove_special_chars_and_parentheses(query)},
).json()
def __get_trending_movies(self) -> dict:
return requests.get(f"{self.url}/movies/trending").json()