Source code for dscleaner.filewriter

from .utils import path_splitter
import os
import pyemddf
import pandas as pd
[docs]class FileWriter: """ Writes to a file. The class should be instantiated with a ``with`` statement. Args: file: Accepts either a FileUtil, or IFileInfo Specialization. mode: Allows for `w` for writing or `a` for appending. """ def __init__(self,file, mode = 'w'): from .ifileinfo import IFileInfo from .fileutil import FileUtil if(issubclass(type(file),IFileInfo)): self.file = file elif(issubclass(type(file),FileUtil)): self.file = file.file else: raise TypeError("Wrong argument type") self.mode = mode def __enter__(self): return self def __exit__(self,type,value,traceback): self.close()
[docs] def close(self): #self.file.close() pass
[docs] def create_file(self,new_filepath,samplerate = None): """ Creates a new file with the extension given in ``new_filepath``. If the source file isn't a soundfile the target file will not be well formated. In order to normalize, you should run ``FileUtil.standardize`` method before. Args: new_filepath - The diretory and name the new file will have, it will convert based on file extension. samplerate (Optional) - The samplerate the file should have, if not supplied it will use the own `file` samplerate. """ file_exists = (os.path.isfile(new_filepath)) if(samplerate == None): samplerate = self.file.getSamplerate() path = path_splitter(new_filepath) if(path['file_name'] == None): raise TypeError("It doesn't contain a filename") try: os.mkdir(path['path']) #tries to create the directory except (OSError,FileNotFoundError) as e: pass if(path['extension']=='csv'): pd.DataFrame(self.file.getSamples()).to_csv(path['full_path'], header= None,index = None) else: import soundfile as sf if(self.mode == 'a' and file_exists): with sf.SoundFile(path['full_path'], mode = 'r+') as wfile:,sf.SEEK_END) wfile.write(self.file.getSamples()) else: sf.write(path['full_path'], self.file.getSamples(), samplerate,format=path['extension']) # writes to the new file return
[docs] def create_file_EMDDF(self, new_filepath, json_file, samplerate = None): """ Creates a soundfile with the EMD-DF format, recurs to the ``pyemddf`` package NOTE: Only works on wave and wave64 files. Args: samplerate(optional): Samplerate of the file. json_file: a JSON file with the metadata fields, you can get a template for it by executing ``pyemddf.create_template_file()``. """ if(not(os.path.exists(json_file))): raise FileNotFoundError("There is no file called" + json_file) self.create_file(new_filepath,samplerate) path = path_splitter(new_filepath) pyemddf.addChunks(path['full_path'],json_file)