tapshex/csvreader.py

"""Tapshex wrapper for DCTAP CSV reader."""

from dctap.csvreader import csvreader
from tapshex.classes import Shape, StatementTemplate


def dctap_csvreader(
    csvfile_str=None,
    config_dict=None,
    shape_class=Shape,
    state_class=StatementTemplate,
):
    """From CSV string, return DCTAP shapes dict."""
    # Open file object not passable because it can only be read once.
    tapshapes = csvreader(
        csvfile_str=csvfile_str,
        config_dict=config_dict,
        shape_class=shape_class,
        state_class=state_class,
    )
    return tapshapes


def tapshex_postprocess(tapshapes):
    """Postprocess DCTAP shapes dict."""
    for shape in tapshapes.values():
        shape.postprocess()
    return tapshapes