X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=sort_db.py;h=143c7d1349c44be69406dc4137f3c1d88c10d23a;hb=22ff22eb1b087f197008ceaf926d952b226f07bf;hp=4418973011c9b1444c2670fdf4e9eebf1efb8566;hpb=c2ea4e82718b903aa123dd77490f36657383b0ca;p=bookmarks_db.git

diff --git a/sort_db.py b/sort_db.py
index 4418973..143c7d1 100755
--- a/sort_db.py
+++ b/sort_db.py
@@ -1,4 +1,4 @@
-#! /usr/bin/env python
+#! /usr/bin/env python3
 """Sort bookmarks DB according to a rule:
       -a - by AddDate
       -v - by LastVisit
@@ -19,23 +19,16 @@ __license__ = "GNU GPL"
 import sys
 
 
-class SortBy(object):
+class SortKey(object):
     def __init__(self, sort_by):
         self.sort_by = sort_by
 
-    def __call__(self, o1, o2):
+    def __call__(self, o1):
         try:
-            attr1 = int(getattr(o1, self.sort_by))
+            return int(getattr(o1, self.sort_by))
         except (AttributeError, TypeError, ValueError):
             return 1
 
-        try:
-            attr2 = int(getattr(o2, self.sort_by))
-        except (AttributeError, TypeError, ValueError):
-            return -1
-
-        return cmp(attr1, attr2)
-
 
 def walk_linear(linear, walker):
     for object in linear:
@@ -91,8 +84,8 @@ def run():
     linear = root_folder.linear
     del linear[0]  # exclude root folder from sorting
 
-    by = SortBy(sort_by)
-    linear.sort(by)
+    sort_key = SortKey(sort_by)
+    linear.sort(key=sort_key)
 
     from writers import writer
     output_filename = "%s-sorted_by-%s" % (writer.filename, sort_by)