main.py

import yaml

from get_standard_csv import *
from get_need_data import *
from post_preparation import *
from notion_property import *


def main():
    # 基本信息
    with open("config_private.yaml", "r", encoding="utf-8") as f:
        config = yaml.load(f, Loader=yaml.FullLoader)
    database_id = config["database_id"]
    token = config["token"]

    path_raw = "alipay_raw.csv"
    path_st = "alipay_standard_good.csv"

    # get_standard_csv(path_raw, path_st)

    df = get_need_data(path_st)
    for i in range(len(df)):
        content = df.iloc[i]["商品说明"]
        price = df.iloc[i]["金额"]
        category = df.iloc[i]["交易分类"]
        date = df.iloc[i]["交易时间"]
        counterparty = df.iloc[i]["交易对方"]
        remarks = df.iloc[i]["备注"]
        # remarks = ""
        transaction_number = df.iloc[i]["交易订单号"]
        merchant_tracking_number = df.iloc[i]["商家订单号"]
        payment_method = df.iloc[i]["收/付款方式"]

        # print(content, price, category, date, counterparty, remarks, transaction_number, merchant_tracking_number, payment_method)
        properties = notion_property(content, price, category, date, counterparty,
                                     remarks,
                                     transaction_number, 
                                     merchant_tracking_number, 
                                     payment_method
                                    )
        response = post_notion(properties, database_id, token)
        response_result(response)

if __name__ == "__main__":
    main()