import os
import pandas as pd
import ezdxf
from pathlib import Path
 
def load_translation_map(excel_path):
    df = pd.read_excel(excel_path)
    return dict(zip(df.iloc[:, 0], df.iloc[:, 1]))
 
def translate_text_entity(msp, entity, translation_map):
    original_text = entity.dxf.text
    translated_text = translation_map.get(original_text)
    if translated_text:
        msp.add_text(translated_text, dxfattribs={
            'insert': (entity.dxf.insert[0], entity.dxf.insert[1] - 0.75),
            'height': 1,  # 設(shè)置字體大小
            'color': 1    # 設(shè)置顏色為紅色
        })
 
def translate_dwg(dwg_path, translation_map):
    doc = ezdxf.readfile(dwg_path)
    msp = doc.modelspace()
 
    # 翻譯模型空間中的文本
    for entity in msp.query('TEXT MTEXT'):
        translate_text_entity(msp, entity, translation_map)
 
    # 翻譯塊中的文本
    for block in doc.blocks:
        for entity in block.query('TEXT MTEXT'):
            translate_text_entity(msp, entity, translation_map)
 
    return doc
 
def process_directory(directory, translation_map, output_folder):
    for root, dirs, files in os.walk(directory):
        # 跳過(guò)translated_drawings文件夾
        dirs[:] = [d for d in dirs if d != output_folder.name]
        for name in files:
            if name.endswith('.dxf'):
                file_path = Path(root) / name
                relative_path = file_path.relative_to(directory)
                output_path = output_folder / relative_path
                output_path.parent.mkdir(parents=True, exist_ok=True)
                print(f"Processing file: {file_path}")
                doc = translate_dwg(file_path, translation_map)
                doc.saveas(str(output_path))
                print(f"File saved: {output_path}")
 
def main():
    translation_map = load_translation_map('3.xlsx')
    source_folder = Path('.')  # Current directory
    output_folder = Path('translated_drawings')
    output_folder.mkdir(exist_ok=True)
    process_directory(source_folder, translation_map, output_folder)
 
if __name__ == "__main__":
    main()

import os
import pandas as pd
import ezdxf
from pathlib import Path
 
def extract_text_from_dxf(file_path):
    try:
        doc = ezdxf.readfile(file_path)
        msp = doc.modelspace()
        texts = []
 
        # 提取模型空間中的文本
        for entity in msp.query('TEXT MTEXT'):
            texts.append(entity.dxf.text)
 
        # 提取塊中的文本
        for block in doc.blocks:
            for entity in block.query('TEXT MTEXT'):
                texts.append(entity.dxf.text)
 
        return texts
    except IOError:
        print(f"Could not read file: {file_path}")
        return []
    except ezdxf.DXFStructureError:
        print(f"Invalid or corrupted DXF file: {file_path}")
        return []
 
def process_directory(directory):
    all_texts = []
    for root, dirs, files in os.walk(directory):
        for name in files:
            if name.endswith('.dxf'):
                file_path = Path(root) / name
                texts = extract_text_from_dxf(file_path)
                for text in texts:
                    all_texts.append([file_path, text])
 
    return all_texts
 
def main():
    directory = Path('.')  # 當(dāng)前目錄
    texts = process_directory(directory)
 
    # 將提取的文本保存到Excel文件中
    df = pd.DataFrame(texts, columns=['File Path', 'Text'])
    df.to_excel('extracted_texts.xlsx', index=False)
    print("Texts have been extracted and saved to 'extracted_texts.xlsx'.")
 
if __name__ == "__main__":
    main()