用了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 np
dir = 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') #可以导出csv
return 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"] = None
data.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') #可以导出csv
return 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 bank
bean = _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
效果如下: