import pandas as pd

finance_root = "C:\\Users\\lanyuanxiaoyao\\Documents\\Tushare"


def load_balance_sheet():
    return pd.read_csv(f"{finance_root}\\财务报表\\资产负债表1990-2024.csv")


def load_income():
    return pd.read_csv(f"{finance_root}\\财务报表\\利润表1990-2024.csv")


def load_cashflow():
    return pd.read_csv(f"{finance_root}\\财务报表\\现金流量表1990-2024.csv")


def load_finance():
    finance_df = load_balance_sheet()
    finance_df = pd.merge(finance_df, load_income(), on=["ts_code", "end_date"])
    finance_df = pd.merge(finance_df, load_cashflow(), on=["ts_code", "end_date"])
    return finance_df