From 842bc2568b972ef7a202729feaecf3523b922ca8 Mon Sep 17 00:00:00 2001
From: foyou <yimi.0822@qq.com>
Date: Mon, 21 Aug 2023 12:18:09 +0800
Subject: [PATCH] add /v2/file/scan

---
 src/aligo/core/Config.py |  1 +
 src/aligo/core/File.py   | 17 +++++++++++++++--
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/aligo/core/Config.py b/src/aligo/core/Config.py
index c651ea11..3302f96d 100644
--- a/src/aligo/core/Config.py
+++ b/src/aligo/core/Config.py
@@ -46,6 +46,7 @@
 V2_FILE_MOVE = '/v2/file/move'
 V2_FILE_COPY = '/v2/file/copy'
 V2_FILE_WALK = '/v2/file/walk'
+V2_FILE_SCAN = '/v2/file/scan'
 V2_FILE_SEARCH = '/v2/file/search'
 V2_FILE_CREATE = '/v2/file/create'
 V2_FILE_COMPLETE = '/v2/file/complete'
diff --git a/src/aligo/core/File.py b/src/aligo/core/File.py
index 5520c69e..2688c43b 100644
--- a/src/aligo/core/File.py
+++ b/src/aligo/core/File.py
@@ -2,11 +2,11 @@
 from typing import Iterator
 
 from aligo.core import BaseAligo
-from aligo.core.Config import ADRIVE_V3_FILE_LIST, V2_FILE_WALK
+from aligo.core.Config import ADRIVE_V3_FILE_LIST, V2_FILE_WALK, V2_FILE_SCAN
 from aligo.request import BatchGetFileRequest, GetFileListRequest, GetFileRequest, BatchRequest, BatchSubRequest
 from aligo.response import BatchSubResponse, GetFileListResponse
 from aligo.types import BaseFile
-from aligo.types.Enum import BaseFileType
+from aligo.types.Enum import BaseFileType, BaseFileCategory
 
 
 class File(BaseAligo):
@@ -52,3 +52,16 @@ def _core_walk_file(
             'url_expire_sec': url_expire_sec,
             'limit': limit,
         }, GetFileListResponse)
+
+    def _core_scan_file(
+            self,
+            drive_id: str = None,
+            category: BaseFileCategory = None,
+            limit: int = 1000,
+    ) -> Iterator[BaseFile]:
+        """..."""
+        yield from self._list_file(V2_FILE_SCAN, {
+            'drive_id': drive_id or self.default_drive_id,
+            'category': category,
+            'limit': limit,
+        }, GetFileListResponse)