Spaces:
Running
Running
File size: 2,398 Bytes
5301c48 |
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 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
# 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
|