File size: 1,395 Bytes
4780a80 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
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() |