Spaces:
Running
Running
| # Copyright (c) Meta Platforms, Inc. and affiliates. | |
| # All rights reserved. | |
| # | |
| # This source code is licensed under the terms described in the LICENSE file in | |
| # the root directory of this source tree. | |
| # Excel parser logic | |
| import os | |
| from typing import Dict, Any | |
| from starfish.data_ingest.parsers.base_parser import BaseParser | |
| class ExcelParser(BaseParser): | |
| """Parser for Excel files""" | |
| def __init__(self): | |
| super().__init__() | |
| self.supported_extensions = [".xlsx", ".xls"] | |
| self.metadata = {} | |
| def parse(self, file_path: str) -> str: | |
| """Parse an Excel file into text | |
| Args: | |
| file_path: Path to the Excel file | |
| Returns: | |
| Extracted text from the Excel file | |
| """ | |
| try: | |
| import openpyxl | |
| except ImportError: | |
| raise ImportError("openpyxl is required for Excel parsing. Install it with: pip install openpyxl") | |
| # Load workbook and extract metadata | |
| wb = openpyxl.load_workbook(file_path) | |
| self.metadata = { | |
| "file_path": file_path, | |
| "sheets": wb.sheetnames, | |
| "creator": wb.properties.creator, | |
| "created": wb.properties.created, | |
| "modified": wb.properties.modified, | |
| "last_modified_by": wb.properties.lastModifiedBy, | |
| } | |
| # Extract text from all sheets | |
| all_text = [] | |
| for sheet_name in wb.sheetnames: | |
| sheet = wb[sheet_name] | |
| sheet_text = [] | |
| sheet_text.append(f"--- Sheet: {sheet_name} ---") | |
| for row in sheet.iter_rows(values_only=True): | |
| row_text = [str(cell) for cell in row if cell is not None] | |
| if row_text: | |
| sheet_text.append("\t".join(row_text)) | |
| all_text.append("\n".join(sheet_text)) | |
| return "\n\n".join(all_text) | |
| def get_metadata(self) -> Dict[str, Any]: | |
| """Get Excel file metadata | |
| Returns: | |
| Dictionary containing file metadata | |
| """ | |
| return self.metadata | |
| def is_supported(self, file_path: str) -> bool: | |
| """Check if the file is supported by this parser | |
| Args: | |
| file_path: Path to the file | |
| Returns: | |
| True if the file is supported, False otherwise | |
| """ | |
| return os.path.splitext(file_path)[1].lower() in self.supported_extensions | |