Spaces:
Sleeping
Sleeping
| import os | |
| import shutil | |
| import sys | |
| import logging | |
| from datetime import datetime | |
| # Set up logging | |
| logging.basicConfig( | |
| filename=f"conversion_log_{datetime.now().strftime('%Y%m%d_%H%M%S')}.log", | |
| level=logging.INFO, | |
| format="%(asctime)s - %(levelname)s - %(message)s" | |
| ) | |
| # For .docx conversion | |
| try: | |
| from docx2pdf import convert as docx_convert | |
| except ImportError: | |
| logging.error("docx2pdf not installed. Run 'pip install docx2pdf'.") | |
| sys.exit(1) | |
| # For .pptx conversion | |
| try: | |
| import win32com.client | |
| except ImportError: | |
| logging.error("pywin32 not installed. Run 'pip install pywin32'.") | |
| sys.exit(1) | |
| try: | |
| import comtypes.client | |
| except ImportError: | |
| logging.warning("comtypes not installed. Run 'pip install comtypes' for fallback.") | |
| def check_irm_status(filepath): | |
| """Check if IRM is enabled on a PPTX file.""" | |
| try: | |
| ppt_app = win32com.client.Dispatch("PowerPoint.Application") | |
| pres = ppt_app.Presentations.Open(os.path.abspath(filepath), ReadOnly=True, WithWindow=False) | |
| irm_enabled = pres.Permission.Enabled | |
| pres.Close() | |
| ppt_app.Quit() | |
| return irm_enabled | |
| except Exception as e: | |
| logging.error(f"Failed to check IRM for {filepath}: {e}") | |
| return None | |
| def convert_pptx_print(filepath, output_pdf): | |
| """Attempt PPTX to PDF conversion using print-to-PDF.""" | |
| try: | |
| ppt_app = win32com.client.Dispatch("PowerPoint.Application") | |
| pres = ppt_app.Presentations.Open(os.path.abspath(filepath), ReadOnly=True, WithWindow=False) | |
| pres.PrintOut(PrintToFile=output_pdf) | |
| pres.Close() | |
| ppt_app.Quit() | |
| logging.info(f"Converted {filepath} to {output_pdf} using print-to-PDF.") | |
| return True | |
| except Exception as e: | |
| logging.error(f"Print-to-PDF failed for {filepath}: {e}") | |
| return False | |
| def convert_pptx_win32(filepath, output_pdf): | |
| """Attempt PPTX to PDF conversion using win32com SaveAs.""" | |
| try: | |
| ppt_app = win32com.client.Dispatch("PowerPoint.Application") | |
| pres = ppt_app.Presentations.Open(os.path.abspath(filepath), ReadOnly=True, WithWindow=False) | |
| pres.SaveAs(output_pdf, 32) # 32 = ppSaveAsPDF | |
| pres.Close() | |
| ppt_app.Quit() | |
| logging.info(f"Converted {filepath} to {output_pdf} using win32com SaveAs.") | |
| return True | |
| except Exception as e: | |
| logging.error(f"win32com SaveAs failed for {filepath}: {e}") | |
| return False | |
| def convert_pptx_comtypes(filepath, output_pdf): | |
| """Attempt PPTX to PDF conversion using comtypes SaveAs.""" | |
| try: | |
| ppt_app = comtypes.client.CreateObject("PowerPoint.Application") | |
| pres = ppt_app.Presentations.Open(os.path.abspath(filepath), ReadOnly=True, WithWindow=False) | |
| pres.SaveAs(output_pdf, 32) # 32 = ppSaveAsPDF | |
| pres.Close() | |
| ppt_app.Quit() | |
| logging.info(f"Converted {filepath} to {output_pdf} using comtypes SaveAs.") | |
| return True | |
| except Exception as e: | |
| logging.error(f"comtypes SaveAs failed for {filepath}: {e}") | |
| return False | |
| def main(input_dir, output_dir): | |
| if not os.path.isdir(input_dir): | |
| logging.error(f"{input_dir} is not a directory.") | |
| sys.exit(1) | |
| os.makedirs(output_dir, exist_ok=True) | |
| for root, _, files in os.walk(input_dir): | |
| rel_dir = os.path.relpath(root, input_dir) | |
| target_subdir = os.path.join(output_dir, rel_dir) | |
| os.makedirs(target_subdir, exist_ok=True) | |
| for file in files: | |
| filepath = os.path.join(root, file) | |
| base, ext = os.path.splitext(file) | |
| output_pdf = os.path.join(target_subdir, base + '.pdf') | |
| try: | |
| if ext.lower() == '.docx': | |
| docx_convert(filepath, output_pdf) | |
| logging.info(f"Converted {filepath} to {output_pdf}") | |
| elif ext.lower() == '.pptx': | |
| # Check IRM status | |
| irm_status = check_irm_status(filepath) | |
| if irm_status is not None: | |
| logging.info(f"IRM enabled for {filepath}: {irm_status}") | |
| # Try print-to-PDF first (bypasses edit restrictions) | |
| if not convert_pptx_print(filepath, output_pdf): | |
| # Fallback to SaveAs methods | |
| if not convert_pptx_win32(filepath, output_pdf): | |
| if 'comtypes' in sys.modules and not convert_pptx_comtypes(filepath, output_pdf): | |
| logging.warning(f"Failed to convert {filepath}. Likely IRM or strict restrictions.") | |
| elif not 'comtypes' in sys.modules: | |
| logging.warning(f"Failed to convert {filepath}. Install comtypes for additional fallback.") | |
| elif ext.lower() == '.pdf': | |
| shutil.copy(filepath, output_pdf) | |
| logging.info(f"Copied {filepath} to {output_pdf}") | |
| except Exception as e: | |
| logging.error(f"Error processing {filepath}: {e}") | |
| if __name__ == "__main__": | |
| main("OrigFolder", "ConvertedFiles") # Use OrigFolder directly |