import loompy
import pandas as pd
import os


def loom_to_pandas_df(file_name="testfiles/test.loom"):
    """
    :param file_name: the name of the loom file (default is:
    "testfiles/test.loom")
    :return: pandas DataFrame
    """
    df = pd.DataFrame  # Empty DataFrame - to return in case of wrong formatting
    if file_name[-5:] != ".loom":
        print(f"{file_name} is not a valid file name. Only use '.loom' files!")

    with loompy.connect(file_name) as ds:
        try:
            df = pd.DataFrame(data=ds[:, :],
                              index=ds.ra.Gene,
                              columns=ds.ca.input_name)
        except AttributeError or ValueError or OSError:
            raise Exception(f"The loom file {file_name} does not exist or has "
                            f"no valid format!")

    return df


print(os.getcwd())
os.chdir("testfiles")
print(os.getcwd())
# print(os.listdir())
test_df = loom_to_pandas_df("AllelicExpressionPatterns-mouse-brain-SS2.loom")
test_df.info()
test_df.describe()
print(".........")
print(test_df)