|
|
package com.langtech.plm.tqsx;
|
|
|
|
|
|
import java.io.File;
|
|
|
import java.io.FileInputStream;
|
|
|
import java.io.FileNotFoundException;
|
|
|
import java.io.FileOutputStream;
|
|
|
import java.io.IOException;
|
|
|
import java.io.InputStream;
|
|
|
import java.io.OutputStream;
|
|
|
import java.nio.file.Files;
|
|
|
import java.nio.file.Paths;
|
|
|
import java.util.HashMap;
|
|
|
import java.util.List;
|
|
|
import java.util.Map;
|
|
|
|
|
|
import org.apache.poi.hssf.usermodel.HSSFCell;
|
|
|
import org.apache.poi.hssf.usermodel.HSSFName;
|
|
|
import org.apache.poi.hssf.usermodel.HSSFRow;
|
|
|
import org.apache.poi.hssf.usermodel.HSSFSheet;
|
|
|
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
|
|
|
import org.apache.poi.ss.util.CellReference;
|
|
|
import org.apache.poi.xssf.usermodel.XSSFCell;
|
|
|
import org.apache.poi.xssf.usermodel.XSSFName;
|
|
|
import org.apache.poi.xssf.usermodel.XSSFRow;
|
|
|
import org.apache.poi.xssf.usermodel.XSSFSheet;
|
|
|
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
|
|
|
|
|
|
import com.aspose.words.Bookmark;
|
|
|
import com.aspose.words.Document;
|
|
|
import com.aspose.words.Field;
|
|
|
import com.aspose.words.SaveFormat;
|
|
|
import com.teamcenter.rac.aif.AIFDesktop;
|
|
|
import com.teamcenter.rac.aif.AbstractAIFApplication;
|
|
|
import com.teamcenter.rac.aif.kernel.AbstractAIFSession;
|
|
|
import com.teamcenter.rac.aif.kernel.InterfaceAIFComponent;
|
|
|
import com.teamcenter.rac.kernel.TCComponent;
|
|
|
import com.teamcenter.rac.kernel.TCComponentDataset;
|
|
|
import com.teamcenter.rac.kernel.TCComponentItem;
|
|
|
import com.teamcenter.rac.kernel.TCComponentItemRevision;
|
|
|
import com.teamcenter.rac.kernel.TCComponentTcFile;
|
|
|
import com.teamcenter.rac.kernel.TCException;
|
|
|
import com.teamcenter.rac.kernel.TCProperty;
|
|
|
import com.teamcenter.rac.util.FileUtility;
|
|
|
|
|
|
|
|
|
|
|
|
public class PropertyToWordOrExcelOperation extends KOperation {
|
|
|
|
|
|
public PropertyToWordOrExcelOperation(AbstractAIFApplication app, String opName) {
|
|
|
super(app, opName, "ÕýÔÚÖ´ÐС£¡£¡£");
|
|
|
}
|
|
|
|
|
|
final String preName = "Connor_LY6_WordExcel_Report_ItemType";
|
|
|
|
|
|
@Override
|
|
|
public boolean init() throws Exception {
|
|
|
new Thread().sleep(1000);
|
|
|
InterfaceAIFComponent targetComponent = app.getTargetComponent();
|
|
|
TCComponentItemRevision revision = (TCComponentItemRevision) targetComponent;
|
|
|
|
|
|
// »ñÈ¡Ê×Ñ¡ÏîÀïµÄÖµ
|
|
|
String[] prefVals = KUtil.getPrefVals(session, preName);
|
|
|
if(prefVals == null || prefVals.length == 0) {
|
|
|
KUtil.info(AIFDesktop.getActiveDesktop(), preName + "¸ÃÊ×Ñ¡ÏîδÅäÖã¬ÇëÁªÏµ¹ÜÀíÔ±£¡");
|
|
|
return false;
|
|
|
}
|
|
|
String relatioin = prefVals[0];
|
|
|
|
|
|
// »ñÈ¡°æ±¾¶ÔÏóϵÄÊý¾Ý¼¯¶ÔÏó
|
|
|
TCComponent[] relatedComponents = revision.getRelatedComponents(relatioin);
|
|
|
if(relatedComponents == null || relatedComponents.length == 0) {
|
|
|
KUtil.info(AIFDesktop.getActiveDesktop(), "Ñ¡ÖжÔÏóµÄ " + relatioin + " ¹ØϵÏÂûÓÐÊý¾Ý¼¯");
|
|
|
return false;
|
|
|
}
|
|
|
//´¦ÀíÊ×Ñ¡Ïî
|
|
|
HashMap<String, String> nameMap = new HashMap<String, String>();//Êý¾Ý¼¯ÀïÃæµÄÃû³ÆºÍ¶ÔÏó(item,rev,revMaster)¼¯ºÏ
|
|
|
if(prefVals.length >= 2) {
|
|
|
for (int i = 1; i < prefVals.length; i++) {
|
|
|
//Item.object_name=object_name
|
|
|
String[] split = prefVals[i].split("=");
|
|
|
nameMap.put(split[1],split[0]);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// ±éÀúÊý¾Ý¼¯
|
|
|
for (TCComponent tcComponent : relatedComponents) {
|
|
|
if (tcComponent instanceof TCComponentDataset) {
|
|
|
TCComponentDataset dataset = (TCComponentDataset) tcComponent;
|
|
|
// ¹¤¾ß°üÖлñȡϵͳÁÙʱ±äÁ¿µÄ·¾¶
|
|
|
File datasetFile = KUtil.getDatasetFile(dataset);
|
|
|
String type = tcComponent.getType();
|
|
|
String datasetType = "";
|
|
|
String newPath = "";
|
|
|
if("MSExcel".equals(type) || "MSExcelX".equals(type) || "MSWord".equals(type) || "MSWordX".equals(type)) {
|
|
|
switch (type) {
|
|
|
case "MSExcel":
|
|
|
datasetType = "excel";
|
|
|
newPath = dealExcel(datasetFile,revision,nameMap);
|
|
|
break;
|
|
|
case "MSExcelX":
|
|
|
datasetType = "excel";
|
|
|
newPath = dealExcelX(datasetFile,revision,nameMap);
|
|
|
break;
|
|
|
case "MSWord":
|
|
|
datasetType = "word";
|
|
|
newPath = dealWord2(datasetFile,revision,nameMap,"doc");
|
|
|
break;
|
|
|
case "MSWordX":
|
|
|
datasetType = "word";
|
|
|
newPath = dealWord2(datasetFile,revision,nameMap,"docx");
|
|
|
break;
|
|
|
}
|
|
|
//¸øÊý¾Ý¼¯ÉèÖÃеÄÃüÃûµÄÒýÓÃ
|
|
|
dataset.setFiles(new String[] {newPath}, new String[] {datasetType});
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
KUtil.info(AIFDesktop.getActiveDesktop(), "ÌáÈ¡¶ÔÏó°æ±¾ÊôÐÔµ½Word ExcelÖ´ÐÐÍê³É");
|
|
|
return true;
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public void execute() throws Exception {
|
|
|
|
|
|
}
|
|
|
|
|
|
private String dealExcel(File datasetFile, TCComponentItemRevision revision, HashMap<String, String> nameMap) throws IOException, TCException {
|
|
|
FileInputStream is = new FileInputStream(datasetFile);
|
|
|
HSSFWorkbook workbook = new HSSFWorkbook(is);
|
|
|
List<HSSFName> allNames = workbook.getAllNames();
|
|
|
for (HSSFName hssfName : allNames) {
|
|
|
String name = hssfName.getNameName();
|
|
|
//Ãû³Æ¹ÜÀíÆ÷´æÔÚÊ×Ñ¡ÏîµÄÃû³Æ
|
|
|
if (nameMap.containsKey(name)) {
|
|
|
CellReference cellReference = new CellReference(hssfName.getRefersToFormula());
|
|
|
HSSFSheet sheet = workbook.getSheet(cellReference.getSheetName());
|
|
|
HSSFRow row = sheet.getRow(cellReference.getRow());
|
|
|
HSSFCell cell = row.getCell(cellReference.getCol());
|
|
|
String preConfigValue = getPreConfigValue(revision,nameMap,nameMap.get(name));
|
|
|
System.out.println("ÐÐΪ£º" + cellReference.getRow() + "====" + "ÁÐΪ£º" + cellReference.getCol() + "===="
|
|
|
+ "ֵΪ£º " + preConfigValue);
|
|
|
if(cell == null) {
|
|
|
cell = row.createCell(cellReference.getCol());
|
|
|
}
|
|
|
cell.setCellValue(preConfigValue);
|
|
|
|
|
|
}
|
|
|
}
|
|
|
is.close();
|
|
|
FileOutputStream os = new FileOutputStream(datasetFile);
|
|
|
workbook.write(os);
|
|
|
workbook.close();
|
|
|
return datasetFile.getAbsolutePath();
|
|
|
}
|
|
|
|
|
|
private String dealExcelX(File datasetFile, TCComponentItemRevision revision, HashMap<String, String> nameMap) throws IOException, TCException {
|
|
|
FileInputStream is = new FileInputStream(datasetFile);
|
|
|
XSSFWorkbook workbook = new XSSFWorkbook(is);
|
|
|
List<XSSFName> allNames = workbook.getAllNames();
|
|
|
for (XSSFName xssfName : allNames) {
|
|
|
String name = xssfName.getNameName();
|
|
|
//Ãû³Æ¹ÜÀíÆ÷´æÔÚÊ×Ñ¡ÏîµÄÃû³Æ
|
|
|
if (nameMap.containsKey(name)) {
|
|
|
CellReference cellReference = new CellReference(xssfName.getRefersToFormula());
|
|
|
XSSFSheet sheet = workbook.getSheet(cellReference.getSheetName());
|
|
|
XSSFRow row = sheet.getRow(cellReference.getRow());
|
|
|
XSSFCell cell = row.getCell(cellReference.getCol());
|
|
|
String preConfigValue = getPreConfigValue(revision,nameMap,nameMap.get(name));
|
|
|
System.out.println("ÐÐΪ£º" + cellReference.getRow() + "====" + "ÁÐΪ£º" + cellReference.getCol() + "===="
|
|
|
+ "ֵΪ£º " + preConfigValue);
|
|
|
if(cell == null) {
|
|
|
cell = row.createCell(cellReference.getCol());
|
|
|
cell.setCellValue(preConfigValue);
|
|
|
}
|
|
|
else {
|
|
|
cell.setCellValue(preConfigValue);
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
is.close();
|
|
|
FileOutputStream os = new FileOutputStream(datasetFile);
|
|
|
workbook.write(os);
|
|
|
workbook.close();
|
|
|
return datasetFile.getAbsolutePath();
|
|
|
}
|
|
|
|
|
|
private String dealWord(File datasetFile, TCComponentItemRevision revision, HashMap<String, String> nameMap,String type) throws IOException, TCException {
|
|
|
String path = datasetFile.getAbsolutePath();
|
|
|
InputStream inputStream = Files.newInputStream(Paths.get(path));
|
|
|
Map<String, String> dataMap = new HashMap<>();
|
|
|
for(String key : nameMap.keySet()) {
|
|
|
String preConfigValue = getPreConfigValue(revision,nameMap,nameMap.get(key));
|
|
|
dataMap.put(key, preConfigValue);
|
|
|
}
|
|
|
String newPath = "";
|
|
|
if(type.equals("doc")) {
|
|
|
newPath = path.replace(".doc", "temp.doc");
|
|
|
WordUtils.replaceBookmarksByDoc(inputStream, Files.newOutputStream(Paths.get(newPath)), dataMap);
|
|
|
}
|
|
|
else if(type.equals("docx")) {
|
|
|
newPath = path.replace(".docx", "temp.docx");
|
|
|
WordUtils.replaceBookmarksByDocx(inputStream, Files.newOutputStream(Paths.get(newPath)), dataMap);
|
|
|
}
|
|
|
return newPath;
|
|
|
}
|
|
|
|
|
|
private String dealWord2(File datasetFile, TCComponentItemRevision revision, HashMap<String, String> nameMap,String type) throws FileNotFoundException, Exception {
|
|
|
String path = datasetFile.getAbsolutePath();
|
|
|
|
|
|
Document document2 = new Document(new FileInputStream(path));
|
|
|
OutputStream outputStream = null;
|
|
|
String newPath = path.replace(".doc", "temp.doc");
|
|
|
outputStream = new FileOutputStream(newPath);
|
|
|
Map<String, String> dataMap = new HashMap<>();
|
|
|
|
|
|
for(String key : nameMap.keySet()) {
|
|
|
String preConfigValue = getPreConfigValue(revision,nameMap,nameMap.get(key));
|
|
|
|
|
|
System.out.println("dataMap("+key+","+preConfigValue+")");
|
|
|
dataMap.put(key, preConfigValue);
|
|
|
}
|
|
|
// ±éÀúÎĵµÖÐËùÓеÄÊéÇ©
|
|
|
for (Bookmark bookmark : document2.getRange().getBookmarks()) {
|
|
|
String bookmarkName = bookmark.getName();
|
|
|
System.out.println("ÊéÇ©Ãû£º"+bookmarkName);
|
|
|
if(dataMap.containsKey(bookmarkName))
|
|
|
{
|
|
|
System.out.println("Ìí¼ÓÄÚÈÝ");
|
|
|
bookmark.setText(dataMap.get(bookmarkName));
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// ±éÀúÎĵµÖеÄËùÓÐÓò
|
|
|
for (Field field : document2.getRange().getFields()) {
|
|
|
// »ñÈ¡ÓòµÄÃû³Æ
|
|
|
String fieldName = field.getFieldCode();
|
|
|
System.out.println("ÓòÃû£º" + fieldName);
|
|
|
|
|
|
for(String key : dataMap.keySet()) {
|
|
|
// ¸ù¾Ý×Ö¶ÎÀàÐÍ»òÆäËûÌõ¼þÉèÖÃÖµ
|
|
|
if(fieldName.contains(key)) {// ¼ÙÉèÄúÏë´¦Àí±íµ¥ÊäÈëÓò
|
|
|
System.out.println("Ìí¼ÓÄÚÈÝ");
|
|
|
field.setResult(dataMap.get(key)); // ÉèÖÃÓòµÄÖµ
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
if(type.equals("doc")) {
|
|
|
document2.save(outputStream, SaveFormat.DOC);
|
|
|
}else if(type.equals("docx")) {
|
|
|
document2.save(outputStream, SaveFormat.DOCX);
|
|
|
}
|
|
|
return newPath;
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
* µÃµ½Ê×Ñ¡ÏîÖС°=¡±×ó±ß¶ÔÓ¦µÄÊôÐÔÖµ
|
|
|
* @param revision
|
|
|
* @param nameMap
|
|
|
* @param value
|
|
|
* @return
|
|
|
* @throws TCException
|
|
|
*/
|
|
|
private String getPreConfigValue(TCComponentItemRevision revision, HashMap<String, String> nameMap,String value) throws TCException {
|
|
|
String[] split = value.split("\\.");
|
|
|
String cellValue = "";
|
|
|
TCProperty property = null;
|
|
|
if(split.length == 2) {
|
|
|
switch (split[0]) {
|
|
|
case "Item":
|
|
|
property = revision.getItem().getTCProperty(split[1]);
|
|
|
break;
|
|
|
case "Rev":
|
|
|
property = revision.getTCProperty(split[1]);
|
|
|
break;
|
|
|
case "RevMaster":
|
|
|
property = revision.getRelatedComponent("IMAN_master_form_rev").getTCProperty(split[1]);
|
|
|
break;
|
|
|
default:
|
|
|
break;
|
|
|
}
|
|
|
}
|
|
|
//È¡¹ØϵÎļþ¼ÐϵÄÊôÐÔ
|
|
|
else if(split.length == 3) {
|
|
|
String relation = split[0];
|
|
|
TCComponent relatedComponent = revision.getRelatedComponent(relation);
|
|
|
TCComponentItemRevision centerRev = null;
|
|
|
if(relatedComponent instanceof TCComponentItemRevision) {
|
|
|
centerRev = (TCComponentItemRevision) relatedComponent;
|
|
|
}else if(relatedComponent instanceof TCComponentItem) {
|
|
|
centerRev = ((TCComponentItem) relatedComponent).getLatestItemRevision();
|
|
|
}
|
|
|
if(centerRev != null) {
|
|
|
switch (split[1]) {
|
|
|
case "Item":
|
|
|
property = centerRev.getItem().getTCProperty(split[2]);
|
|
|
break;
|
|
|
case "Rev":
|
|
|
property = centerRev.getTCProperty(split[2]);
|
|
|
break;
|
|
|
case "RevMaster":
|
|
|
property = centerRev.getRelatedComponent("IMAN_master_form_rev").getTCProperty(split[2]);
|
|
|
break;
|
|
|
default:
|
|
|
break;
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
if(property == null) {
|
|
|
return "";
|
|
|
}
|
|
|
else {
|
|
|
cellValue = property.getDisplayValue();
|
|
|
return cellValue;
|
|
|
}
|
|
|
}
|
|
|
}
|
