Compare commits

...

2 Commits

Author SHA1 Message Date
zzs
b835ed6cb3 feat: replace with control 2025-03-27 15:16:04 +08:00
zzs
ca9bb2771c chore: do clear 2025-03-27 10:13:08 +08:00

View File

@ -1,5 +1,21 @@
import json
import time
def Replace(bookmark_name_value_map):
"""
替换所有书签
Args:
bookmark_name_value_map: 书签名称和新值的映射
"""
doc = XSCRIPTCONTEXT.getDocument()
bookmarks = doc.getBookmarks()
for bookmark_name, new_value in bookmark_name_value_map.items():
if bookmarks.hasByName(bookmark_name):
bookmark = bookmarks.getByName(bookmark_name)
anchor = bookmark.getAnchor()
anchor.setString(new_value)
def ReplaceOne(bookmark_obj_collection, bookmark_name, new_value):
"""
@ -16,30 +32,6 @@ def ReplaceOne(bookmark_obj_collection, bookmark_name, new_value):
anchor.setString(new_value)
def ReplaceAll(bookmark_obj_collection, bookmark_name_value_map):
"""
替换所有书签
Args:
bookmark_obj_collection: 书签对象集合
bookmark_name_value_map: 书签名称和新值的映射
"""
for bookmark_name, new_value in bookmark_name_value_map.items():
ReplaceOne(bookmark_obj_collection, bookmark_name, new_value)
def Replace(bookmark_name_value_map):
"""
替换所有书签
Args:
bookmark_name_value_map: 书签名称和新值的映射
"""
doc = XSCRIPTCONTEXT.getDocument()
bookmarks = doc.getBookmarks()
ReplaceAll(bookmarks, bookmark_name_value_map)
def ReplaceWithJSON(json_str):
"""
替换所有书签
@ -51,21 +43,6 @@ def ReplaceWithJSON(json_str):
Replace(bookmark_name_value_map)
def DebugCallReplace():
"""
调试调用
"""
Replace({"bb1": "P1", "bb2": "P2", "bb3": "P3", "bb4": "P4"})
def DebugCallReplaceList():
"""
调试调用替换list
"""
Replace({"list": "1\r2\r3\r4"})
def QueryAll():
"""
查询所有书签表格中的书签获取对应表格index
@ -85,27 +62,14 @@ def QueryAll():
doc = XSCRIPTCONTEXT.getDocument()
bookmarks = doc.getBookmarks()
bookmark_names = bookmarks.getElementNames()
filtered_bookmarks = [bk_name for bk_name in bookmark_names if not bk_name.startswith('_') and ':' not in bk_name]
filtered_bookmarks = [
bk_name
for bk_name in bookmark_names
if not bk_name.startswith("_") and ":" not in bk_name
]
result = {"text": [], "table": []}
# tables = doc.getTextTables()
# bookmark_in_table_position = QueryBookmarkPositionInTable(tables, bookmarks)
existing_table_indices = {}
for bk_name in filtered_bookmarks:
# if bk_name in bookmark_in_table_position:
# table_index = bookmark_in_table_position[bk_name]["tableIndex"]
# if table_index in existing_table_indices:
# index = existing_table_indices[table_index]
# result["table"][index]["bookmark"].append(bk_name)
# else:
# new_entry = {"tableIndex": table_index, "bookmark": [bk_name]}
# result["table"].append(new_entry)
# existing_table_indices[table_index] = (
# len(result["table"]) - 1
# ) # 更新索引记录
# else:
result["text"].append(bk_name)
return result
@ -116,59 +80,59 @@ def QueryAllWithJSON():
def QueryBookmarkPositionInTable(tables, bookmarks):
"""
查询书签在表格中的位置
{'bk1': {'tableIndex': 0, 'row': 0, 'col': 0}, 'bk2': {'tableIndex': 0, 'row': 0, 'col': 2}}
查询书签在表格中的位置
参数:
tables: 表格集合 (XIndexAccess)
bookmarks: 书签集合 (XIndexAccess)
返回:
一个字典格式为 {'书签名': {'tableIndex': 表格索引, 'row': 行号}}
"""
result = {}
bookmark_names = bookmarks.getElementNames()
# 遍历所有书签
for b in range(bookmarks.getCount()):
bookmark = bookmarks.getByIndex(b)
bookmark_name = bookmark.getName()
x_range = bookmark.getAnchor() # 获取书签的锚点
x_text = x_range.getText() # 获取书签所在的文本对象
bookmark_position = {}
# 遍历所有表格
for t in range(tables.getCount()):
table = tables.getByIndex(t)
for table_index in range(tables.getCount()):
table = tables.getByIndex(table_index)
# 遍历表格中的每一行
rows = table.getRows().getCount()
for i in range(rows):
# 动态确定当前行的最大列数
max_cols = 0
while True:
try:
cell = table.getCellByPosition(max_cols, i)
max_cols += 1
except Exception:
break
col_count = table.getColumns().getCount()
row_count = table.getRows().getCount()
for j in range(max_cols):
cell = table.getCellByPosition(j, i)
cell_text = cell.getText()
for row in range(row_count):
for col in range(col_count):
cell = table.getCellByPosition(col, row)
cell_text_obj = cell.getText()
for bk_name in bookmark_names:
bookmark_obj = bookmarks.getByName(bk_name)
bookmark_text_obj = bookmark_obj.getAnchor().getText()
if cell_text_obj == bookmark_text_obj:
bookmark_position[bk_name] = {
"tableIndex": table_index,
"col": col,
"row": row,
if cell_text == x_text:
result[bookmark_name] = {
'tableIndex': t,
'row': i,
'col': j
}
return bookmark_position
def CreateRow(table, row_count):
"""
创建表格行
"""
table.getRows().insertByIndex(table.getRows().getCount(), row_count)
def FindTableIndex(table_bookmarks, location_bookmark_name):
"""
找到表格索引
"""
handle_table_index = -1
for table_info in table_bookmarks:
if location_bookmark_name in table_info["bookmark"]:
handle_table_index = table_info["tableIndex"]
break
else:
continue
break
else:
continue
break
if handle_table_index == -1:
raise ValueError(f"未找到书签 {location_bookmark_name} 对应的表格")
return handle_table_index
return result
def InsertRowWithJSON(json_str):
"""
@ -194,7 +158,9 @@ def InsertRow(location_bookmark_name, data, start_row_index=-1):
if location_bookmark_name not in bookmark_in_table_position:
raise ValueError(f"未找到书签 {location_bookmark_name} 对应的表格")
handle_table_index = bookmark_in_table_position[location_bookmark_name]["tableIndex"]
handle_table_index = bookmark_in_table_position[location_bookmark_name][
"tableIndex"
]
try:
handle_table = tables.getByIndex(handle_table_index)
@ -207,33 +173,29 @@ def InsertRow(location_bookmark_name, data, start_row_index=-1):
if not data or len(data) == 0:
return
if any(len(row) != col_count for row in data):
raise ValueError(f"数据列数不匹配,表格有 {col_count}")
try:
row_count = handle_table.getRows().getCount()
rows_to_insert = len(data)
if rows_to_insert > 0:
# 确定插入位置
if start_row_index != -1 and 0 <= start_row_index < row_count:
insert_pos = start_row_index + 1 # 在指定行下方插入
else:
insert_pos = row_count # 插入到表格末尾
# 批量插入所有新行
handle_table.getRows().insertByIndex(insert_pos, rows_to_insert)
# 填充数据到新插入的行
for data_row_idx, row_data in enumerate(data):
target_row = insert_pos + data_row_idx # 计算目标行索引
for col_idx, cell_value in enumerate(row_data):
cell = handle_table.getCellByPosition(col_idx, target_row)
cell.setString(str(cell_value))
except Exception as e:
raise RuntimeError(f"插入行时发生错误: {str(e)}")
raise RuntimeError(f"err: {str(e)}")
def BatchInsertRowWithContentControl(data_array):
"""
@ -245,13 +207,15 @@ def BatchInsertRowWithContentControl(data_array):
bookmark_in_table_position = QueryBookmarkPositionInTable(tables, bookmarks)
for arr_obj in data_array:
location_bookmark_name = arr_obj.get('location_bookmark_name')
data = arr_obj.get('data')
start_row_index = arr_obj.get('start_row_index', -1) # 默认值为 -1
location_bookmark_name = arr_obj.get("location_bookmark_name")
data = arr_obj.get("data")
start_row_index = arr_obj.get("start_row_index", -1) # 默认值为 -1
if location_bookmark_name not in bookmark_in_table_position:
raise ValueError(f"未找到书签 {location_bookmark_name} 对应的表格")
handle_table_index = bookmark_in_table_position[location_bookmark_name]["tableIndex"]
handle_table_index = bookmark_in_table_position[location_bookmark_name][
"tableIndex"
]
try:
handle_table = tables.getByIndex(handle_table_index)
@ -264,45 +228,51 @@ def BatchInsertRowWithContentControl(data_array):
if not data or len(data) == 0:
return
if any(len(row) != col_count for row in data):
raise ValueError(f"数据列数不匹配,表格有 {col_count}")
# if any(len(row) != col_count for row in data):
# raise ValueError(f"数据列数不匹配,表格有 {col_count} 列")
try:
row_count = handle_table.getRows().getCount()
rows_to_insert = len(data)
if rows_to_insert > 0:
# 确定插入位置
if start_row_index != -1 and 0 <= start_row_index < row_count:
insert_pos = start_row_index + 1 # 在指定行下方插入
else:
insert_pos = row_count # 插入到表格末尾
# 批量插入所有新行
handle_table.getRows().insertByIndex(insert_pos, rows_to_insert)
# 填充数据到新插入的行
for data_row_idx, row_data in enumerate(data):
target_row = insert_pos + data_row_idx # 计算目标行索引
for col_idx, cell_value in enumerate(row_data):
cell = handle_table.getCellByPosition(col_idx, target_row)
# cell.setString(str(cell_value))
# 创建文本控件并赋值 -----------------------------
cell_text = cell.getText()
cell_text.setString("") # 清空原有内容
text_control = doc.createInstance("com.sun.star.text.TextField.Input")
text_control.setPropertyValue("Content", str(cell_value))
# === 清空单元格 ===
cell_text.setString("")
# === 创建内容控件 ===
content_control = doc.createInstance(
"com.sun.star.text.ContentControl"
)
# === 插入控件到单元格 ===
cursor = cell_text.createTextCursor()
cell_text.insertTextContent(cursor, text_control, False)
cell_text.insertTextContent(cursor, content_control, False)
# === 设置控件内容 ===
cc_text = content_control.getText()
cc_cursor = cc_text.createTextCursor()
cc_cursor.setString(str(cell_value))
except Exception as e:
raise RuntimeError(f"插入行时发生错误: {str(e)}")
def BatchInsertRow(data_array):
"""
批量插入行
@ -314,13 +284,15 @@ def BatchInsertRow(data_array):
bookmark_in_table_position = QueryBookmarkPositionInTable(tables, bookmarks)
for arr_obj in data_array:
location_bookmark_name = arr_obj.get('location_bookmark_name')
data = arr_obj.get('data')
start_row_index = arr_obj.get('start_row_index', -1) # 默认值为 -1
location_bookmark_name = arr_obj.get("location_bookmark_name")
data = arr_obj.get("data")
start_row_index = arr_obj.get("start_row_index", -1) # 默认值为 -1
if location_bookmark_name not in bookmark_in_table_position:
raise ValueError(f"未找到书签 {location_bookmark_name} 对应的表格")
handle_table_index = bookmark_in_table_position[location_bookmark_name]["tableIndex"]
handle_table_index = bookmark_in_table_position[location_bookmark_name][
"tableIndex"
]
try:
handle_table = tables.getByIndex(handle_table_index)
@ -339,50 +311,39 @@ def BatchInsertRow(data_array):
try:
row_count = handle_table.getRows().getCount()
rows_to_insert = len(data)
if rows_to_insert > 0:
# 确定插入位置
if start_row_index != -1 and 0 <= start_row_index < row_count:
insert_pos = start_row_index + 1 # 在指定行下方插入
else:
insert_pos = row_count # 插入到表格末尾
# 批量插入所有新行
handle_table.getRows().insertByIndex(insert_pos, rows_to_insert)
# 填充数据到新插入的行
for data_row_idx, row_data in enumerate(data):
target_row = insert_pos + data_row_idx # 计算目标行索引
for col_idx, cell_value in enumerate(row_data):
cell = handle_table.getCellByPosition(col_idx, target_row)
cell.setString(str(cell_value))
except Exception as e:
raise RuntimeError(f"插入行时发生错误: {str(e)}")
def InsertRowWithJSONDefineCol(json_str):
"""
插入行
"""
def DebugCallInsertRow():
"""
调试调用
"""
InsertRow("t1", [["1", "2"], ["4", "5"]], 0)
def DeleteRowWithJSON(json_str):
"""
删除行
"""
data = json.loads(json_str)
DeleteRow(data["location_bookmark_name"], data["start_row_index"], data["delete_row_count"])
DeleteRow(
data["location_bookmark_name"],
data["start_row_index"],
data["delete_row_count"],
)
def DeleteRow(location_bookmark_name, start_row_index, delete_row_count=-1):
"""
@ -393,10 +354,6 @@ def DeleteRow(location_bookmark_name, start_row_index, delete_row_count=-1):
start_row_index: 起始行位置(删除行包含本行)
delete_row_count: 删除的行数
"""
# bookmark_info = QueryAll()
# table_bookmarks = bookmark_info["table"]
# handle_table_index = FindTableIndex(table_bookmarks, location_bookmark_name)
doc = XSCRIPTCONTEXT.getDocument()
tables = doc.getTextTables()
bookmarks = doc.getBookmarks()
@ -404,8 +361,9 @@ def DeleteRow(location_bookmark_name, start_row_index, delete_row_count=-1):
if location_bookmark_name not in bookmark_in_table_position:
raise ValueError(f"未找到书签 {location_bookmark_name} 对应的表格")
handle_table_index = bookmark_in_table_position[location_bookmark_name]["tableIndex"]
handle_table_index = bookmark_in_table_position[location_bookmark_name][
"tableIndex"
]
handle_table = tables.getByIndex(handle_table_index)
row_count = handle_table.getRows().getCount()
@ -419,14 +377,7 @@ def DeleteRow(location_bookmark_name, start_row_index, delete_row_count=-1):
try:
handle_table.getRows().removeByIndex(start_row_index, delete_row_count)
except Exception as e:
raise RuntimeError(f"删除行时发生错误: {str(e)}")
def DebugCallDeleteRow():
"""
调试调用
"""
DeleteRow("h1", 1, 2)
raise RuntimeError(f"err: {str(e)}")
def LocateBookmark(bookmark_name):
@ -438,7 +389,7 @@ def LocateBookmark(bookmark_name):
doc = XSCRIPTCONTEXT.getDocument()
if False == doc.getBookmarks().hasByName(bookmark_name):
return
return
bookmark = doc.getBookmarks().getByName(bookmark_name)
@ -467,11 +418,6 @@ def LocateBookmark(bookmark_name):
except Exception as e:
raise RuntimeError(f"定位书签失败: {str(e)}") from e
def DebugCallLocateBookmarkLo1():
LocateBookmark("lo1")
def DebugCallLocateBookmarkLo2():
LocateBookmark("lo2")
def InsertBookmark(bookmark_name):
"""
@ -481,21 +427,21 @@ def InsertBookmark(bookmark_name):
"""
doc = XSCRIPTCONTEXT.getDocument()
controller = doc.getCurrentController()
# 获取当前选区
selection = controller.getSelection()
# 检查是否已存在同名书签,如果存在则先删除
bookmarks = doc.getBookmarks()
if bookmarks.hasByName(bookmark_name):
bookmarks.getByName(bookmark_name).dispose()
try:
# 创建书签
if selection.getCount() > 0:
# 尝试获取选区的第一个元素
range_to_bookmark = selection.getByIndex(0)
# 创建书签并附加到选区
bookmark = doc.createInstance("com.sun.star.text.Bookmark")
bookmark.attach(range_to_bookmark)
@ -504,21 +450,13 @@ def InsertBookmark(bookmark_name):
# 没有选中内容,创建点书签
cursor = controller.getViewCursor()
text_cursor = cursor.getText().createTextCursorByRange(cursor)
# 创建书签并附加到光标位置
bookmark = doc.createInstance("com.sun.star.text.Bookmark")
bookmark.attach(text_cursor)
bookmark.setName(bookmark_name)
except Exception as e:
raise RuntimeError(f"插入书签失败: {str(e)}") from e
def DebugCallInsertBookmark():
"""
调试调用
"""
InsertBookmark("bookmark_" + str(time.time()))
def UpdateBookmark(bookmark_name, new_value):
@ -541,11 +479,6 @@ def UpdateBookmark(bookmark_name, new_value):
old_bookmark = bookmarks.getByName(bookmark_name)
old_bookmark.setName(new_value)
def DebugCallUpdateBookmark():
"""
调试调用
"""
UpdateBookmark("ole_bookmark", "new_bookmark")
def DeleteBookmark(bookmark_name):
"""
@ -566,12 +499,6 @@ def DeleteBookmark(bookmark_name):
except Exception as e:
pass
def DebugCallDeleteBookmark():
"""
调试调用
"""
DeleteBookmark("remove_me")
def CreateTable(location_bookmark_name, data):
"""
@ -581,6 +508,7 @@ def CreateTable(location_bookmark_name, data):
data: 二维数组用于填充表格数据
"""
def ReplaceTextAndInsertTableRow(json_str):
"""
替换文本和表格
@ -589,23 +517,27 @@ def ReplaceTextAndInsertTableRow(json_str):
Replace(data["text"])
BatchInsertRow(data["table"])
def ReplaceTextAndInsertTableRowWithContentControl(json_str):
data = json.loads(json_str)
# BatchInsertRowWithContentControl(data["table"])
ReplaceBookmarksWithControls(data["text"])
BatchInsertRowWithContentControl(data["table"])
def returnWithJSON(data):
return json.dumps(data, ensure_ascii=False)
def SaveDocument():
# 获取当前文档对象
model = XSCRIPTCONTEXT.getDocument()
try:
# 检查文档是否支持保存XStorable 接口)
from com.sun.star.frame import XStorable
xstorable = model.uno_getAdapter(XStorable)
# 调用保存方法(覆盖原文件)
xstorable.store()
return True
@ -613,47 +545,59 @@ def SaveDocument():
print("err:", e)
return False
def DebugCallReplaceBookmarksWithControls():
ReplaceBookmarksWithControls({"合同编号_A": "11111", "合同名称_A": "22222"})
def ReplaceBookmarksWithControls(bookmark_name_value_map={}):
# 获取文档对象
doc = XSCRIPTCONTEXT.getDocument()
bookmarks = doc.getBookmarks()
for name in reversed([bm.getName() for bm in bookmarks]):
# 过滤包含下划线前缀的书签
bookmark_names = [name for name in bookmarks.getElementNames() if not name.startswith("_")]
# 遍历所有需要处理的书签
for name in bookmark_names:
try:
if name.startswith('_'):
# 检查书签是否存在
if not bookmarks.hasByName(name):
continue
# 获取书签对象及其锚点
bookmark = bookmarks.getByName(name)
anchor = bookmark.getAnchor()
# 获取书签内容
cursor = anchor.getText().createTextCursorByRange(anchor)
original_text = cursor.getString()
# 创建文本控件并设置内容
text_control = doc.createInstance("com.sun.star.text.TextField.Input")
if (bookmark_name_value_map.get(name)):
text_control.setPropertyValue("Content", bookmark_name_value_map.get(name))
else
text_control.setPropertyValue("Content", original_text)
# 替换原书签内容为控件
cursor.setString("")
anchor.getText().insertTextContent(cursor, text_control, False)
# 重新插入原书签到控件后
new_cursor = cursor.getText().createTextCursorByRange(cursor)
new_cursor.goRight(1, False)
doc.getBookmarks().addNew(name, new_cursor)
# 获取书签的原始内容
original_text = anchor.getString()
replace_value = bookmark_name_value_map.get(name, original_text)
# === 第1步创建内容控件 ===
content_control = doc.createInstance("com.sun.star.text.ContentControl")
# 获取书签所在文本对象并清空原内容
text = anchor.getText()
anchor.setString("")
# 插入内容控件
text.insertTextContent(anchor, content_control, True)
# 设置控件内容
cc_cursor = content_control.getText().createTextCursor()
cc_cursor.setString(replace_value)
# === 第2步清理旧书签 ===
bookmark.dispose()
# === 第3步创建新书签 ===
cc_anchor = content_control.getText().getAnchor()
new_bookmark = doc.createInstance("com.sun.star.text.Bookmark")
new_bookmark.setName(name)
new_bookmark.attach(cc_anchor)
except Exception as e:
print(f"error{str(e)}")
print(f"处理书签 '{name}' 时出错: {e}")
def NextEditableZone(current_index):
"""
根据索引定位可编辑区域支持循环
根据索引定位名称包含'permission'的书签支持循环
Args:
current_index (int/str): 0 开始的索引支持字符串或整数类型
"""
@ -661,41 +605,56 @@ def NextEditableZone(current_index):
try:
current_index = int(current_index)
except (ValueError, TypeError):
print(f"err {current_index} cant convert to int")
print(f"错误:无法将 {current_index} 转换为整数")
return
doc = XSCRIPTCONTEXT.getDocument()
controller = doc.getCurrentController()
view_cursor = controller.getViewCursor()
# 收集所有可编辑控件的锚点
text_fields = doc.getTextFields().createEnumeration()
editable_anchors = []
while text_fields.hasMoreElements():
field = text_fields.nextElement()
if field.supportsService("com.sun.star.text.TextField.Input"):
try:
anchor = field.getAnchor()
editable_anchors.append(anchor)
except Exception:
continue
bookmarks = doc.getBookmarks()
bookmark_names = bookmarks.getElementNames()
# 无控件时直接返回
# 过滤包含permission的书签不区分大小写
permission_bookmarks = [name for name in bookmark_names if not name.startswith("_")]
editable_anchors = []
for bm_name in permission_bookmarks:
try:
bookmark = bookmarks.getByName(bm_name)
anchor = bookmark.getAnchor()
editable_anchors.append(anchor)
except Exception as e:
print(f"err '{bm_name}' e: {e}")
continue
# 无匹配书签时直接返回
if not editable_anchors:
print("文档中无可编辑控件")
return
# 计算有效索引(循环逻辑)
total = len(editable_anchors)
effective_index = current_index % total # 此处不再报错
effective_index = current_index % total # 自动处理越界
# 定位到目标控件
# 定位到目标书签
target_anchor = editable_anchors[effective_index]
view_cursor.gotoRange(target_anchor, False)
controller.select(target_anchor) # 选中控件(可选)
try:
# 跳转到书签起始位置
view_cursor.gotoRange(target_anchor, False)
# 选中整个书签范围
controller.select(target_anchor)
# 窗口可见性处理
try:
window = controller.getFrame().getContainerWindow()
window.setVisible(True)
window.toFront()
except Exception:
pass
except Exception as e:
raise RuntimeError(f"err: {str(e)}") from e
g_exportedScripts = (
@ -708,18 +667,8 @@ g_exportedScripts = (
DeleteRow,
DeleteRowWithJSON,
ReplaceTextAndInsertTableRow,
DebugCallReplace,
DebugCallReplaceList,
DebugCallInsertRow,
DebugCallDeleteRow,
DebugCallLocateBookmarkLo1,
DebugCallLocateBookmarkLo2,
DebugCallInsertBookmark,
DebugCallUpdateBookmark,
DebugCallDeleteBookmark,
ReplaceBookmarksWithControls,
ReplaceTextAndInsertTableRowWithContentControl,
DebugCallReplaceBookmarksWithControls,
SaveDocument,
NextEditableZone
NextEditableZone,
)