mirror of
https://github.com/yyhuni/xingrin.git
synced 2026-02-02 04:33:10 +08:00
55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
"""
|
|
工作空间相关的 Prefect Tasks
|
|
|
|
负责扫描工作空间的创建、验证和管理
|
|
"""
|
|
|
|
from pathlib import Path
|
|
from prefect import task
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@task(
|
|
name="create_scan_workspace",
|
|
description="创建并验证 Scan 工作空间目录",
|
|
retries=2,
|
|
retry_delay_seconds=5
|
|
)
|
|
def create_scan_workspace_task(scan_workspace_dir: str) -> Path:
|
|
"""
|
|
创建并验证 Scan 工作空间目录
|
|
|
|
Args:
|
|
scan_workspace_dir: Scan 工作空间目录路径
|
|
|
|
Returns:
|
|
Path: 创建的 Scan 工作空间路径对象
|
|
|
|
Raises:
|
|
OSError: 目录创建失败或不可写
|
|
"""
|
|
scan_workspace_path = Path(scan_workspace_dir)
|
|
|
|
# 创建目录
|
|
try:
|
|
scan_workspace_path.mkdir(parents=True, exist_ok=True)
|
|
logger.info("✓ Scan 工作空间已创建: %s", scan_workspace_path)
|
|
except OSError as e:
|
|
logger.error("创建 Scan 工作空间失败: %s - %s", scan_workspace_dir, e)
|
|
raise
|
|
|
|
# 验证目录是否可写
|
|
test_file = scan_workspace_path / ".test_write"
|
|
try:
|
|
test_file.touch()
|
|
test_file.unlink()
|
|
logger.info("✓ Scan 工作空间验证通过（可写）: %s", scan_workspace_path)
|
|
except OSError as e:
|
|
error_msg = f"Scan 工作空间不可写: {scan_workspace_path}"
|
|
logger.error(error_msg)
|
|
raise OSError(error_msg) from e
|
|
|
|
return scan_workspace_path
|