Finished sort2 to sort by two keys

1 files changed, 11 insertions, 0 deletions

diff --git a/sta.py b/sta.py
index 684d0ac..aec249b 100755
--- a/sta.py
+++ b/sta.py
@@ -75,6 +75,17 @@ class Trace:
             else:
                 return 0
         self.iter = sorted(self.iter, key=sortfunc)
+    
+    def sort2(self, key1, key2):
+        """Return the records sorted by two keys""" 
+        def sortfunc(record):
+            score = 0
+            if key1 in record:
+                score += record[key1]
+            if key2 in record:
+                score = float(str(score) + '.' + str(record[key2]))
+            return score
+        self.iter = sorted(self.iter, key=sortfunc)
 
     def print_records(self):
         """Prints all records in the trace"""