feat：整理手动中止功能

app/modules/filemanager/storages/smb.py

@@ -8,7 +8,7 @@ from smbclient import ClientConfig, register_session, reset_connection_cache
 from smbprotocol.exceptions import SMBException, SMBResponseException, SMBAuthenticationError
 
 from app import schemas
-from app.core.config import settings
+from app.core.config import settings, global_vars
 from app.log import logger
 from app.modules.filemanager import StorageBase
 from app.modules.filemanager.storages import transfer_process
@@ -438,12 +438,14 @@ class SMB(StorageBase, metaclass=WeakSingleton):
                 with open(local_path, "wb") as dst_file:
                     downloaded_size = 0
                     while True:
+                        if global_vars.is_transfer_stopped(fileitem.path):
+                            logger.info(f"【SMB】{fileitem.path} 下载已取消！")
+                            return None
                         chunk = src_file.read(self.chunk_size)
                         if not chunk:
                             break
                         dst_file.write(chunk)
                         downloaded_size += len(chunk)
-
                         # 更新进度
                         if file_size:
                             progress = (downloaded_size * 100) / file_size
@@ -485,12 +487,14 @@ class SMB(StorageBase, metaclass=WeakSingleton):
                 with smbclient.open_file(smb_path, mode="wb") as dst_file:
                     uploaded_size = 0
                     while True:
+                        if global_vars.is_transfer_stopped(path.as_posix()):
+                            logger.info(f"【SMB】{path} 上传已取消！")
+                            return None
                         chunk = src_file.read(self.chunk_size)
                         if not chunk:
                             break
                         dst_file.write(chunk)
                         uploaded_size += len(chunk)
-
                         # 更新进度
                         if file_size:
                             progress = (uploaded_size * 100) / file_size