2020-06-22 15:22:28 +01:00
|
|
|
'''
|
|
|
|
This file is part of tmdb-bot.
|
|
|
|
|
|
|
|
tmdb-bot is free software: you can redistribute it and/or modify
|
|
|
|
it under the terms of the GNU Affero General Public License version 3 as published by
|
|
|
|
the Free Software Foundation.
|
|
|
|
|
|
|
|
tmdb-bot is distributed in the hope that it will be useful,
|
|
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
GNU Affero General Public License for more details.
|
|
|
|
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
|
|
along with tmdb-bot. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
'''
|
2020-06-22 12:15:28 +01:00
|
|
|
from html import escape
|
|
|
|
|
|
|
|
from mautrix.types import TextMessageEventContent, MediaMessageEventContent, MessageType, Format, RelatesTo, RelationType
|
|
|
|
|
|
|
|
from maubot import Plugin, MessageEvent
|
|
|
|
from maubot.handlers import command
|
|
|
|
|
2020-06-24 10:17:19 +01:00
|
|
|
from tmdb.tmdb_api import Movie, TvShow
|
2020-06-23 16:34:34 +01:00
|
|
|
from tmdb.database import Database
|
2020-06-22 12:15:28 +01:00
|
|
|
|
|
|
|
|
2020-06-22 15:21:18 +01:00
|
|
|
class TmdbBot(Plugin):
|
|
|
|
db: Database
|
2020-06-22 13:41:31 +01:00
|
|
|
|
|
|
|
async def start(self) -> None:
|
|
|
|
await super().start()
|
|
|
|
self.db = Database(self.database)
|
2020-06-23 14:47:40 +01:00
|
|
|
|
2020-06-24 10:17:19 +01:00
|
|
|
async def send_html_message(self, evt: MessageEvent, text_message, html_message) -> None:
|
2020-06-22 12:15:28 +01:00
|
|
|
content = TextMessageEventContent(
|
|
|
|
msgtype=MessageType.TEXT, format=Format.HTML,
|
|
|
|
body=f"{text_message}",
|
|
|
|
formatted_body=f"{html_message}")
|
|
|
|
await evt.respond(content)
|
2020-06-23 14:47:40 +01:00
|
|
|
|
2020-06-24 10:17:19 +01:00
|
|
|
async def send_image(self, evt: MessageEvent, title, image) -> None:
|
2020-06-23 14:47:40 +01:00
|
|
|
mxc_uri = await self.client.upload_media(image)
|
2020-06-22 12:15:28 +01:00
|
|
|
content = MediaMessageEventContent(
|
|
|
|
msgtype=MessageType.IMAGE,
|
2020-06-23 14:47:40 +01:00
|
|
|
body=f"Image {title}",
|
2020-06-22 12:15:28 +01:00
|
|
|
url=f"{mxc_uri}")
|
|
|
|
await evt.respond(content)
|
|
|
|
|
2020-06-23 14:47:40 +01:00
|
|
|
def construct_html_message(self, movie, overview_length = 200, cast_length = 3) -> str:
|
|
|
|
if len(movie.overview) > overview_length:
|
|
|
|
three_dotts = " [...]"
|
|
|
|
else:
|
|
|
|
three_dotts = ""
|
|
|
|
|
|
|
|
cast = "Acting: "
|
|
|
|
for actor in movie.cast[:cast_length]:
|
|
|
|
cast+= f'{actor}, '
|
|
|
|
cast = cast[:-2]
|
2020-06-24 09:51:42 +01:00
|
|
|
html_message = f"""<p><a href="{movie.web_url}"><b>{escape(movie.title)}</b></a></p>
|
2020-06-23 14:47:40 +01:00
|
|
|
<p>{escape(movie.overview)[:overview_length]}{three_dotts}</p>
|
|
|
|
<p>{cast}</p>
|
2020-06-24 10:17:19 +01:00
|
|
|
<p>Taken from www.themoviedb.org</p>"""
|
2020-06-23 14:47:40 +01:00
|
|
|
return html_message
|
|
|
|
|
|
|
|
async def send_movie_info(self, evt: MessageEvent, movie) -> None:
|
|
|
|
html_message = self.construct_html_message(movie)
|
2020-06-24 10:17:19 +01:00
|
|
|
await self.send_html_message(evt, f'{movie.title}', html_message)
|
|
|
|
await self.send_image(evt, movie.title, movie.get_image_binary())
|
2020-06-23 14:47:40 +01:00
|
|
|
|
2020-06-22 12:15:28 +01:00
|
|
|
|
|
|
|
@command.new("movie-id", help="Movie lookup by id")
|
|
|
|
@command.argument("message", pass_raw=True, required=True)
|
|
|
|
async def movie_id(self, evt: MessageEvent, message: str = "") -> None:
|
|
|
|
movie = Movie()
|
2020-06-22 15:21:18 +01:00
|
|
|
language = self.db.get_language(evt.sender)
|
|
|
|
if language:
|
|
|
|
movie.set_language(language)
|
2020-06-22 12:15:28 +01:00
|
|
|
movie.query_details(message)
|
|
|
|
await self.send_movie_info(evt, movie)
|
|
|
|
|
|
|
|
|
|
|
|
@command.new("movie-search", help="Movie lookup by Title")
|
|
|
|
@command.argument("message", pass_raw=True, required=True)
|
|
|
|
async def movie_search(self, evt: MessageEvent, message: str = "") -> None:
|
|
|
|
movie = Movie()
|
2020-06-22 13:41:31 +01:00
|
|
|
language = self.db.get_language(evt.sender)
|
|
|
|
if language:
|
|
|
|
movie.set_language(language)
|
2020-06-22 12:15:28 +01:00
|
|
|
movie.search_title(message)
|
|
|
|
if movie.valid:
|
|
|
|
await self.send_movie_info(evt, movie)
|
|
|
|
else:
|
|
|
|
content = TextMessageEventContent(
|
|
|
|
msgtype=MessageType.NOTICE, format=Format.HTML,
|
|
|
|
body=f"No movie found!")
|
|
|
|
await evt.respond(content)
|
2020-06-22 13:41:31 +01:00
|
|
|
|
|
|
|
@command.new("movie-language", help="Set language for lookup")
|
|
|
|
@command.argument("message", pass_raw=True, required=True)
|
|
|
|
async def movie_language(self, evt: MessageEvent, message: str = "") -> None:
|
|
|
|
self.db.set_language(evt.sender, message)
|
|
|
|
content = TextMessageEventContent(
|
|
|
|
msgtype=MessageType.NOTICE, format=Format.HTML,
|
2020-06-22 15:21:18 +01:00
|
|
|
body=f"Language set to {message}!")
|
2020-06-22 13:41:31 +01:00
|
|
|
await evt.respond(content)
|
2020-06-24 10:17:19 +01:00
|
|
|
|
|
|
|
@command.new("tvshow-search", help="TV Show lookup by Title")
|
|
|
|
@command.argument("message", pass_raw=True, required=True)
|
|
|
|
async def tvshow_search(self, evt: MessageEvent, message: str = "") -> None:
|
|
|
|
movie = TvShow()
|
|
|
|
language = self.db.get_language(evt.sender)
|
|
|
|
if language:
|
|
|
|
movie.set_language(language)
|
|
|
|
movie.search_title(message)
|
|
|
|
if movie.valid:
|
|
|
|
await self.send_movie_info(evt, movie)
|
|
|
|
else:
|
|
|
|
content = TextMessageEventContent(
|
|
|
|
msgtype=MessageType.NOTICE, format=Format.HTML,
|
|
|
|
body=f"No tv show found!")
|
|
|
|
await evt.respond(content)
|