Read pandas dataframe according to file extension

read_auto.py

import os
import pandas as pd

def read_auto(datapath):
    _, ext = os.path.splitext(datapath)

    if ext == ".csv":
        return pd.read_csv(datapath)
    elif ext == ".tsv":
        return pd.read_csv(datapath, sep="\t")
    elif ext in [".parquet", ".pq", ".parq"]:
        return pd.read_parquet(datapath)
    elif ext in [".xlsx", ".xls"]:
        return pd.read_excel(datapath)
    else:
        raise Exception("Unsupported file type. Should be one of csv, tsv, parquet, or xlsx.")