Repository URL to install this package:
#!/usr/bin/python
# coding=utf-8
#
# Copyright (C) 2018-2025 by dream-alpha
#
# In case of reuse of this source code please do not remove this copyright.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program 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 General Public License for more details.
#
# For more information on the GNU General Public License see:
# <http://www.gnu.org/licenses/>.
from twisted.internet import threads, reactor
from . import tmdbsimple as tmdb
from .Utils import cleanText
from .Debug import logger
from .SearchTMDB import SearchTMDB
from .SearchMovie import SearchMovie
from .Utils import getApiKey
class ScreenTMDB():
def __init__(self, text, callback):
tmdb.API_KEY = getApiKey()
self.result1 = []
self.result2 = {}
self.callback = callback
self.text = cleanText(text)
logger.debug("text: %s", self.text)
self.search_iteration = 0
self.search_words = self.text.split(" ")
if self.text:
self.search([])
else:
self.callback({})
def search(self, result):
if self.search_iteration and self.search_words:
del self.search_words[-1]
text = " ".join(self.search_words)
else:
text = self.text
if not result and text:
self.search_iteration += 1
logger.debug("iteration: %s, text: >%s<", self.search_iteration, text)
threads.deferToThread(self.getData, text, self.search)
else:
self.gotData(result)
def getData(self, text, callback):
result = SearchTMDB().getResult(self.result1, text)
reactor.callFromThread(callback, result) # pylint: disable=E1101
def gotData(self, result):
logger.info("result: %s", result)
if result:
current = result[0][0]
logger.debug("current: %s", current)
ident = current[1]
media = current[2]
cover_url = current[3]
if media in ["movie", "tv"]:
result = SearchMovie().getResult(self.result2, ident, media)
result["cover_url"] = cover_url
logger.debug("result: %s", result)
self.callback(result)