scott/maubot-tmdb
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Merge branch 'develop' of codeberg.org:lomion/tmdb-bot into develop

Browse source
This commit is contained in:
lomion 2021-03-04 21:28:38 +01:00
parent e3ce109ceb 91ac13cf29
commit 41180653ac
2 changed files with 15 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
test_tmdb.py
View file

@ -107,6 +107,11 @@ class TestTmdbMethods(unittest.TestCase):
movie.search_title('The Flash') 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]) self.assertEqual('Carlos Valdes', movie.cast[2])
def test_poster_path(self):
movie = Movie()
id = movie.search_title('Dune')
self.assertEqual(movie.poster_url, "http://image.tmdb.org/t/p/w92/9HNZTw2D3cM1yA08FF5SeWEO9eX.jpg")
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

17
tmdb/tmdb.py
View file

@ -16,7 +16,7 @@ along with tmdb-bot. If not, see <https://www.gnu.org/licenses/>.
from html import escape from html import escape
import re import re
from mautrix.types import TextMessageEventContent, MediaMessageEventContent, MessageType, Format from mautrix.types import TextMessageEventContent, MediaMessageEventContent, MessageType, Format, ImageInfo
from maubot import Plugin, MessageEvent from maubot import Plugin, MessageEvent
from maubot.handlers import command from maubot.handlers import command
@ -100,12 +100,15 @@ class TmdbBot(Plugin):
await evt.respond(content) await evt.respond(content)
async def send_image(self, evt: MessageEvent, title, image) -> None: async def send_image(self, evt: MessageEvent, title, image) -> None:
mxc_uri = await self.client.upload_media(image) if image:
content = MediaMessageEventContent( mxc_uri = await self.client.upload_media(image, mime_type='image/jpg')
msgtype=MessageType.IMAGE, content = MediaMessageEventContent(
body=f"Image {title}", msgtype=MessageType.IMAGE,
url=f"{mxc_uri}") body=f"Image {title}",
await evt.respond(content) url=f"{mxc_uri}",
info=ImageInfo(mimetype='image/jpg'))
await evt.respond(content)
#await self.client.send_image(evt.room_id, url=mxc_uri, info=ImageInfo(mimetype='image/jpg'))
def split_title_year(self, message : str) -> (str, int): def split_title_year(self, message : str) -> (str, int):
m = re.search(r'^(.*) (y:\d\d\d\d)', message) m = re.search(r'^(.*) (y:\d\d\d\d)', message)