| import os | |
| import shutil | |
| from datetime import datetime, timedelta | |
| from .config import settings | |
| from .logger import logger | |
| class FileManager: | |
| def __init__(self): | |
| self.storage_path = settings.PDF_STORAGE_PATH | |
| os.makedirs(self.storage_path, exist_ok=True) | |
| def save_pdf(self, pdf_content: bytes, filename: str) -> str: | |
| """Save PDF content to storage.""" | |
| try: | |
| filepath = os.path.join(self.storage_path, filename) | |
| with open(filepath, 'wb') as f: | |
| f.write(pdf_content) | |
| logger.info(f"PDF saved successfully: {filename}") | |
| return filepath | |
| except Exception as e: | |
| logger.error(f"Error saving PDF {filename}: {str(e)}") | |
| raise | |
| def cleanup_old_files(self): | |
| """Remove PDFs older than MAX_PDF_AGE_DAYS.""" | |
| try: | |
| cutoff_date = datetime.now() - timedelta(days=settings.MAX_PDF_AGE_DAYS) | |
| for filename in os.listdir(self.storage_path): | |
| filepath = os.path.join(self.storage_path, filename) | |
| if os.path.getctime(filepath) < cutoff_date.timestamp(): | |
| os.remove(filepath) | |
| logger.info(f"Removed old PDF: {filename}") | |
| except Exception as e: | |
| logger.error(f"Error cleaning up old files: {str(e)}") | |
| raise | |
| file_manager = FileManager() |