更新Python脚本，基本完成查询功能