python case

case.py

def linkbase_type(filepath):
    """Get the type of linkbase from the filename."""

    cases = (
        (filepath.endswith("_cal.xml"), "calculation"),
        (filepath.endswith('_def.xml'), "definition"),
        (filepath.endswith('_lab.xml'), "label"),
        (filepath.endswith('_pre.xml'), "presentation"),
        (filepath.endswith('.xsd'), "reference"),
        (filepath.endswith('.xml'), "filing"),
        (True, "unknown_linkbase")
    )

    return next(value for case, value in cases if case)