Source code for cis_interface.drivers.PandasFileInputDriver

from cis_interface import serialize
from cis_interface.drivers.FileInputDriver import FileInputDriver


[docs]class PandasFileInputDriver(FileInputDriver): r"""Class to handle input from a Pandas csv file. Args: name (str): Name of the input queue to send messages to. args (str or dict): Path to the file that messages should be read from. delimiter (str, optional): String that should be used to separate columns. Defaults to '\t'. **kwargs: Additional keyword arguments are passed to parent class. """ def __init__(self, name, args, **kwargs): file_keys = ['delimiter'] icomm_kws = kwargs.get('icomm_kws', {}) icomm_kws.setdefault('comm', 'PandasFileComm') icomm_kws.setdefault('recv_converter', serialize.pandas2numpy) for k in file_keys: if k in kwargs: icomm_kws[k] = kwargs.pop(k) kwargs['icomm_kws'] = icomm_kws super(PandasFileInputDriver, self).__init__(name, args, **kwargs) self.debug('(%s)', args)
[docs] def update_serializer(self, msg): r"""Update the serializer for the output comm based on input.""" pass