python代码封装