Source code for dtale.cli.loaders.csv_loader
import pandas as pd
from dtale.cli.clickutils import get_loader_options
'''
IMPORTANT!!! These global variables are required for building any customized CLI loader.
When build_loaders runs startup it will search for any modules containing the global variable LOADER_KEY.
'''
LOADER_KEY = 'csv'
LOADER_PROPS = ['path', 'parse_dates']
# IMPORTANT!!! This function is required for building any customized CLI loader.
[docs]def find_loader(kwargs):
"""
CSV implementation of data loader which will return a function if any of the
`click` options based on LOADER_KEY & LOADER_PROPS have been used, otherwise return None
:param kwargs: Optional keyword arguments to be passed from `click`
:return: data loader function for CSV implementation
"""
csv_opts = get_loader_options(LOADER_KEY, kwargs)
if len([f for f in csv_opts.values() if f]):
def _csv_loader():
csv_arg_parsers = { # TODO: add additional arg parsers
'parse_dates': lambda v: v.split(',') if v else None
}
kwargs = {k: csv_arg_parsers.get(k, lambda v: v)(v) for k, v in csv_opts.items() if k != 'path'}
return pd.read_csv(csv_opts['path'], **kwargs)
return _csv_loader
return None