From: W. Trevor King <wking@drexel.edu>
Date: Thu, 8 Mar 2012 15:24:03 +0000 (-0500)
Subject: Make `index` argument optional and return curve in `remove curve from playlist`.
X-Git-Url: http://git.tremily.us/?p=hooke.git;a=commitdiff_plain;h=5560a9e827a46ebc09b3cb539911aef8383b00a5

Make `index` argument optional and return curve in `remove curve from playlist`.
---

diff --git a/hooke/plugin/playlist.py b/hooke/plugin/playlist.py
index ca99339..334c133 100644
--- a/hooke/plugin/playlist.py
+++ b/hooke/plugin/playlist.py
@@ -337,7 +337,7 @@ class RemoveCommand (PlaylistCommand):
         super(RemoveCommand, self).__init__(
             name='remove curve from playlist',
             arguments=[
-                Argument(name='index', type='int', optional=False, help="""
+                Argument(name='index', type='int', optional=True, help="""
 Index of target curve.
 """.strip()),
                 ],
@@ -345,8 +345,11 @@ Index of target curve.
 
     def _run(self, hooke, inqueue, outqueue, params):
         playlist = self._playlist(hooke, params)
-        playlist.pop(params['index'])
+        if params['index'] is None:
+            params['index'] = playlist.index()
+        curve = playlist.pop(params['index'])
         playlist.jump(playlist.index())
+        outqueue.put(curve)
 
 
 class ApplyCommand (PlaylistCommand):