Fix menu (#93)

.pylintrc

@@ -17,6 +17,7 @@ disable=
     too-many-statements,
     use-maxsplit-arg,
     consider-using-from-import,
+    unspecified-encoding,
 
     super-with-arguments, # Python 2.7 compatibility
-    raise-missing-from, # Python 2.7 compatibility
+    raise-missing-from, # Python 2.7 compatibility

resources/lib/modules/iptvmanager.py

@@ -66,7 +66,7 @@ def send_epg():  # pylint: disable=no-method-argument
 
         today = datetime.today()
 
-        results = dict()
+        results = {}
         for key, channel in CHANNELS.items():
             iptv_id = channel.get('iptv_id')
 

resources/lib/modules/menu.py

@@ -128,14 +128,7 @@ def generate_titleitem(item):
             }
 
             visible = True
-            if isinstance(item.episodes, list) and not item.episodes:
-                # We know that we don't have episodes
-                title = '[COLOR gray]' + item.title + '[/COLOR]'
-                visible = kodiutils.get_setting_bool('interface_show_unavailable')
-
-            else:
-                # We have episodes, or we don't know it
-                title = item.title
+            title = item.title
 
             context_menu = []
             if item.uuid:

resources/lib/viervijfzes/content.py

@@ -454,30 +454,33 @@ def get_recommendation_categories(self):
 
         # Categories regexes
         regex_articles = re.compile(r'<article[^>]+>(.*?)</article>', re.DOTALL)
-        regex_category = re.compile(r'<h1.*?>(.*?)</h1>(?:.*?<div class="visually-hidden">(.*?)</div>)?', re.DOTALL)
+        regex_category = re.compile(r'<h1.*?>(.*?)</h1>(?:.*?<div class=\"visually-hidden\">(.*?)</div>)?', re.DOTALL)
 
         categories = []
         for result in regex_articles.finditer(raw_html):
             article_html = result.group(1)
 
             match_category = regex_category.search(article_html)
-            category_title = match_category.group(1).strip()
-            if match_category.group(2):
-                category_title += ' [B]%s[/B]' % match_category.group(2).strip()
-
-            # Extract programs and lookup in all_programs so we have more metadata
-            programs = []
-            for program in self._extract_programs(article_html):
-                try:
-                    rich_program = next(rich_program for rich_program in all_programs if rich_program.path == program.path)
-                    programs.append(rich_program)
-                except StopIteration:
-                    programs.append(program)
-
-            episodes = self._extract_videos(article_html)
-
-            categories.append(
-                Category(uuid=hashlib.md5(category_title.encode('utf-8')).hexdigest(), title=category_title, programs=programs, episodes=episodes))
+            category_title = None
+            if match_category:
+                category_title = match_category.group(1).strip()
+                if match_category.group(2):
+                    category_title += ' [B]%s[/B]' % match_category.group(2).strip()
+
+            if category_title:
+                # Extract programs and lookup in all_programs so we have more metadata
+                programs = []
+                for program in self._extract_programs(article_html):
+                    try:
+                        rich_program = next(rich_program for rich_program in all_programs if rich_program.path == program.path)
+                        programs.append(rich_program)
+                    except StopIteration:
+                        programs.append(program)
+
+                episodes = self._extract_videos(article_html)
+
+                categories.append(
+                    Category(uuid=hashlib.md5(category_title.encode('utf-8')).hexdigest(), title=category_title, programs=programs, episodes=episodes))
 
         return categories
 
@@ -595,35 +598,35 @@ def _parse_program_data(data):
         """
         # Create Program info
         program = Program(
-            uuid=data['id'],
-            path=data['link'].lstrip('/'),
-            channel=data['pageInfo']['brand'],
-            title=data['title'],
-            description=html_to_kodi(data['description']),
-            aired=datetime.fromtimestamp(data.get('pageInfo', {}).get('publishDate')),
-            poster=data['images']['poster'],
-            thumb=data['images']['teaser'],
-            fanart=data['images']['hero'],
+            uuid=data.get('id'),
+            path=data.get('link').lstrip('/'),
+            channel=data.get('pageInfo').get('brand'),
+            title=data.get('title'),
+            description=html_to_kodi(data.get('description')),
+            aired=datetime.fromtimestamp(data.get('pageInfo', {}).get('publishDate', 0.0)),
+            poster=data.get('images').get('poster'),
+            thumb=data.get('images').get('teaser'),
+            fanart=data.get('images').get('teaser'),
         )
 
         # Create Season info
         program.seasons = {
             key: Season(
-                uuid=playlist['id'],
-                path=playlist['link'].lstrip('/'),
-                channel=playlist['pageInfo']['brand'],
-                title=playlist['title'],
+                uuid=playlist.get('id'),
+                path=playlist.get('link').lstrip('/'),
+                channel=playlist.get('pageInfo').get('brand'),
+                title=playlist.get('title'),
                 description=html_to_kodi(playlist.get('description')),
-                number=playlist['episodes'][0]['seasonNumber'],  # You did not see this
+                number=playlist.get('episodes')[0].get('seasonNumber'),  # You did not see this
             )
-            for key, playlist in enumerate(data['playlists']) if playlist['episodes']
+            for key, playlist in enumerate(data.get('playlists', [])) if playlist.get('episodes')
         }
 
         # Create Episodes info
         program.episodes = [
-            ContentApi._parse_episode_data(episode, playlist['id'])
-            for playlist in data['playlists']
-            for episode in playlist['episodes']
+            ContentApi._parse_episode_data(episode, playlist.get('id'))
+            for playlist in data.get('playlists', [])
+            for episode in playlist.get('episodes')
         ]
 
         return program