From 832574e06bc344b0db3815cc0a171f34ee74e650 Mon Sep 17 00:00:00 2001
From: Stefan Bethke <stb@lassitu.de>
Date: Sun, 2 Mar 2014 22:19:36 +0000
Subject: [PATCH] Alle neuen Shows am Stueck downloaden, dafuer aber mit
 Timeout von drei Stunden pro Show.

---
 src/tivomirror/tivomirror | 43 ++++++++++++++++++++++++++++++++-------
 1 file changed, 36 insertions(+), 7 deletions(-)

diff --git a/src/tivomirror/tivomirror b/src/tivomirror/tivomirror
index fa3dd5d..2f4250c 100755
--- a/src/tivomirror/tivomirror
+++ b/src/tivomirror/tivomirror
@@ -1,16 +1,22 @@
 #!/usr/local/bin/python
 
-# $Schlepperbande: src/tivomirror/tivomirror,v 1.50 2013/03/16 09:54:12 stb Exp $
+# $Schlepperbande: src/tivomirror/tivomirror,v 1.51 2013/08/06 22:17:16 stb Exp $
 #
 # Stefans Script, um die Sendungen vom Tivo runterzuladen und in MPEG4
 # zu transkodieren.
 # Wird auf disklesslibber per Crontab-Eintrag stuendlich gestartet:
 # flock -n /tmp/tivomirror.log -c 'tivomirror >.tivomirror.log 2>&1 </dev/null'
 
+import sys
+reload(sys)  
+sys.setdefaultencoding('utf-8')
+
 import anydbm
 import cookielib
 import datetime
 import getopt
+import errno
+import functools
 import os
 import re
 import requests
@@ -34,25 +40,25 @@ arset = dict()
 arset["Futurama"] = 169
 
 includes = dict()
-includes['Breaking Bad'] = 1
+includes['Almost Human'] = 1
+includes['Brooklyn Nine-Nine'] = 1
 includes['Castle'] = 1
-includes['Dirty Jobs'] = 1
 includes['Family Guy'] = 1
-includes['Futurama'] = 1
 includes["John Oliver's New York Stand-Up Show"] = 1
 includes['Late Show With David Letterman'] = 1
 includes['Louie'] = 1
 includes['Mad Men'] = 1
-includes['Mockingbird Lane'] = 1
 includes['Modern Family'] = 1
 includes['MythBusters'] = 1
 includes['NCIS'] = 1
 includes['NFL Football'] = 1
+includes['Person of Interest'] = 1
 includes['Sesame Street'] = 1
 includes['The Big Bang Theory'] = 1
 includes['The Colbert Report'] = 1
 includes['The Daily Show With Jon Stewart'] = 1
 includes['The Late Late Show With Craig Ferguson'] = 1
+includes['The Tonight Show Starring Jimmy Fallon'] = 1
 
 includes['CONSTITUTION USA With Peter Sagal'] = 1;
 
@@ -77,6 +83,28 @@ session.verify = False
 session.auth = requests.auth.HTTPDigestAuth("tivo", mak)
 
 
+class TimeoutError(Exception):
+    pass
+
+def timeout(seconds=10, error_message=os.strerror(errno.ETIMEDOUT)):
+	def decorator(func):
+		def _handle_timeout(signum, frame):
+			raise TimeoutError(error_message)
+
+		def wrapper(*args, **kwargs):
+			signal.signal(signal.SIGALRM, _handle_timeout)
+			signal.alarm(seconds)
+			try:
+				result = func(*args, **kwargs)
+			finally:
+				signal.alarm(0)
+			return result
+
+		return functools.wraps(func)(wrapper)
+
+	return decorator
+
+
 def trimDescription(desc):
 	desc = desc.strip()
 	i = desc.rfind(". Copyright Tribune Media Services, Inc.");
@@ -202,6 +230,7 @@ def quit_process(pid):
 		pass
 
 
+@timeout(7200)
 def download(file, url, mak, target):
 	global browser, sidcookie
 	#url = re.sub("tivo.lassitu.de:80", "localhost:8888", url)
@@ -308,8 +337,8 @@ def mirror(dom, downloaddb):
 			if getattr(downloaddb, "sync", None) and callable(downloaddb.sync):
 				downloaddb.sync()
 			# stop after the first successful download since the tivo hangs easily
-			print "Stopping after one successful transfer"
-			break
+			#print "Stopping after one successful transfer"
+			#break
 		except TivoException, e:
 			print "Error processing \"%s\": %s" % (item.name, e)
 		break