import changed

Author: fakechris

src/pyres/__init__.py

@@ -4,6 +4,28 @@
 import simplejson
 
 import types
+
+def my_import(name):
+    mod = __import__(name)    
+    components = name.split('.')    
+    for comp in components[1:]:        
+        mod = getattr(mod, comp)    
+    return mod
+
+def safe_str_to_class(s):
+    lst = s.split(".")
+    klass = lst[-1]
+    mod_list = lst[:-1]
+    module = ".".join(mod_list)
+    try:
+        mod = my_import(module)
+        if hasattr(mod, klass):
+            return getattr(mod, klass)
+        else:
+            return None
+    except ImportError:
+        return None
+
 def str_to_class(s):
     lst = s.split(".")
     klass = lst[-1]

src/pyres/job.py

@@ -1,4 +1,4 @@
-from pyres import ResQ, str_to_class
+from pyres import ResQ, str_to_class, safe_str_to_class
 from pyres.failure import Failure
 class Job(object):
     def __init__(self, queue, payload, resq, worker):
@@ -9,7 +9,7 @@ def __init__(self, queue, payload, resq, worker):
 
     def perform(self):
         payload_class_str = self._payload["class"]
-        payload_class = str_to_class(payload_class_str)
+        payload_class = safe_str_to_class(payload_class_str)
         args = self._payload.get("args", None)
         if args:
             return payload_class.perform(*args)