fixed reading specific level sorting conditions

CHANGELOG

@@ -6,3 +6,6 @@
 - Global options stored in ~/.tdrc.
 - Dropped colorama.
 - Sorting by multiple conditions.
+
+== 0.2.1 (12.04.2013):
+- Fixed reading specifing level sorting conditions from .td/.tdrc files.

td/model.py

@@ -339,7 +339,8 @@ def _modify(submodel, i):
                         _new.append(_mark(v, i) + [_modify(v[4], i + 1)])
                 else:
                     _new.append(_mark(v, i) + [_modify(v[4], i + 1)])
-            for index, reverse in sortLevels.get(i) or sortAll:
+            levels = sortLevels.get(i) or sortLevels.get(str(i))
+            for index, reverse in levels or sortAll:
                 _new = sorted(_new, key=lambda e: e[index], reverse=reverse)
             return _new
         return _modify(self.data, 1)

tests/test_model.py

@@ -368,6 +368,15 @@ def test_sort_second_level_by_name_and_priority(self):
             ["testname3", 5, "", False, []]
         ]
 
+    def test_sort_with_level_specified_as_string(self):
+        # JSON corner case
+        sort = ([], {"1": [(0, True)]})
+        result = self.model.modify(sort=sort)
+        assert result == [
+            ["testname2", 3, "", False, []],
+            ["testname1", 4, "", True, []]
+        ]
+
     def test_done_all_levels(self):
         self.addSecondLevel()
         done = ([(None, None, True)], {})