mirror of
https://github.com/pacnpal/Pac-cogs.git
synced 2025-12-20 02:41:06 -05:00
Fixed FFmpeg Management:
Updated extraction logic to handle BtbN's new archive structure with binaries in bin directory Added fallback to old structure for backward compatibility Improved binary verification and permissions handling Enhanced error handling during extraction and verification Fixed Exception Handling: Added missing FileCleanupError exception Updated file operations to use proper exceptions Improved error propagation and logging Better cleanup error handling Improved Resource Management: Better cleanup of failed downloads Proper handling of temporary files Enhanced queue management Improved file deletion with retries Enhanced Error Reporting: More detailed error messages Better logging of cleanup operations Proper error propagation through the system Improved user feedback The system now: Properly handles FFmpeg's new archive structure Better manages file operations and cleanup Provides more detailed error messages Cleans up resources properly Shows appropriate error messages for different failure types
This commit is contained in:
@@ -9,6 +9,8 @@ from datetime import datetime
|
||||
from pathlib import Path
|
||||
from typing import Optional
|
||||
|
||||
from .exceptions import FileCleanupError
|
||||
|
||||
logger = logging.getLogger("VideoArchiver")
|
||||
|
||||
def secure_delete_file(file_path: str, passes: int = 3, timeout: int = 30) -> bool:
|
||||
@@ -21,6 +23,9 @@ def secure_delete_file(file_path: str, passes: int = 3, timeout: int = 30) -> bo
|
||||
|
||||
Returns:
|
||||
bool: True if file was successfully deleted, False otherwise
|
||||
|
||||
Raises:
|
||||
FileCleanupError: If file deletion fails after all attempts
|
||||
"""
|
||||
if not os.path.exists(file_path):
|
||||
return True
|
||||
@@ -30,9 +35,9 @@ def secure_delete_file(file_path: str, passes: int = 3, timeout: int = 30) -> bo
|
||||
# Get file size before starting
|
||||
try:
|
||||
file_size = os.path.getsize(file_path)
|
||||
except OSError:
|
||||
except OSError as e:
|
||||
file_size = 0
|
||||
logger.warning(f"Could not get size of {file_path}, assuming 0")
|
||||
logger.warning(f"Could not get size of {file_path}: {e}")
|
||||
|
||||
# Ensure file is writable
|
||||
try:
|
||||
@@ -40,6 +45,7 @@ def secure_delete_file(file_path: str, passes: int = 3, timeout: int = 30) -> bo
|
||||
os.chmod(file_path, current_mode | stat.S_IWRITE)
|
||||
except OSError as e:
|
||||
logger.warning(f"Could not modify permissions of {file_path}: {e}")
|
||||
raise FileCleanupError(f"Permission error: {str(e)}")
|
||||
|
||||
# Overwrite file content
|
||||
if file_size > 0:
|
||||
@@ -78,11 +84,13 @@ def secure_delete_file(file_path: str, passes: int = 3, timeout: int = 30) -> bo
|
||||
while os.path.exists(file_path):
|
||||
if (datetime.now() - start_time).total_seconds() > timeout:
|
||||
logger.error(f"Timeout while trying to delete {file_path}")
|
||||
return False
|
||||
raise FileCleanupError(f"Deletion timeout for {file_path}")
|
||||
time.sleep(0.1)
|
||||
|
||||
return True
|
||||
|
||||
except FileCleanupError:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Error during secure deletion of {file_path}: {e}")
|
||||
# Last resort: try force delete
|
||||
@@ -92,6 +100,7 @@ def secure_delete_file(file_path: str, passes: int = 3, timeout: int = 30) -> bo
|
||||
Path(file_path).unlink(missing_ok=True)
|
||||
except Exception as e2:
|
||||
logger.error(f"Force delete failed for {file_path}: {e2}")
|
||||
raise FileCleanupError(f"Force delete failed: {str(e2)}")
|
||||
return not os.path.exists(file_path)
|
||||
|
||||
def cleanup_downloads(download_path: str) -> None:
|
||||
@@ -99,22 +108,26 @@ def cleanup_downloads(download_path: str) -> None:
|
||||
|
||||
Args:
|
||||
download_path: Path to the downloads directory to clean
|
||||
|
||||
Raises:
|
||||
FileCleanupError: If cleanup fails
|
||||
"""
|
||||
try:
|
||||
if not os.path.exists(download_path):
|
||||
return
|
||||
|
||||
errors = []
|
||||
# Delete all files in the directory
|
||||
for entry in os.scandir(download_path):
|
||||
try:
|
||||
path = entry.path
|
||||
if entry.is_file():
|
||||
if not secure_delete_file(path):
|
||||
logger.error(f"Failed to delete file: {path}")
|
||||
errors.append(f"Failed to delete file: {path}")
|
||||
elif entry.is_dir():
|
||||
shutil.rmtree(path, ignore_errors=True)
|
||||
except Exception as e:
|
||||
logger.error(f"Error processing {entry.path}: {e}")
|
||||
errors.append(f"Error processing {entry.path}: {str(e)}")
|
||||
continue
|
||||
|
||||
# Clean up empty subdirectories
|
||||
@@ -125,7 +138,13 @@ def cleanup_downloads(download_path: str) -> None:
|
||||
if not os.listdir(dir_path): # Check if directory is empty
|
||||
os.rmdir(dir_path)
|
||||
except Exception as e:
|
||||
logger.error(f"Error removing directory {name}: {e}")
|
||||
errors.append(f"Error removing directory {name}: {str(e)}")
|
||||
|
||||
if errors:
|
||||
raise FileCleanupError("\n".join(errors))
|
||||
|
||||
except FileCleanupError:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Error during cleanup of {download_path}: {e}")
|
||||
raise FileCleanupError(f"Cleanup failed: {str(e)}")
|
||||
|
||||
Reference in New Issue
Block a user