上传文件时,可以添加校验参数强制校验key和hash

Author: ramwin

qiniu/exceptions.py

@@ -0,0 +1,11 @@
+#!/usr/bin/env python3
+# -*- coding: UTF-8 -*-
+# Xiang Wang @ 2019-09-17 15:25:42
+
+
+class QiniuException(Exception):
+    pass
+
+
+class UploadException(QiniuException):
+    pass

qiniu/services/storage/uploader.py

@@ -5,7 +5,9 @@
 import time
 
 from qiniu import config, Auth
-from qiniu.utils import urlsafe_base64_encode, crc32, file_crc32, _file_iter, rfc_from_timestamp
+from qiniu import config
+from qiniu.exceptions import UploadException
+from qiniu.utils import urlsafe_base64_encode, crc32, file_crc32, _file_iter, rfc_from_timestamp, etag
 from qiniu import http
 from .upload_progress_recorder import UploadProgressRecorder
 
@@ -47,6 +49,7 @@ def put_data(
 def put_file(up_token, key, file_path, params=None,
              mime_type='application/octet-stream', check_crc=False,
              progress_handler=None, upload_progress_recorder=None, keep_last_modified=False, hostscache_dir=None,
+             raise_exception=False,
              part_size=None, version=None, bucket_name=None):
     """上传文件到七牛
 
@@ -63,6 +66,7 @@ def put_file(up_token, key, file_path, params=None,
         version                   分片上传版本 目前支持v1/v2版本 默认v1
         part_size                 分片上传v2必传字段 默认大小为4MB 分片大小范围为1 MB - 1 GB
         bucket_name               分片上传v2字段 空间名称
+        raise_exception:          上传后自动校验key和hash, 如果不一致就报错
 
     Returns:
         一个dict变量，类似 {"hash": "<Hash string>", "key": "<Key string>"}
@@ -84,6 +88,9 @@ def put_file(up_token, key, file_path, params=None,
             ret, info = _form_put(up_token, key, input_stream, params, mime_type,
                                   crc, hostscache_dir, progress_handler, file_name,
                                   modify_time=modify_time, keep_last_modified=keep_last_modified)
+    if raise_exception is True:
+        if (ret["key"] != key) or (ret["hash"] != etag(file_path)):
+            raise UploadException("数据校验不正确")
     return ret, info