Source code for neuxus.nodes.store

import os

import pandas as pd

from neuxus.node import Node


class ToCsv(Node):
    """Convert a signal to a CSV file

    Args:
      - file (str): CSV file to write
      - sep (str): Separator between rows, default is ';'
      - decimal (str): Character recognized as decimal separator, default is ','

    Example:
        ToCsv(port4, 'log.csv')

    """

    def __init__(self, input_port, file, sep=';', decimal=','):
        Node.__init__(self, input_port, None)
        filename, file_extension = os.path.splitext(file)
        self._file = filename + '.csv'
        self._sep = sep
        self._decimal = decimal
        self._first_iter = True
        chan = self.input.channels
        pd.DataFrame([] * len(chan), [], chan).to_csv(self._file, mode='w', sep=self._sep, decimal=self._decimal)

        Node.log_instance(self, {'file': self._file})

    def update(self):
        for chunk in self.input:
            chunk.to_csv(self._file, mode='a+', header=False, sep=self._sep, decimal=self._decimal)