From 80726b309665b5c8f1c5134119701a536f4c3a80 Mon Sep 17 00:00:00 2001
From: Dustyn Gibson <miigotu@gmail.com>
Date: Fri, 27 Nov 2015 21:37:16 -0800
Subject: [PATCH] Add country tag to KODI 12+ tvshow.nfo
---
sickbeard/metadata/kodi_12plus.py | 13 ++++++++++++-
1 file changed, 12 insertions(+), 1 deletion(-)
diff --git a/sickbeard/metadata/kodi_12plus.py b/sickbeard/metadata/kodi_12plus.py
index 87fe23f1f..4ba5ceb76 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"]
--
GitLab