diff --git a/pywxdump/api/api.py b/pywxdump/api/api.py index d20beab..3fa1886 100644 --- a/pywxdump/api/api.py +++ b/pywxdump/api/api.py @@ -251,21 +251,22 @@ def get_img(): @api.route('/api/video/', methods=["GET", 'POST']) def get_video(videoPath): - # savePath = request.args.get("savePath") - # savePath = request.json.get("savePath", savePath) - savePath = "video/" + videoPath # 这个是从url中获取的 + savePath = os.path.join('video',videoPath) # 这个是从url中获取的 wx_path = read_session(g.sf, "wx_path") video_path = os.path.join(wx_path, videoPath) - if not savePath: + if os.path.exists(savePath): + return send_file(savePath) + if not videoPath: return ReJson(1002) - # 判断savePath路径的文件夹是否存在 savePath = os.path.join(g.tmp_path, savePath) if not os.path.exists(os.path.dirname(savePath)): os.makedirs(os.path.dirname(savePath)) - - - shutil.copy(video_path, savePath) + try: + shutil.copy(video_path, savePath) + except PermissionError: + if not os.path.exists(savePath): + return ReJson(5002) return send_file(savePath) @api.route('/api/audio/', methods=["GET", 'POST'])