LocalFileSource

classView source ↗

class LocalFileSource(
    path: str | List[str],
    format: str | dict | FileFormat = None,
    initial_last_modified: str | datetime = None,
)

Bases: SourcePlugin

Categories: source

Class for managing the configuration of local-file-based data inputs.

Attributes: format (FileFormat): The format of the file. If not provided, it will be inferred from the file extension of the data. path (str | List[str]): The path where the files can be found. It can be a single path or a list of paths. initial_last_modified (str | None): If not None, only the files modified after this date and time will be considered.

Initializes the LocalFileSource with the given path, and optionally a format and a date and time after which the files were modified.

Parameters

parameter

pathstr | List[str]

The path where the files can be found. It can be a single path or a list of paths.

parameter

formatstr | dict | FileFormat, optional

The format of the file. If not provided, it will be inferred from the file extension of the data. Can be either a string with the format, a FileFormat object or a dictionary with the format as the 'type' key and any additional format-specific information. Currently supported formats are 'csv', 'parquet', 'ndjson', 'jsonl' and 'log'.

parameter

initial_last_modifiedstr | datetime, optional

If provided, only the files modified after this date and time will be considered. The date and time can be provided as a string in ISO 8601 format or as a datetime object. If no timezone is provided, UTC will be assumed.

Raises

InputConfigurationError FormatConfigurationError

Properties

property

formatFileFormat

The format of the file or files. If not provided, it will be inferred from the file extension in the path.

property

initial_last_modifiedstr

The date and time after which the files were modified.

property

initial_valuesdict

Return a dictionary with the initial values to be stored after execution of the plugin. They will be accessible in the next execution of the plugin. The dictionary must have the parameter names as keys and the initial values as values, all the type string.

Returns: dict: A dictionary with the initial values of the parameters of the plugin.

property

pathstr | List[str]

The path or paths to the files to load.

class

View source ↗

SupportedFormats

class SupportedFormats(*args, **kwds)

Bases: Enum

Enum for the supported formats for the LocalFileSource.

attribute

csv

attribute

ndjson

attribute

log

attribute

parquet

Methods

method

View source ↗

chunk

def chunk(working_dir: str) -> list[str | None | list[str | None]]

Trigger the import of the data. This must be implemented in any class that inherits from this class unless directly implementing streaming. The method will receive a folder where it must store the data as parquet files, and return a list of the paths of the files created. This files will then be loaded and mapped to the dataset function in positional order, so if you want file.parquet to be the first argument of the dataset function, you must return it first. If you want a parameter to receive multiple files, return a list of the paths. For example, you would give the following return to provide a first argument with a single file and a second argument with two files: return ["file1.parquet", ["file2.parquet", "file3.parquet"]]

Parameters:

parameter

working_dirstr

The folder where the files must be stored

Returns:

Union[str, Tuple[str, ...], List[str]]: The path of the file(s) created, in the order they must be mapped to the dataset function