增加了类包装，函数中有小改动

其实独立出一个包更好，之后发布到PyPI，可以直接pip安装。

有时间我会做一下。