浏览代码

symuid-list: refactor

Fabian Peter Hammerle 6 年之前
父节点
当前提交
b5dec4dbf3
共有 1 个文件被更改,包括 3 次插入6 次删除
  1. 3 6
      symuid-list

+ 3 - 6
symuid-list

@@ -4,13 +4,10 @@
 import symuid
 
 
-def walk_attr(root_path, path_ignore_regex):
-    for track in symuid.Track.walk(root_path, path_ignore_regex):
-        yield {'path': track.path, 'play_count': track.get_play_count_sum()}
-
-
 def symuid_list(path, path_ignore_regex, filter_expression, sort_expression):
-    attr_it = walk_attr(path, path_ignore_regex)
+    # use generators until sort is required
+    attr_it = ({'path': track.path, 'play_count': track.get_play_count_sum()}
+               for track in symuid.Track.walk(path, path_ignore_regex))
     if filter_expression:
         attr_it = filter(lambda a: eval(filter_expression, a), attr_it)
     if sort_expression: