Added --limit to buy command

eyeonus · Nov 20, 2014 · 3ec13d1 · 3ec13d1
1 parent 16ec9d0
commit 3ec13d1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/commands/buy_cmd.py b/commands/buy_cmd.py
@@ -30,6 +30,11 @@
             metavar='N.NN',
             type=float,
         ),
+    ParseArgument('--limit',
+            help='Maximum number of results to list.',
+            default=None,
+            type=int,
+        ),
     MutuallyExclusiveGroup(
         ParseArgument('--price-sort', '-P',
                 help='(When using --near) Sort by price not distance',
@@ -129,6 +134,10 @@ def run(results, cmdenv, tdb):
             results.summary.sort = "Dist"
             results.rows.sort(key=lambda result: result.dist)
 
+    limit = cmdenv.limit or 0
+    if limit > 0:
+        results.rows = results.rows[:limit]
+
     return results
 
 #######################################################################