def search_image(self, search_query, manual_select=False): """ search google images with the given query, returns first/best match optional parameter: manual_select (bool), will show selectdialog to allow manual select by user """ image = "" images_list = [] for img in self.get_data(search_query): img = img.replace(" ", "%20") # fix for spaces in url if xbmcvfs.exists(img): if not manual_select: # just return the first image found (assuming that will be the best match) return img else: # manual lookup, list results and let user pick one listitem = xbmcgui.ListItem(label=img) listitem.setArt({'icon': img}) images_list.append(listitem) if manual_select and images_list: dialog = DialogSelect("DialogSelect.xml", "", listing=images_list, window_title="%s - Google" % xbmc.getLocalizedString(283)) dialog.doModal() selected_item = dialog.result del dialog if selected_item != -1: selected_item = images_list[selected_item] if sys.version_info.major == 3: image = selected_item.getLabel() else: image = selected_item.getLabel().decode("utf-8") return image
def select_art(items, manual_select=False, art_type=""): """select the preferred image from the list""" image = None if manual_select: # show selectdialog to manually select the item results_list = [] # add none and browse entries listitem = xbmcgui.ListItem(label=xbmc.getLocalizedString(231), iconImage="DefaultAddonNone.png") results_list.append(listitem) listitem = xbmcgui.ListItem(label=xbmc.getLocalizedString(1030), iconImage="DefaultFolder.png") results_list.append(listitem) for item in items: labels = [item["contributedby"], item["dateadded"], item["language"], item["source"]] label = " / ".join(labels) listitem = xbmcgui.ListItem(label=label, iconImage=item["thumb"]) results_list.append(listitem) if manual_select and results_list: dialog = DialogSelect("DialogSelect.xml", "", listing=results_list, window_title=art_type) dialog.doModal() selected_item = dialog.result del dialog if selected_item == 0: image = "" if selected_item == 1: # browse for image dialog = xbmcgui.Dialog() image = dialog.browse(2, xbmc.getLocalizedString(1030), 'files', mask='.gif').decode("utf-8") del dialog elif selected_item > 1: # user has selected an image from online results image = items[selected_item - 2]["image"] elif items: # just grab the first item as best match image = items[0]["image"] return image
def lookup_tvdb(self, searchtitle, channel, manual_select=False): '''helper to select a match on tvdb''' tvdb_match = None searchtitle = searchtitle.lower() tvdb_result = self.artutils.thetvdb.search_series(searchtitle, True) searchchannel = channel.lower().split("hd")[0].replace(" ", "") match_results = [] if tvdb_result: for item in tvdb_result: item["score"] = 0 itemtitle = item["seriesName"].lower() network = item["network"].lower().replace(" ", "") # high score if channel name matches if network in searchchannel or searchchannel in network: item["score"] += 800 # exact match on title - very high score if searchtitle == itemtitle: item["score"] += 1000 # match title by replacing some characters if re.sub('\*|,|.\"|\'| |:|;', '', searchtitle) == re.sub('\*|,|.\"|\'| |:|;', '', itemtitle): item["score"] += 750 # add SequenceMatcher score to the results stringmatchscore = SM(None, searchtitle, itemtitle).ratio() if stringmatchscore > 0.7: item["score"] += stringmatchscore * 500 # prefer items with native language as we've searched with localized info enabled if item["overview"]: item["score"] += 250 # prefer items with artwork if item["banner"]: item["score"] += 1 if item["score"] > 500 or manual_select: match_results.append(item) # sort our new list by score match_results = sorted(match_results, key=itemgetter("score"), reverse=True) if match_results and manual_select: # show selectdialog to manually select the item listitems = [] for item in match_results: thumb = "http://thetvdb.com/banners/%s" % item["banner"] if item["banner"] else "" listitem = xbmcgui.ListItem(label=item["seriesName"], iconImage=thumb, label2=item["overview"]) listitems.append(listitem) dialog = DialogSelect( "DialogSelect.xml", "", listing=listitems, window_title="%s - TVDB" % xbmc.getLocalizedString(283)) dialog.doModal() selected_item = dialog.result del dialog if selected_item != -1: tvdb_match = match_results[selected_item]["id"] else: match_results = [] if not tvdb_match and match_results: # just grab the first item as best match tvdb_match = match_results[0]["id"] return tvdb_match
def select_best_match(results, prefyear="", preftype="", preftitle="", manual_select=False): '''helper to select best match or let the user manually select the best result from the search''' details = {} # score results if one or more preferences are given if results and (prefyear or preftype or preftitle): newdata = [] preftitle = preftitle.lower() for item in results: item["score"] = 0 itemtitle = item["title"] if item.get( "title") else item["name"] itemtitle = itemtitle.lower() itemorgtitle = item["original_title"] if item.get( "original_title") else item["original_name"] itemorgtitle = itemorgtitle.lower() # high score if year matches if prefyear: if item.get("first_air_date" ) and prefyear in item["first_air_date"]: item["score"] += 800 # matches preferred year if item.get("release_date" ) and prefyear in item["release_date"]: item["score"] += 800 # matches preferred year # find exact match on title if preftitle and preftitle == itemtitle: item["score"] += 1000 # exact match! if preftitle and preftitle == itemorgtitle: item["score"] += 1000 # exact match! # match title by replacing some characters if preftitle and get_compare_string( preftitle) == get_compare_string(itemtitle): item["score"] += 750 if preftitle and get_compare_string( preftitle) == get_compare_string(itemorgtitle): item["score"] += 750 # add SequenceMatcher score to the results if preftitle: stringmatchscore = SM( None, preftitle, itemtitle).ratio() + SM( None, preftitle, itemorgtitle).ratio() if stringmatchscore > 1.6: item["score"] += stringmatchscore * 250 # higher score if result ALSO matches our preferred type or native language # (only when we already have a score) if item["score"]: if preftype and (item["media_type"] in preftype) or ( preftype in item["media_type"]): item["score"] += 250 # matches preferred type if item["original_language"] == KODI_LANGUAGE: item["score"] += 500 # native language! if KODI_LANGUAGE.upper() in item.get("origin_country", []): item["score"] += 500 # native language! if KODI_LANGUAGE in item.get("languages", []): item["score"] += 500 # native language! if item["score"] > 500 or manual_select: newdata.append(item) results = sorted(newdata, key=itemgetter("score"), reverse=True) if results and manual_select: # show selectdialog to manually select the item results_list = [] for item in results: title = item["name"] if "name" in item else item["title"] if item.get("premiered"): year = item["premiered"].split("-")[0] else: year = item.get("first_air_date", "").split("-")[0] if item["poster_path"]: thumb = "http://image.tmdb.org/t/p/original%s" % item[ "poster_path"] else: thumb = "" label = "%s (%s) - %s" % (title, year, item["media_type"]) listitem = xbmcgui.ListItem(label=label, iconImage=thumb, label2=item["overview"]) results_list.append(listitem) if manual_select and results_list: dialog = DialogSelect("DialogSelect.xml", "", listing=results_list, window_title="%s - TMDB" % xbmc.getLocalizedString(283)) dialog.doModal() selected_item = dialog.result del dialog if selected_item != -1: details = results[selected_item] else: results = [] if not details and results: # just grab the first item as best match details = results[0] return details
def lookup_tvdb(self, searchtitle, channel, manual_select=False, tempmanualselect=False): """helper to select a match on tvdb""" tvdb_match = None searchtitle = searchtitle.lower() tvdb_result = self._mutils.thetvdb.search_series(searchtitle, True) searchchannel = channel.lower().split("hd")[0].replace(" ", "") if " FHD" in channel: searchchannel = channel.lower().split("fhd")[0].replace(" ", "") if " HD" in channel: searchchannel = channel.lower().split("hd")[0].replace(" ", "") if " SD" in channel: searchchannel = channel.lower().split("sd")[0].replace(" ", "") match_results = [] if tvdb_result: for item in tvdb_result: item["score"] = 0 if not item["seriesName"]: continue # seriesname can be None in some conditions itemtitle = item["seriesName"].lower() if not item["network"]: continue # network can be None in some conditions network = item["network"].lower().replace(" ", "") # high score if channel name matches if network in searchchannel or searchchannel in network: item["score"] += 800 # exact match on title - very high score if searchtitle == itemtitle: item["score"] += 1000 # match title by replacing some characters if re.sub('\*|,|.\"|\'| |:|;', '', searchtitle) == re.sub('\*|,|.\"|\'| |:|;', '', itemtitle): item["score"] += 750 # add SequenceMatcher score to the results stringmatchscore = SM(None, searchtitle, itemtitle).ratio() if stringmatchscore > 0.7: item["score"] += stringmatchscore * 500 # prefer items with native language as we've searched with localized info enabled try: if item["overview"]: item["score"] += 250 except KeyError: log_msg( "pvrartwork.py - Overview Key Error in lookup_tvb: %s" % searchchannel) # prefer items with artwork if item["banner"]: item["score"] += 1 if item["score"] > 500 or manual_select: match_results.append(item) # sort our new list by score match_results = sorted(match_results, key=itemgetter("score"), reverse=True) # original code: if match_results and manual_select:. part of "auto refresh" fix. if match_results and manual_select and tempmanualselect: # show selectdialog to manually select the item listitems = [] for item in match_results: thumb = "http://thetvdb.com%s" % item["poster"] if item[ "poster"] else "" try: listitem = xbmcgui.ListItem(label=item["seriesName"], label2=item["overview"]) except KeyError: listitem = xbmcgui.ListItem(label=item["seriesName"]) log_msg( "pvrartwork.py - Overview Key Error in lookup_tvb: %s" % searchchannel) listitem.setArt({'icon': thumb}) listitems.append(listitem) dialog = DialogSelect("DialogSelect.xml", "", listing=listitems, window_title="%s - TVDB" % xbmc.getLocalizedString(283)) dialog.doModal() selected_item = dialog.result del dialog if selected_item != -1: tvdb_match = match_results[selected_item]["id"] else: match_results = [] if not tvdb_match and match_results: # just grab the first item as best match tvdb_match = match_results[0]["id"] return tvdb_match
def manual_set_pvr_artwork(self, title, channel, genre): '''manual override artwork options''' artwork = self.get_pvr_artwork(title, channel, genre) cache_str = artwork["cachestr"] # show dialogselect with all artwork options abort = False while not abort: listitems = [] for arttype in ["thumb", "poster", "fanart", "banner", "clearart", "clearlogo", "discart", "landscape", "characterart"]: listitem = xbmcgui.ListItem(label=arttype, iconImage=artwork["art"].get(arttype, "")) listitem.setProperty("icon", artwork["art"].get(arttype, "")) listitems.append(listitem) dialog = DialogSelect("DialogSelect.xml", "", listing=listitems, windowtitle=xbmc.getLocalizedString(13511), multiselect=False) dialog.doModal() selected_item = dialog.result del dialog if selected_item == -1: abort = True else: # show results for selected art type artoptions = [] selected_item = listitems[selected_item] image = selected_item.getProperty("icon").decode("utf-8") label = selected_item.getLabel().decode("utf-8") heading = "%s: %s" % (xbmc.getLocalizedString(13511), label) if image: # current image listitem = xbmcgui.ListItem(label=xbmc.getLocalizedString(13512), iconImage=image) listitem.setProperty("icon", image) artoptions.append(listitem) # none option listitem = xbmcgui.ListItem(label=xbmc.getLocalizedString(231), iconImage="DefaultAddonNone.png") listitem.setProperty("icon", "DefaultAddonNone.png") artoptions.append(listitem) # browse option listitem = xbmcgui.ListItem(label=xbmc.getLocalizedString(1024), iconImage="DefaultFolder.png") listitem.setProperty("icon", "DefaultFolder.png") artoptions.append(listitem) # add remaining images as option allarts = artwork["art"].get(label + "s", []) if len(allarts) > 1: for item in allarts: listitem = xbmcgui.ListItem(label=item, iconImage=item) listitem.setProperty("icon", item) artoptions.append(listitem) dialog = DialogSelect("DialogSelect.xml", "", listing=artoptions, window_title=heading) dialog.doModal() selected_item = dialog.result del dialog if image and selected_item == 1: artwork["art"][label] = "" elif image and selected_item > 2: artwork["art"][label] = artoptions[selected_item].getProperty("icon").decode("utf-8") elif (image and selected_item == 2) or not image and selected_item == 0: # manual browse... dialog = xbmcgui.Dialog() image = dialog.browse(2, xbmc.getLocalizedString(1030), 'files', mask='.gif|.png|.jpg').decode("utf-8") del dialog if image: artwork["art"][label] = image # save results in cache self.artutils.cache.set(cache_str, artwork, expiration=timedelta(days=120))