From 4080f8071e96d4dbcc5ae8af10b051e14fea30ac Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8F=AD=E6=89=AC?= <xingjun.wxj@alibaba-inc.com>
Date: Tue, 1 Nov 2022 12:57:04 +0800
Subject: [PATCH] temp

---
 modelscope/hub/api.py               | 11 +++++++++++
 modelscope/msdatasets/ms_dataset.py | 14 ++++++++++++++
 2 files changed, 25 insertions(+)

diff --git a/modelscope/hub/api.py b/modelscope/hub/api.py
index 7468e5e3..0262fc1d 100644
--- a/modelscope/hub/api.py
+++ b/modelscope/hub/api.py
@@ -646,6 +646,17 @@ class HubApi:
     def check_local_cookies(self, use_cookies) -> CookieJar:
         return self._check_cookie(use_cookies=use_cookies)
 
+    def count_uv_by_channel(self, dataset_name: str, namespace: str, channel: str):
+        # todo: 1. check args  2.
+
+        url = f'{self.endpoint}/api/v1/datasets/{namespace}/{dataset_name}/download/uv/{channel}'
+        cookies = ModelScopeConfig.get_cookies()
+        r = requests.post(url, cookies=cookies, headers=self.headers)
+        resp = r.json()
+        raise_on_error(resp)
+        print(resp)
+        return resp['Message']
+
 
 class ModelScopeConfig:
     path_credential = expanduser(DEFAULT_CREDENTIALS_PATH)
diff --git a/modelscope/msdatasets/ms_dataset.py b/modelscope/msdatasets/ms_dataset.py
index 0c537df7..a7d29990 100644
--- a/modelscope/msdatasets/ms_dataset.py
+++ b/modelscope/msdatasets/ms_dataset.py
@@ -727,3 +727,17 @@ class MsDataset:
         resp_msg = _delete_manager.delete(object_name=object_name)
         logger.info(f'Object {object_name} successfully removed!')
         return resp_msg
+
+
+if __name__ == '__main__':
+    from modelscope.hub.api import HubApi
+    api = HubApi()
+    # api.login('c252d64a-ce7b-4c0c-b583-7bedf628c7da')  # online
+    # api.login('aa14716f-e2de-4f26-bf49-254d81eb8ac6')   # test
+
+    channel = 'local'  # dsw
+    dataset_name = 'small_coco_for_test'
+    namespace = 'wangxingjun778test'
+    resp = api.count_uv_by_channel(
+        dataset_name=dataset_name, namespace=namespace, channel=channel)
+    print(resp)