Mercurial > hg > auditok
changeset 276:f0252da17455
Refactor RegionSaverWorker
author | Amine Sehili <amine.sehili@gmail.com> |
---|---|
date | Sat, 21 Sep 2019 11:39:49 +0100 |
parents | a1388f0d18d3 |
children | 0a5374fcb700 |
files | auditok/workers.py |
diffstat | 1 files changed, 11 insertions(+), 6 deletions(-) [+] |
line wrap: on
line diff
--- a/auditok/workers.py Fri Sep 20 21:02:08 2019 +0200 +++ b/auditok/workers.py Sat Sep 21 11:39:49 2019 +0100 @@ -371,24 +371,29 @@ class RegionSaverWorker(Worker): def __init__( - self, name_format, filetype=None, timeout=0.2, logger=None, **kwargs + self, + filename_format, + audio_format=None, + timeout=0.2, + logger=None, + **audio_parameters ): - self._name_format = name_format - self._filetype = filetype - self._audio_kwargs = kwargs + self._filename_format = filename_format + self._audio_format = audio_format + self._audio_parameters = audio_parameters self._debug_format = '[SAVE]: Detection {id} saved as "{filename}"' Worker.__init__(self, timeout=timeout, logger=logger) def _process_message(self, message): _id, audio_region = message - filename = self._name_format.format( + filename = self._filename_format.format( id=_id, start=audio_region.meta.start, end=audio_region.meta.end, duration=audio_region.duration, ) filename = audio_region.save( - filename, self._filetype, **self._audio_kwargs + filename, self._audio_format, **self._audio_parameters ) if self._logger: message = self._debug_format.format(id=_id, filename=filename)