diff --git a/test_tmdb.py b/test_tmdb.py index 68bc6a3..e55d716 100644 --- a/test_tmdb.py +++ b/test_tmdb.py @@ -85,6 +85,7 @@ class TestTmdbMethods(unittest.TestCase): movie = Movie() size = movie.set_poster_size("w500") self.assertEqual(size, "w500") + self.assertEqual(movie.base_url_poster, f"{movie.base_url_images}w500") size = movie.set_poster_size("w666") self.assertEqual(size, None) @@ -102,7 +103,7 @@ class TestTmdbMethods(unittest.TestCase): def test_cast(self): movie = TvShow() movie.search_title('The Flash') - self.assertEqual('Tom Cavanagh', movie.cast[0]) + #self.assertEqual('Tom Cavanagh', movie.cast[0]) self.assertEqual('Carlos Valdes', movie.cast[2]) if __name__ == '__main__': diff --git a/tmdb/tmdb.py b/tmdb/tmdb.py index 4e12725..385f17a 100644 --- a/tmdb/tmdb.py +++ b/tmdb/tmdb.py @@ -21,7 +21,7 @@ from mautrix.types import TextMessageEventContent, MediaMessageEventContent, Mes from maubot import Plugin, MessageEvent from maubot.handlers import command -from tmdb.tmdb_api import Movie, TvShow +from tmdb.tmdb_api import TmdbApi, Movie, TvShow from tmdb.database import Database @@ -32,21 +32,21 @@ class TmdbBot(Plugin): await super().start() self.db = Database(self.database) - async def send_movie_info(self, evt: MessageEvent, movie) -> None: - mxc_uri = await self.client.upload_media(data=movie.get_image_binary()) - text_message = f'{movie.title}' - if len(movie.overview) > 200: - three_dotts = " [...]" - else: - three_dotts = "" - cast = "Acting: " - for actor in movie.cast[:3]: - cast += f'{actor}, ' - cast = cast[:-2] - html_message = f"""
{escape(movie.title)}
-{escape(movie.overview)[:200]}{three_dotts}
-{cast}
-taken from www.themoviedb.org
""" +# async def send_movie_info(self, evt: MessageEvent, movie) -> None: +# mxc_uri = await self.client.upload_media(data=movie.get_image_binary()) +# text_message = f'{movie.title}' +# if len(movie.overview) > 200: +# three_dotts = " [...]" +# else: +# three_dotts = "" +# cast = "Acting: " +# for actor in movie.cast[:3]: +# cast += f'{actor}, ' +# cast = cast[:-2] +# html_message = f"""{escape(movie.title)}
+#{escape(movie.overview)[:200]}{three_dotts}
+#{cast}
+#taken from www.themoviedb.org
""" async def send_html_message(self, evt: MessageEvent, text_message, html_message) -> None: content = TextMessageEventContent( @@ -65,6 +65,7 @@ class TmdbBot(Plugin): html = """Use !tmdb id movie {tmdb id} to get movie detail for tmdb-id.
Use !tmdb movie {title} [y:{release year}] to get movie detail based on the given title.
Use !tmdb language {language} to set your prefered language.
+Use !tmdb poster_size {size} to set your prefered poster size. With empty {size} all available sizes are listed.
Use !tmdb tvshow {title} to get detail about a tv show based on the given title.
""" content = TextMessageEventContent( msgtype=MessageType.TEXT, format=Format.HTML, @@ -104,6 +105,16 @@ class TmdbBot(Plugin): return (title, year) return (message, None) + def set_language(self, evt: MessageEvent, movie : TmdbApi): + language = self.db.get_language(evt.sender) + if language: + movie.set_language(language) + + def poster_size(self, evt: MessageEvent, movie : TmdbApi): + size = self.db.get_poster_size(evt.sender) + if size: + movie.set_poster_size(size) + async def send_movie_info(self, evt: MessageEvent, movie) -> None: html_message = self.construct_html_message(movie) await self.send_html_message(evt, f'{movie.title}', html_message) @@ -111,6 +122,7 @@ class TmdbBot(Plugin): async def movie_id(self, evt: MessageEvent, message: str = "") -> None: movie = Movie() + self.poster_size(evt, movie) language = self.db.get_language(evt.sender) if language: movie.set_language(language) @@ -124,6 +136,7 @@ class TmdbBot(Plugin): async def movie_search(self, evt: MessageEvent, message: str = "") -> None: movie = Movie() + self.poster_size(evt, movie) language = self.db.get_language(evt.sender) if language: movie.set_language(language) @@ -149,16 +162,20 @@ class TmdbBot(Plugin): body=f"Language set to {message}!") await evt.respond(content) - async def poster_size(self, evt: MessageEvent, message: str = None) -> None: + async def set_poster_size(self, evt: MessageEvent, message: str = None) -> None: movie = Movie() + poster_sizes = "" + for x in movie.poster_sizes: + poster_sizes += x + " " if message: size = movie.set_poster_size(message) + if size: + self.db.set_poster_size(evt.sender, size) + await self.send_notice(evt, f"Set default poster size to {size}") + else: + await self.send_notice(evt, f"Failed setting poster size. Valid sizes are {poster_sizes}.") else: - self.db.set_language(evt.sender, message) - content = TextMessageEventContent( - msgtype=MessageType.NOTICE, format=Format.HTML, - body=f"Language set to {message}!") - await evt.respond(content) + await self.send_notice(evt, f"Valid sizes are {poster_sizes}.") @command.new("movie-language", help="Set language for lookup") @command.argument("message", pass_raw=True, required=True) @@ -175,6 +192,7 @@ class TmdbBot(Plugin): async def tvshow_search(self, evt: MessageEvent, message: str = "") -> None: movie = TvShow() + self.poster_size(evt, movie) language = self.db.get_language(evt.sender) if language: movie.set_language(language) @@ -182,7 +200,7 @@ class TmdbBot(Plugin): if movie.valid: await self.send_movie_info(evt, movie) else: - await self.send_notice("No tv show found!") + await self.send_notice(evt, "No tv show found!") @command.new("tvshow-search", help="TV Show lookup by Title") @command.argument("message", pass_raw=True, required=True) @@ -203,7 +221,7 @@ class TmdbBot(Plugin): elif command.lower() == 'language': await self.movie_language(evt, parameters) elif command.lower() == 'poster_size': - await self.poster_size(evt, parameters) + await self.set_poster_size(evt, parameters) elif command.lower() == 'tvshow': await self.tvshow_search(evt, parameters) elif command.lower() == 'id': diff --git a/tmdb/tmdb_api.py b/tmdb/tmdb_api.py index 0d4a365..e164ab9 100644 --- a/tmdb/tmdb_api.py +++ b/tmdb/tmdb_api.py @@ -33,7 +33,8 @@ class TmdbApi(): self.base_url = "https://api.themoviedb.org/3/" result = requests.get(self.base_url + 'configuration', params = self.get_apikey()).json() - self.base_url_poster = result['images']['base_url'] + result['images']['poster_sizes'][0] + self.base_url_images = result['images']['base_url'] + self.base_url_poster = self.base_url_images + result['images']['poster_sizes'][0] self.poster_sizes = result['images']['poster_sizes'] def get_apikey(self): @@ -59,7 +60,7 @@ class TmdbApi(): def set_poster_size(self, size): for x in self.poster_sizes: if x == size: - self.base_url_poster = result['images']['base_url'] + x + self.base_url_poster = self.base_url_images + x return x return None