import os.path

import pandas as pd

# finance_root = "/Users/lanyuanxiaoyao/SynologyDrive/data/Tushare"
finance_root = "C:\\Users\\lanyuanxiaoyao\\Documents\\Tushare"


def load_balance_sheet():
    return pd.read_csv(os.path.join(finance_root, "财务报表", "资产负债表1990-2024.csv"))


def load_income():
    return pd.read_csv(os.path.join(finance_root, "财务报表", "利润表1990-2024.csv"))


def load_cashflow():
    return pd.read_csv(os.path.join(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