1 files changed, 5 insertions, 2 deletions

diff --git a/sta.py b/sta.py
index 3f244e8..4b0ee7e 100755
--- a/sta.py
+++ b/sta.py
@@ -67,11 +67,14 @@ class Trace:
                 return False
         self.iter = filter(func, self.iter)
 
-    def sort(self, key):
-        """Return the records sorted by some key""" 
+    def sort(self, key, alt=None):
+        """Return the records sorted by some key, or if that key does not exist,
+            an alternate.""" 
         def sortfunc(record):
             if key in record:
                 return record[key]
+            if alt in record:
+                return record[alt]
             else:
                 return 0
         self.iter = sorted(self.iter, key=sortfunc)