from UniversalParser import WordParser

# wtl = WordParser.parse_tables('test.docx')
# wtl.modify_cell_by_coordinate(1, (2, 1), '测试中，位置')
# wtl.save_as()

word = WordParser.parse_tables('111.docx')
word.batch_modify_cells_by_coordinate(
    table_loc = 1,
    coordinate_values = {
        (2, 2): 'Hello World'
        , (5, 2): '测试合并单元格赋值'
        , (3, 2): '测试2010'
    }
)
word.refresh_tables()
print(word.tables[0])
word.manager.save_as_xml()
word.save_as('output.docx')



# import re
# from pprint import pprint

# wtl = Word.parse_tables('test.docx')

# replace_cells = {
#     'lbxx': '蜡笔小新'
#     , 'hmbb': '海绵宝宝'
#     , 'gtxy': '工藤新一'
#     , 'D4': '123'
#     , 'E5': '456'
# }

# wtl.save_from_modify(
#     new_word_path = 'temp.docx'
#     # , patt = re.compile(r'[+]{2}(.+?)[+]{2}')
#     , **replace_cells
# )



# wtl.refresh_tables()
# pprint(wtl.tables) # 获取当前 Word 的所有表格



## 保存测试
word.to_html(title='测试', out_path='测试.html', encoding='utf-8')

# wtl.manager.save_as_xml() # 提取出表格所在的XML


## 修改测试
# from UniversalParser._tools import *

# content = read_content_from_zip('test.docx').replace('\r\n', '').replace('\n', '')
# print(content)
# save_modify_content_to_zip('test.docx', content.replace('ok', 'HelloWorld'))