用了beancount来记账后,发现每月的账单手动记是一件极其麻烦的事情。
然后再github搜索一通后,有double-entry-generator(https://github.com/deb-sig/double-entry-generator)能转换支付宝/微信的账单,但是没有自己用的建行和招商。
所有自己写了一个py:
- 建行能导出的账单是excel格式的直接用;
- 招商导出的是pdf,幸好还能转成txt,然后自己整理成csv
- 后续可能会加上深农商和农行,也可能不会,这两张卡用的不多。
import os
import re
import pandas as pd
import numpy as npdir = os.path.dirname(os.path.abspath(__file__))#input_file = "xxxxxx.csv"
input_file = input("输入账单路径:")
bank = input("输入账单归属银行[CCB、CMB、SRCB、ABC]:") #暂时只有CCB和CMB
output_file = dir + "\\" + bank + ".bean"
config_file = dir + "\\config\\" + bank + ".conf" #当前config文件夹下的CCB.conf和CMB.conf。#CCB账单excel的整理
def CCB_data(_file_):data = pd.read_excel(_file_)data.columns = ["No","text0","CCY","Yuan","time","cost","bal","text1","text2"]data = data.iloc[2:]#data["time"] = data["time"].str[0:4] + "-" + data["time"].str[4:6] + "-" + data["time"].str[6:8]data["time"] = pd.to_datetime(data["time"]).astype(str)data.loc[:,"account"] = data["text0"] + ":" + data["text2"]data["CCY"] = data["CCY"].replace("人民币元","CNY")data = data[["time","cost","CCY","account","text0","text1","text2"]]#pd.DataFrame(data).to_csv(output_file+".csv",encoding='utf-8-sig') #可以导出csvreturn data#CMB账单csv的整理
def CMB_data(_file_):data = pd.read_csv(_file_)data.columns = ["time","CCY","cost","bal","text0","text2"]data["time"] = pd.to_datetime(data["time"]).astype(str)data["cost"] = data["cost"].astype(str)data.loc[:,"text1"] = Nonedata.loc[:,"account"] = data["text0"] + ":" + data["text2"]data = data[["time","cost","CCY","account","text0","text1","text2"]]#pd.DataFrame(data).to_csv(output_file+".csv",encoding='utf-8-sig') #可以导出csvreturn data#关键字查找替换
def replace(_data_,_config_): conf = pd.read_csv(_config_)for conf in conf.itertuples():_data_ = _data_.str.replace(".*"+conf[1]+".*",conf[2],regex = True)return _data_#输出beancount格式的字段
def to_bean(_data_):global bankbean = _data_["time"] + ' * \"' +_data_["text0"] + '\"\n\t'if _data_["text1"].notnull().any():bean = bean + 'notes: \"' + _data_["text1"] + '\"\n\t' bean = bean + 'id: \"' + _data_["text2"] + '\"\n\t'bean = bean + "Assets:Bank:" + bank + " " + _data_["cost"] + " " + _data_["CCY"] + "\n\t"bean = bean + _data_["account"] + "\n"return bean#整理账单格式
if bank == "CCB":DATA = CCB_data(input_file)
if bank == "CMB":DATA = CMB_data(input_file)#按conf文件查找替换关键字
DATA["account"]= replace(DATA["account"],config_file)
#print(DATA)#整理为beancount格式的字段
bean = to_bean(DATA).to_frame()
#print(bean)#输出文件CCB.bean或者CMB.beantext = ""
for bean in bean.itertuples():text = text + str(bean[1]) + "\n"file = open(output_file,'w')
file.write(text)
file.close()
config文件,格式:【关键字,替换后】,例如:
Text,Replace_text
饿了么,Expenses:Food:Others
效果如下: