Adding basic tracebacks

mathom · mathom · commit ae2e19f8b50c · 2014-10-15T13:19:50.000-05:00
diff --git a/stackify/error.py b/stackify/error.py
@@ -1,3 +1,6 @@
+import traceback
+import sys
+
 from stackify.formats import JSONObject
 
 
@@ -11,12 +14,17 @@ def __init__(self):
         self.StackTrace = [] # array of TraceFrames
         self.InnerError = None # cause?
 
+    def load_stack(self, tb):
+        stacks = traceback.extract_tb(tb)
+        for filename, lineno, parent, method in stacks:
+            self.StackTrace.append(TraceFrame(filename, lineno, method))
+
 
 class TraceFrame(JSONObject):
-    def __init__(self):
-        self.CodeFileName = None
-        self.LineNum = None
-        self.Method = None
+    def __init__(self, filename, lineno, method):
+        self.CodeFileName = filename
+        self.LineNum = lineno
+        self.Method = method
 
 
 class WebRequestDetail(JSONObject):
@@ -47,3 +55,8 @@ def __init__(self):
         self.CustomerName = None
         self.UserName = None
 
+    def load_exception(self):
+        self.Error = ErrorItem()
+        type_, value, tb = sys.exc_info()
+        self.Error.load_stack(tb)
+
