mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-07-10 01:18:29 +00:00
7160747607
Let backup creation and restore cleanup use unlimited pattern scans so archives for agents with more than 50,000 files are complete. Keep UI preview and dry-run paths bounded for responsiveness while making the dry-run truncated flag safe for optional limits. Add regression coverage for unlimited scans, backup creation, clean-before-restore, and restoring an archive entry past the old 50,000-file boundary.
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
from helpers.api import ApiHandler, Request, Response
|
|
from helpers.backup import BackupService
|
|
|
|
|
|
class BackupTest(ApiHandler):
|
|
@classmethod
|
|
def requires_auth(cls) -> bool:
|
|
return True
|
|
|
|
@classmethod
|
|
def requires_loopback(cls) -> bool:
|
|
return False
|
|
|
|
async def process(self, input: dict, request: Request) -> dict | Response:
|
|
try:
|
|
# Get input parameters
|
|
include_patterns = input.get("include_patterns", [])
|
|
exclude_patterns = input.get("exclude_patterns", [])
|
|
include_hidden = input.get("include_hidden", True)
|
|
max_files = input.get("max_files", 1000)
|
|
|
|
# Support legacy string patterns format for backward compatibility
|
|
patterns_string = input.get("patterns", "")
|
|
if patterns_string and not include_patterns:
|
|
# Parse patterns string into arrays
|
|
lines = [line.strip() for line in patterns_string.split('\n') if line.strip() and not line.strip().startswith('#')]
|
|
for line in lines:
|
|
if line.startswith('!'):
|
|
exclude_patterns.append(line[1:])
|
|
else:
|
|
include_patterns.append(line)
|
|
|
|
if not include_patterns:
|
|
return {
|
|
"success": True,
|
|
"files": [],
|
|
"total_count": 0,
|
|
"truncated": False
|
|
}
|
|
|
|
# Create metadata object for testing
|
|
metadata = {
|
|
"include_patterns": include_patterns,
|
|
"exclude_patterns": exclude_patterns,
|
|
"include_hidden": include_hidden
|
|
}
|
|
|
|
backup_service = BackupService()
|
|
matched_files = await backup_service.test_patterns(metadata, max_files=max_files)
|
|
truncated = max_files is not None and len(matched_files) >= max_files
|
|
|
|
return {
|
|
"success": True,
|
|
"files": matched_files,
|
|
"total_count": len(matched_files),
|
|
"truncated": truncated
|
|
}
|
|
|
|
except Exception as e:
|
|
return {
|
|
"success": False,
|
|
"error": str(e)
|
|
}
|