Merge pull request #116 from degangliu/master

万象内容识别接口的增加

Author: dt3310321

qcloud_cos/cos_client.py

@@ -383,6 +383,78 @@ def get_object(self, Bucket, Key, **kwargs):
 
         return response
 
+    def get_object_sensitive_content_recognition(self, Bucket, Key, DetectType, **kwargs):
+        """文件内容识别接口 https://cloud.tencent.com/document/product/460/37318
+        
+        :param Bucket(string): 存储桶名称.
+        :param Key(string): COS路径.
+        :param DetectType(int): 内容识别标志,位计算 1:porn, 2:terrorist, 4:politics, 8:ads
+        :param kwargs(dict): 设置下载的headers.
+        :return(dict): 下载成功返回的结果,dict类型.
+
+        .. code-block:: python
+
+            config = CosConfig(Region=region, SecretId=secret_id, SecretKey=secret_key, Token=token)  # 获取配置对象
+            client = CosS3Client(config)
+            # 下载cos上的文件到本地
+            response = client.get_object_sensitive_content_recognition(
+                Bucket='bucket',
+                DetectType=CiDetectType.PORN | CiDetectType.POLITICS,
+                Key='test.png'
+            )
+            print response
+        """
+        headers = mapped(kwargs)
+        final_headers = {}
+        params = {}
+        for key in headers:
+            if key.startswith("response"):
+                params[key] = headers[key]
+            else:
+                final_headers[key] = headers[key]
+        headers = final_headers
+
+        if 'versionId' in headers:
+            params['versionId'] = headers['versionId']
+            del headers['versionId']
+        params['ci-process'] = 'sensitive-content-recognition'
+        detect_type = ''
+        if DetectType & CiDetectType.PORN > 0 :
+            detect_type += 'porn'
+        if DetectType & CiDetectType.TERRORIST > 0 :
+            if len(detect_type) > 0:
+                detect_type += ','
+            detect_type += 'terrorist'
+        if DetectType & CiDetectType.POLITICS > 0 :
+            if len(detect_type) > 0:
+                detect_type += ','
+            detect_type += 'politics'
+        if DetectType & CiDetectType.ADS > 0 :
+            if len(detect_type) > 0:
+                detect_type += ','
+            detect_type += 'ads'
+        
+        params['detect-type'] = detect_type
+        params = format_values(params)
+
+        url = self._conf.uri(bucket=Bucket, path=Key)
+        logger.info("get object sensitive content recognition, url=:{url} ,headers=:{headers}, params=:{params}".format(
+            url=url,
+            headers=headers,
+            params=params))
+        rt = self.send_request(
+                method='GET',
+                url=url,
+                bucket=Bucket,
+                stream=True,
+                auth=CosS3Auth(self._conf, Key, params=params),
+                params=params,
+                headers=headers)
+
+        data = xml_to_dict(rt.content)
+
+        return data
+
     def get_presigned_url(self, Bucket, Key, Method, Expired=300, Params={}, Headers={}):
         """生成预签名的url
 

qcloud_cos/cos_comm.py

@@ -405,3 +405,9 @@ def get_date(yy, mm, dd):
     date_str = datetime(yy, mm, dd).isoformat()
     final_date_str = date_str+'+08:00'
     return final_date_str
+
+class CiDetectType():
+    PORN = 1
+    TERRORIST = 2
+    POLITICS = 4
+    ADS = 8

qcloud_cos/version.py

@@ -1,2 +1,2 @@
 
-__version__ = '5.1.7.9'
+__version__ = '5.1.8.0'

setup.py

@@ -16,7 +16,7 @@ def long_description():
 
 setup(
     name='cos-python-sdk-v5',
-    version='1.7.9',
+    version='1.8.0',
     url='https://www.qcloud.com/',
     license='MIT',
     author='tiedu, lewzylu, channingliu',

ut/test.py

@@ -10,6 +10,7 @@
 from qcloud_cos import CosConfig
 from qcloud_cos import CosServiceError
 from qcloud_cos import get_date
+from qcloud_cos.cos_comm import CiDetectType
 
 SECRET_ID = os.environ["SECRET_ID"]
 SECRET_KEY = os.environ["SECRET_KEY"]
@@ -1151,6 +1152,16 @@ def test_select_object():
     for event in event_stream:
         print(event)
 
+def test_get_object_sensitive_content_recognition():
+    """测试ci文件内容识别的接口"""
+    print(CiDetectType)
+    response = client.get_object_sensitive_content_recognition(
+        Bucket=test_bucket,
+        Key=test_object,
+        DetectType=(CiDetectType.PORN | CiDetectType.TERRORIST | CiDetectType.POLITICS | CiDetectType.ADS)
+    )
+    print(response)
+    assert response
 
 if __name__ == "__main__":
     setUp()
@@ -1175,6 +1186,8 @@ def test_select_object():
     test_put_get_delete_bucket_inventory()
     test_put_get_traffic_limit()
     test_put_get_delete_bucket_domain()
-    """
     test_select_object()
+    """
+
+    test_get_object_sensitive_content_recognition()
     tearDown()