[voice] add on_complete callback to DCADOpusEncoderPlayable

b1naryth1ef · Jul 17, 2017 · 78ac2c4 · 78ac2c4
1 parent a522a90
commit 78ac2c4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/disco/voice/playable.py b/disco/voice/playable.py
@@ -227,6 +227,7 @@ class DCADOpusEncoderPlayable(BasePlayable, AbstractOpus, OpusEncoder):
     def __init__(self, source, *args, **kwargs):
         self.source = source
         self.command = kwargs.pop('command', 'dcad')
+        self.on_complete = kwargs.pop('on_complete', None)
         super(DCADOpusEncoderPlayable, self).__init__(*args, **kwargs)
 
         self._done = False
@@ -270,13 +271,15 @@ def next_frame(self):
         header = self.proc.stdout.read(OPUS_HEADER_SIZE)
         if len(header) < OPUS_HEADER_SIZE:
             self._done = True
+            self.on_complete()
             return
 
         size = struct.unpack('<h', header)[0]
 
         data = self.proc.stdout.read(size)
         if len(data) < size:
             self._done = True
+            self.on_complete()
             return
 
         return data