Source code for pdb2pqr.io

"""Generic functions used for input and output."""

from pathlib import Path
from typing import List, Tuple

from .factory import input_factory
from .reader_pqr import read_pqr  # noqa: F401
from .reader_qcd import read_qcd  # noqa: F401
from .dx import read_dx, write_cube  # noqa: F401


def read_input(inputfile_path: str) -> Tuple[List[str], List[str]]:
    """Read and parse input files using the appropriate factory class.

    :param inputfile_path: Path to the input file to read
    :type inputfile_path: str

    :return:  List of PDB and ERROR objects
    :rtype:  Tuple[List[str], List[str]]
    """
    file_path = Path(inputfile_path)
    reader = input_factory(file_path.suffix)
    return reader.read(file_path)