refactor: clean code

[Leizhenpeng]

• 引入工厂模式，提高可读性
• 优化代码结构和变量名称

[jingfelix]

回头不用 test.ipynb 做测试了，有点干扰 diff 信息。