Auto-restart echobot on error

echobot/echobot.py

@@ -5,51 +5,60 @@ from selenium import webdriver
 PHANTOMJS_PATH = "tools/phantomjs/bin/phantomjs"
 SITE_URL = "http://127.0.0.1:43110"
 
-browser = webdriver.PhantomJS(executable_path=PHANTOMJS_PATH, service_log_path=os.path.devnull)
-browser.set_window_size(1400, 1000)
-
-browser.get("%s/1MaiL5gfBM1cyb4a8e3iiL8L5gXmoAJu27" % SITE_URL)
-print " * LocalStorage:", browser.execute_script("return JSON.stringify(localStorage)")
-
-# Switch to inner frame
-browser.switch_to.frame(browser.find_element_by_id("inner-iframe"))
-
-# Check if Inbox is active
-assert browser.execute_script("return window.Page.message_lists.getActive().title") == u"Inbox"
-
-# Setup new message checking script
-browser.execute_script("""
-window.replyUnreadMessage = function(cb) {
-	var messages = window.Page.message_lists.getActive().messages
-	var unread_messages = messages.filter(function(message) { return message.read == false })
-	if (unread_messages.length == 0) {
-		cb(false)
-		return false
+def openBrowser():
+	browser = webdriver.PhantomJS(executable_path=PHANTOMJS_PATH, service_log_path=os.path.devnull)
+	browser.set_window_size(1400, 1000)
+
+	browser.get("%s/1MaiL5gfBM1cyb4a8e3iiL8L5gXmoAJu27" % SITE_URL)
+	print " * LocalStorage:", browser.execute_script("return JSON.stringify(localStorage)")
+
+	# Switch to inner frame
+	browser.switch_to.frame(browser.find_element_by_id("inner-iframe"))
+
+	# Check if Inbox is active
+	assert browser.execute_script("return window.Page.message_lists.getActive().title") == u"Inbox"
+
+	# Setup new message checking script
+	browser.execute_script("""
+	window.replyUnreadMessage = function(cb) {
+		var messages = window.Page.message_lists.getActive().messages
+		var unread_messages = messages.filter(function(message) { return message.read == false })
+		if (unread_messages.length == 0) {
+			cb(false)
+			return false
+		}
+		message = unread_messages[0]
+		console.log("New unread message:", message.key)
+		message.handleListClick()
+		message.handleReplyClick()
+		Page.message_create.body = "This is an echo of your message:\\n> " + message.row.body.replace(/\\n/g, "\\n> ") + "\\n\\nGreetings:\\nThe ZeroMail echo bot"
+		Page.projector.scheduleRender()
+		setTimeout(function() {
+			Page.message_create.handleSendClick()
+			cb(true)
+		}, 1000)
 	}
-	message = unread_messages[0]
-	console.log("New unread message:", message.key)
-	message.handleListClick()
-	message.handleReplyClick()
-	Page.message_create.body = "This is an echo of your message:\\n> " + message.row.body.replace(/\\n/g, "\\n> ") + "\\n\\nGreetings:\\nThe ZeroMail echo bot"
-	Page.projector.scheduleRender()
-	setTimeout(function() {
-		Page.message_create.handleSendClick()
-		cb(true)
-	}, 1000)
-}
-
-window.replyTimer = function() {
-	setTimeout(function() { replyUnreadMessage(replyTimer) }, 1000)
-}
-
-replyTimer()
-""")
-
-browser.switch_to.default_content()
+
+	window.replyTimer = function() {
+		setTimeout(function() { replyUnreadMessage(replyTimer) }, 1000)
+	}
+
+	replyTimer()
+	""")
+
+	browser.switch_to.default_content()
+	return browser
+
+browser = openBrowser()
 last_log_line = 0
 while 1:
-	lines = browser.get_log("browser")
-	for line in lines[last_log_line:]:
-		print line["message"].replace("(:)", "")
-		last_log_line += 1
+	try:
+		lines = browser.get_log("browser")
+		for line in lines[last_log_line:]:
+			print line["message"].replace("(:)", "")
+			last_log_line += 1
+	except Exception, err:
+		print "Error", err
+		browser = openBrowser()
+		last_log_line = 0
 	time.sleep(5)