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

其实独立出一个包更好，之后发布到PyPI，可以直接pip安装。

有时间我会做一下。
Merge pull request #1 from jasonlvhit/patch-1

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