-

cool-RR · cool-RR · commit 28823003dca1 · 2012-12-24T01:47:23.000+02:00
diff --git a/python_toolbox/nifty_collections/ordered_dict.py b/python_toolbox/nifty_collections/ordered_dict.py
@@ -7,6 +7,7 @@
 See its documentation for more information.
 '''
 
+import __builtin__
 from UserDict import DictMixin
 
 from python_toolbox import comparison_tools
@@ -162,7 +163,7 @@ def move_to_end(self, key, last=True):
             end[2] = first[1] = link
 
     
-    def sort(self, key=None):
+    def sort(self, key=None, reversed=False):
         '''
         Sort the items according to their keys, changing the order in-place.
         
@@ -172,7 +173,8 @@ def sort(self, key=None):
         key_function = \
                    comparison_tools.process_key_function_or_attribute_name(key)
         sorted_keys = sorted(self.keys(), key=key_function)
-        for key_ in sorted_keys[1:]:
+        step = -1 if reversed else 1
+        for key_ in sorted_keys[1::step]:
             self.move_to_end(key_)
         
             
diff --git a/test_python_toolbox/test_nifty_collections/test_ordered_dict/test.py b/test_python_toolbox/test_nifty_collections/test_ordered_dict/test.py
@@ -41,6 +41,10 @@ def test_sort():
     second_ordered_dict.sort('imag')
     assert second_ordered_dict == \
                            OrderedDict(((3+1j, 'a'), (1+2j, 'b'), (2+3j, 'c')))
+    
+    second_ordered_dict.sort('real', reversed=True)
+    assert second_ordered_dict == \
+                           OrderedDict(((3+1j, 'a'), (2+3j, 'c'), (1+2j, 'b')))
 
     
     
