Fix simple robot with timeout

[bookmarks_db.git] / Robots / bkmk_rurllib_to.py

diff --git a/Robots/bkmk_rurllib_to.py b/Robots/bkmk_rurllib_to.py
new file mode 100644 (file)
index 0000000..2726dbf
--- /dev/null
+++ b/Robots/bkmk_rurllib_to.py
@@ -0,0 +1,28 @@
+"""Simple robot with socket's timeout
+
+This file is a part of Bookmarks database and Internet robot.
+
+"""
+
+__author__ = "Oleg Broytman <phd@phdru.name>"
+__copyright__ = "Copyright (C) 2000-2014 PhiloSoft Design"
+__license__ = "GNU GPL"
+
+__all__ = ['robot_urllib_to']
+
+
+import socket
+socket.setdefaulttimeout(900)
+
+from .bkmk_rurllib import robot_urllib, get_error
+
+
+class robot_urllib_to(robot_urllib):
+   def check_url(self, bookmark):
+      try:
+         return robot_urllib.check_url(self, bookmark)
+      except socket.error, msg:
+         bookmark.error = get_error(msg)
+         self.log(bookmark.error)
+
+      return 1