From 6480138f1b198220ef8618086bdbbe12f9057960 Mon Sep 17 00:00:00 2001
From: chriszero <zerov83@gmail.com>
Date: Mon, 2 Sep 2013 21:52:17 +0200
Subject: [PATCH] oscreen Episoden (z.B. 1-3)

---
 tvsp2xmltv/model.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tvsp2xmltv/model.py b/tvsp2xmltv/model.py
index f45ad4b..556e590 100644
--- a/tvsp2xmltv/model.py
+++ b/tvsp2xmltv/model.py
@@ -153,9 +153,12 @@ def get_xml(self):
 
         try:
             if self.folge:
+                text = self.__generate_xmltv_ns()
                 tmp = SubElement(programme, 'episode-num', {'system': 'xmltv_ns'})
-                tmp.text = self.__generate_xmltv_ns()
+                tmp.text = text
         except ValueError:
+            tmp = SubElement(programme, 'episode-num', {'system': 'onscreen'})
+            tmp.text = self.folge
             pass
 
         if self.sz_hdtv:
@@ -174,7 +177,7 @@ def get_xml(self):
 
 
     def __generate_xmltv_ns(self):
-        # ToDo: folgen mit 111;112 behandeln (Doppelfolgen?) Prüfen ob nur Zahlen im String sind
+        # ToDo: folgen mit 111;112 & 1-3 behandeln (Doppelfolgen?) Prüfen ob nur Zahlen im String sind
         if self.folge:
             ep = int(self.folge) - 1
         else: