Added some additional support to the pyepg and eit grabbers.

diff --git a/src/epggrab/eit.c b/src/epggrab/eit.c
index 9b1a8bf41a7cb888100af7030efaacdba2c4ce45..6bbcb167a745305e942fda3dd1dfad0a5099431b 100644 (file)
--- a/src/epggrab/eit.c
+++ b/src/epggrab/eit.c
@@ -37,7 +37,8 @@ static void _eit_episode_uri
 void eit_callback ( channel_t *ch, int id, time_t start, time_t stop,
                     const char *title, const char *desc,
                     const char *extitem, const char *extdesc,
-                    const char *exttext ) {
+                    const char *exttext,
+                    const uint8_t *genre, int genre_cnt ) {
   int save = 0;
   epg_broadcast_t *ebc;
   epg_episode_t *ee;
@@ -76,6 +77,8 @@ void eit_callback ( channel_t *ch, int id, time_t start, time_t stop,
       save |= epg_episode_set_summary(ee, summary);
     if (description)
       save |= epg_episode_set_description(ee, description);
+    if (genre_cnt)
+      save |= epg_episode_set_genre(ee, genre, genre_cnt);
 
     /* Update */
     save |= epg_broadcast_set_episode(ebc, ee);

diff --git a/src/epggrab/eit.h b/src/epggrab/eit.h
index c30341e5e84df4b8acd35d42b5ba91964c630c40..bc8b17c8d499d678059b06a6dd9e340585ec4a3c 100644 (file)
--- a/src/epggrab/eit.h
+++ b/src/epggrab/eit.h
@@ -26,6 +26,7 @@ void eit_init ( epggrab_module_list_t *list );
 void eit_callback ( struct channel *ch, int id, time_t start, time_t stop,
                     const char *title, const char *desc,
                     const char *extitem, const char *extdesc,
-                    const char *exttext );
+                    const char *exttext,
+                    const uint8_t *genres, int genre_cnt );
 
 #endif

diff --git a/src/epggrab/pyepg.c b/src/epggrab/pyepg.c
index c89d78f945fe1ffd9ad176ac5db1b1ed363fc3a4..5921a75145c0cd552dd3a93a27e299b5115d58df 100644 (file)
--- a/src/epggrab/pyepg.c
+++ b/src/epggrab/pyepg.c
@@ -53,7 +53,12 @@ static int _pyepg_parse_time ( const char *str, time_t *out )
   return ret;
 }
 
-/* TODO: currently limited to at most 10 services */
+static const uint8_t *_pyepg_parse_genre ( htsmsg_t *tags, int *cnt )
+{
+  // TODO: implement this
+  return NULL;
+}
+
 static int _pyepg_parse_channel ( htsmsg_t *data, epggrab_stats_t *stats )
 {
   int save = 0;
@@ -66,8 +71,6 @@ static int _pyepg_parse_channel ( htsmsg_t *data, epggrab_stats_t *stats )
   uint16_t sid[10];
   int sid_idx = 0, sname_idx = 0;
   
-  
-
   if ( data == NULL ) return 0;
 
   if ((attr    = htsmsg_get_map(data, "attrib")) == NULL) return 0;
@@ -148,12 +151,12 @@ static int _pyepg_parse_brand ( htsmsg_t *data, epggrab_stats_t *stats )
     save |= epg_brand_set_summary(brand, str);
   }
   
-  /* Set icon */
-#if TODO_ICON_SUPPORT
-  if ((str = htsmsg_xml_get_cdata_str(tags, "icon"))) {
-    save |= epg_brand_set_icon(brand, str);
+  /* Set image */
+  if ((str = htsmsg_xml_get_cdata_str(tags, "image"))) {
+    save |= epg_brand_set_image(brand, str);
+  } else if ((str = htsmsg_xml_get_cdata_str(tags, "thumb"))) {
+    save |= epg_brand_set_image(brand, str);
   }
-#endif
 
   /* Set season count */
   if (htsmsg_xml_get_cdata_u32(tags, "series-count", &u32) == 0) {
@@ -192,22 +195,17 @@ static int _pyepg_parse_season ( htsmsg_t *data, epggrab_stats_t *stats )
     }
   }
 
-  /* Set title */
-#if TODO_EXTRA_METADATA
-  if ((str = htsmsg_xml_get_cdata_str(tags, "title"))) {
-    save |= epg_season_set_title(season, str);
-  } 
-
   /* Set summary */
   if ((str = htsmsg_xml_get_cdata_str(tags, "summary"))) {
     save |= epg_season_set_summary(season, str);
   }
   
-  /* Set icon */
-  if ((str = htsmsg_xml_get_cdata_str(tags, "icon"))) {
-    save |= epg_season_set_icon(season, str);
+  /* Set image */
+  if ((str = htsmsg_xml_get_cdata_str(tags, "image"))) {
+    save |= epg_season_set_image(season, str);
+  } else if ((str = htsmsg_xml_get_cdata_str(tags, "thumb"))) {
+    save |= epg_season_set_image(season, str);
   }
-#endif
 
   /* Set season number */
   if (htsmsg_xml_get_cdata_u32(tags, "number", &u32) == 0) {
@@ -226,13 +224,14 @@ static int _pyepg_parse_season ( htsmsg_t *data, epggrab_stats_t *stats )
 
 static int _pyepg_parse_episode ( htsmsg_t *data, epggrab_stats_t *stats )
 {
-  int save = 0;
+  int genre_cnt, save = 0;
   htsmsg_t *attr, *tags;
   epg_episode_t *episode;
   epg_season_t *season;
   epg_brand_t *brand;
   const char *str;
-  uint32_t u32;
+  uint32_t u32, pc, pn;
+  const uint8_t *genre;
 
   if ( data == NULL ) return 0;
 
@@ -276,15 +275,23 @@ static int _pyepg_parse_episode ( htsmsg_t *data, epggrab_stats_t *stats )
   if (htsmsg_xml_get_cdata_u32(tags, "number", &u32) == 0) {
     save |= epg_episode_set_number(episode, u32);
   }
+  if (!htsmsg_xml_get_cdata_u32(tags, "part-number", &pn)) {
+    pc = 0;
+    htsmsg_xml_get_cdata_u32(tags, "part-count", &pc);
+    save |= epg_episode_set_part(episode, pn, pc);
+  }
+
+  /* Set image */
+  if ((str = htsmsg_xml_get_cdata_str(tags, "image"))) {
+    save |= epg_episode_set_image(episode, str);
+  } else if ((str = htsmsg_xml_get_cdata_str(tags, "thumb"))) {
+    save |= epg_episode_set_image(episode, str);
+  }
 
   /* Genre */
-  // TODO: can actually have multiple!
-#if TODO_GENRE_SUPPORT
-  if ((str = htsmsg_xml_get_cdata_str(tags, "genre"))) {
-    // TODO: conversion?
-    save |= epg_episode_set_genre(episode, str);
+  if ((genre = _pyepg_parse_genre(tags, &genre_cnt))) {
+    save |= epg_episode_set_genre(episode, genre, genre_cnt);
   }
-#endif
 
   /* TODO: extra metadata */