diff --git a/sickbeard/metadata/kodi_12plus.py b/sickbeard/metadata/kodi_12plus.py index 87fe23f1fd7298e7751608cae5142aff8a763a15..4ba5ceb762fa195d01e1d7e50e9cc534e902c810 100644 --- a/sickbeard/metadata/kodi_12plus.py +++ b/sickbeard/metadata/kodi_12plus.py @@ -19,6 +19,7 @@ import re import datetime +from babelfish import Country import sickbeard from sickbeard.metadata import generic @@ -94,7 +95,7 @@ class KODI_12PlusMetadata(generic.GenericMetadata): @staticmethod def _split_info(info_string): - return {x.strip() for x in re.sub(r'[,/]*', '|', info_string).split('|') if x.strip()} + return {x.strip().title() for x in re.sub(r'[,/]+', '|', info_string).split('|') if x.strip()} def _show_data(self, show_obj): """ @@ -178,6 +179,16 @@ class KODI_12PlusMetadata(generic.GenericMetadata): cur_genre = etree.SubElement(tv_node, "genre") cur_genre.text = genre + if 'country_codes' in show_obj.imdb_info: + for country in self._split_info(show_obj.imdb_info['country_codes']): + try: + cur_country_name = Country(country.upper()).name.title() + except Exception: + continue + + cur_country = etree.SubElement(tv_node, "country") + cur_country.text = cur_country_name + if getattr(myShow, 'firstaired', None): premiered = etree.SubElement(tv_node, "premiered") premiered.text = myShow["firstaired"]