misc - change Iproperties of Autodesk Inventor with Python

import os
from pathlib import Path
from datetime import datetime
import dearpygui.dearpygui as dpg
import win32com.client as win32
import pythoncom
from threading import local, Thread
from concurrent.futures import Future
import queue
import tkinter as tk
# --- CONFIG ---
FOLDER = r"D:\OneDrive\001. PROJECTEN\000. KUBUZ\OPENFRAME\CAD"
MD_OUT = r"D:\OneDrive\005. NOTES\OpenFrame\Part Registry.md"
EXTS = {".ipt", ".iam", ".idw", ".dwg"}
COLUMNS = ["FileName", "PartNumber", "Title", "Subject", "Comments", "Revision", "DateChecked"]
EDITABLE = {"PartNumber", "Title", "Subject", "Comments", "Revision"}  # which columns can be edited
# --- Inventor helpers ---
def open_inventor():
    inv = win32.Dispatch("Inventor.Application")
    inv.Visible = True
    return inv
def get_prop(doc, set_name, prop_name):
    try:
        return str(doc.PropertySets.Item(set_name).Item(prop_name).Value)
    except:
        return ""
def set_prop(doc, set_name, prop_name, value):
    try:
        doc.PropertySets.Item(set_name).Item(prop_name).Value = value
    except:
        pass
def read_metadata(inv, path: Path):
    doc = None
    try:
        doc = inv.Documents.Open(str(path), False)
        rev = get_prop(doc, "Inventor Summary Information", "Revision Number")
        title = get_prop(doc, "Inventor Summary Information", "Title")
        subject = get_prop(doc, "Inventor Summary Information", "Subject")
        partnum = get_prop(doc, "Design Tracking Properties", "Part Number")
        comments = get_prop(doc, "Inventor Summary Information", "Comments")
        date_checked_raw = get_prop(doc, "Design Tracking Properties", "Date Checked")
        # Format date to YYYY-MM-DD only
        try:
            if date_checked_raw:
                # Try to parse and reformat the date
                date_obj = datetime.strptime(date_checked_raw.split()[0], "%Y-%m-%d")
                date_checked = date_obj.strftime("%Y-%m-%d")
            else:
                date_checked = ""
        except:
            # If parsing fails, try to extract just the date part
            if date_checked_raw and " " in date_checked_raw:
                date_checked = date_checked_raw.split()[0]
            else:
                date_checked = date_checked_raw or ""
        return {
            "FilePath": str(path),
            "FileName": path.name,
            "PartNumber": partnum,
            "Title": title,
            "Subject": subject,
            "Comments": comments,
            "Revision": rev,
            "DateChecked": date_checked,
        }
    finally:
        try:
            if doc is not None: doc.Close(True)
        except:
            pass
def apply_updates(inv, rows):
    current_date = datetime.now().strftime("%Y-%m-%d")
    for r in rows:
        p = Path(r["FilePath"])
        if not p.exists():
            continue
        doc = None
        try:
            doc = inv.Documents.Open(str(p), False)
            # Track if any properties changed to update Date Checked
            properties_changed = False
            # Props
            if r["Revision"] != get_prop(doc, "Inventor Summary Information", "Revision Number"):
                set_prop(doc, "Inventor Summary Information", "Revision Number", r["Revision"])
                properties_changed = True
            if r["Title"] != get_prop(doc, "Inventor Summary Information", "Title"):
                set_prop(doc, "Inventor Summary Information", "Title", r["Title"])
                properties_changed = True
            if r["Subject"] != get_prop(doc, "Inventor Summary Information", "Subject"):
                set_prop(doc, "Inventor Summary Information", "Subject", r["Subject"])
                properties_changed = True
            if r["PartNumber"] != get_prop(doc, "Design Tracking Properties", "Part Number"):
                set_prop(doc, "Design Tracking Properties", "Part Number", r["PartNumber"])
                properties_changed = True
            if r["Comments"] != get_prop(doc, "Inventor Summary Information", "Comments"):
                set_prop(doc, "Inventor Summary Information", "Comments", r["Comments"])
                properties_changed = True
            # Update Date Checked if any properties changed
            if properties_changed:
                set_prop(doc, "Design Tracking Properties", "Date Checked", current_date)
            doc.Save()
        finally:
            try:
                if doc is not None: doc.Close(True)
            except:
                pass
def export_markdown(rows, md_path):
    os.makedirs(Path(md_path).parent, exist_ok=True)
    def esc(s):
        s = "" if s is None else str(s)
        return s.replace("|", "\\|").replace("\r", " ").replace("\n", " ")
    lines = []
    lines.append("| File Name | Part Number | Title | Subject | Comments | Revision | Date Checked |")
    lines.append("|---|---|---|---|---|---|---|")
    for r in rows:
        lines.append("| " + " | ".join([
            esc(r["FileName"]),
            esc(r["PartNumber"]),
            esc(r["Title"]),
            esc(r["Subject"]),
            esc(r["Comments"]),
            esc(r["Revision"]),
            esc(r["DateChecked"]),
        ]) + " |")
    Path(md_path).write_text("\n".join(lines), encoding="utf-8")
# --- Data ---
def list_inv_files(folder):
    return [p for p in Path(folder).glob("*") if p.is_file() and p.suffix.lower() in EXTS]
def load_rows(inv, folder):
    files = list_inv_files(folder)
    return [read_metadata(inv, p) for p in files]
# --- DPG UI Helpers ---
DPG_TAGS = {
    "folder_input": "-FOLDER-",
    "folder_dialog": "-FOLDER_DIALOG-",
    "table": "-TABLE-",
    "status": "-STATUS-",
    "edit_FileName": "-FileName-",
    "edit_PartNumber": "-PartNumber-",
    "edit_Title": "-Title-",
    "edit_Subject": "-Subject-",
    "edit_Comments": "-Comments-",
    "edit_Revision": "-Revision-",
    "edit_DateChecked": "-DateChecked-",
}
_rows_state = {
    "rows": [],
    "selected_index": None,
    "worker": None,
}
_thread_local = local()
def _ensure_com_initialized():
    try:
        if not getattr(_thread_local, "com_initialized", False):
            pythoncom.CoInitialize()
            _thread_local.com_initialized = True
    except Exception:
        pass
class COMWorker:
    def __init__(self):
        self._q = queue.Queue()
        self._thread = Thread(target=self._run, daemon=True)
        self._started = False
    def start(self):
        if not self._started:
            self._thread.start()
            self._started = True
    def call(self, op, *args):
        fut = Future()
        self._q.put((op, args, fut))
        return fut.result()
    def stop(self):
        fut = Future()
        try:
            self._q.put(("_stop", (), fut))
            fut.result(timeout=5)
        except Exception:
            pass
    def _run(self):
        inv = None
        pythoncom.CoInitialize()
        try:
            inv = open_inventor()
            while True:
                op, args, fut = self._q.get()
                if op == "_stop":
                    try:
                        fut.set_result(True)
                    except Exception:
                        pass
                    break
                try:
                    if op == "load_rows":
                        folder, = args
                        res = load_rows(inv, folder)
                        fut.set_result(res)
                    elif op == "apply_updates":
                        rows, = args
                        apply_updates(inv, rows)
                        fut.set_result(True)
                    else:
                        fut.set_exception(RuntimeError(f"Unknown op: {op}"))
                except Exception as ex:
                    try:
                        fut.set_exception(ex)
                    except Exception:
                        pass
        except Exception:
            # Swallow worker-level failures; they will surface via Future exceptions
            pass
        finally:
            try:
                if inv is not None:
                    inv.Quit()
            except Exception:
                pass
            try:
                pythoncom.CoUninitialize()
            except Exception:
                pass
def _set_status(text):
    try:
        dpg.set_value(DPG_TAGS["status"], text)
    except Exception:
        pass
def _populate_edit_fields(row_dict):
    dpg.set_value(DPG_TAGS["edit_FileName"], row_dict.get("FileName", ""))
    for c in EDITABLE:
        dpg.set_value(DPG_TAGS[f"edit_{c}"], row_dict.get(c, ""))
def _get_edit_values(base_row):
    updated = dict(base_row)
    for c in EDITABLE:
        updated[c] = dpg.get_value(DPG_TAGS[f"edit_{c}"]) or base_row.get(c, "")
    return updated
def _rebuild_table():
    table_tag = DPG_TAGS["table"]
    # clear existing rows
    children = dpg.get_item_children(table_tag, 1) or []
    for child in children:
        dpg.delete_item(child)
    # add rows
    for idx, r in enumerate(_rows_state["rows"]):
        with dpg.table_row(parent=table_tag):
            # FileName (read-only)
            dpg.add_text(r.get("FileName", ""))
            # Other columns: make editable where allowed
            for c in COLUMNS[1:]:
                if c in EDITABLE:
                    # Right align Revision and DateChecked columns using buttons
                    if c in ["Revision", "DateChecked"]:
                        # Use button with transparent background for right alignment
                        dpg.add_button(label=r.get(c, ""), width=-1, callback=_on_cell_edit, user_data=(idx, c))
                        dpg.bind_item_theme(dpg.last_item(), "right_align_theme")
                    else:
                        dpg.add_input_text(default_value=r.get(c, ""), callback=_on_cell_edit, user_data=(idx, c), width=-1)
                else:
                    # Right align Revision and DateChecked columns for read-only text too
                    if c in ["Revision", "DateChecked"]:
                        # Use button with transparent background for right alignment
                        dpg.add_button(label=r.get(c, ""), width=-1)
                        dpg.bind_item_theme(dpg.last_item(), "right_align_theme")
                    else:
                        dpg.add_text(r.get(c, ""))
def _on_row_select(sender, app_data, user_data):
    try:
        idx = int(user_data)
    except Exception:
        return
    if idx < 0 or idx >= len(_rows_state["rows"]):
        return
    _rows_state["selected_index"] = idx
    _populate_edit_fields(_rows_state["rows"][idx])
    _set_status(f"Selected row {idx+1} of {len(_rows_state['rows'])}")
def _on_cell_edit(sender, app_data, user_data):
    # For buttons, app_data is the button label (current value)
    # For input text, app_data is the new text value
    try:
        idx, col = user_data
    except Exception:
        return
    if idx < 0 or idx >= len(_rows_state["rows"]):
        return
    if col not in EDITABLE:
        return
    # If it's a button (Revision or DateChecked), we need to handle it differently
    if col in ["Revision", "DateChecked"]:
        # For now, just show a message that these fields are auto-managed
        _set_status(f"Row {idx+1} - {col} is auto-managed")
    else:
        # For input text fields
        _rows_state["rows"][idx][col] = app_data or ""
        _set_status(f"Edited row {idx+1} - {col}")
def _on_browse_folder():
    dpg.configure_item(DPG_TAGS["folder_dialog"], show=True)
def _on_folder_chosen(sender, app_data):
    # app_data contains selections dict: {"file_path_name": path}
    try:
        folder = app_data.get("file_path_name", "")
    except Exception:
        folder = ""
    if folder:
        dpg.set_value(DPG_TAGS["folder_input"], folder)
def _on_reload():
    folder = dpg.get_value(DPG_TAGS["folder_input"]) or FOLDER
    if not Path(folder).exists():
        _set_status(f"Folder not found: {folder}")
        return
    try:
        rows = _rows_state["worker"].call("load_rows", folder)
        _rows_state["rows"] = rows
        _rows_state["selected_index"] = None
        _rebuild_table()
        _set_status(f"Loaded {len(_rows_state['rows'])} files from {folder}")
    except Exception as ex:
        _set_status(f"Reload failed: {ex}")
def _on_update_row():
    # No longer needed with inline editing, keep as no-op for compatibility
    _set_status("Rows are edited directly in the table.")
def _on_apply_changes():
    try:
        _rows_state["worker"].call("apply_updates", _rows_state["rows"])
        # Refresh rows from disk
        folder = dpg.get_value(DPG_TAGS["folder_input"]) or FOLDER
        _rows_state["rows"] = _rows_state["worker"].call("load_rows", folder)
        _rows_state["selected_index"] = None
        _rebuild_table()
        # Automatically save markdown after applying changes
        try:
            export_markdown(_rows_state["rows"], MD_OUT)
            _set_status("Changes applied to Inventor files and markdown saved.")
        except Exception as ex:
            _set_status(f"Changes applied to Inventor files, but markdown save failed: {ex}")
    except Exception as ex:
        _set_status(f"Apply failed: {ex}")
def _on_save_markdown():
    try:
        export_markdown(_rows_state["rows"], MD_OUT)
        _set_status(f"Markdown saved: {MD_OUT}")
    except Exception as ex:
        _set_status(f"Save Markdown failed: {ex}")
def _on_exit():
    try:
        if _rows_state.get("worker") is not None:
            _rows_state["worker"].stop()
    except Exception:
        pass
    try:
        pythoncom.CoUninitialize()
    except Exception:
        pass
def get_screen_size():
    """Get screen dimensions and return 80% of screen size"""
    root = tk.Tk()
    screen_width = root.winfo_screenwidth()
    screen_height = root.winfo_screenheight()
    root.destroy()
    # Return 50% of screen size
    return int(screen_width * 0.5), int(screen_height * 0.5)
def center_window(window_width, window_height):
    """Calculate position to center window on screen"""
    root = tk.Tk()
    screen_width = root.winfo_screenwidth()
    screen_height = root.winfo_screenheight()
    root.destroy()
    # Calculate center position
    x = (screen_width - window_width) // 2
    y = (screen_height - window_height) // 2
    return x, y
def main():
    # Start COM worker thread that owns Inventor COM apartment
    worker = COMWorker()
    worker.start()
    _rows_state["worker"] = worker
    # Preload
    if Path(FOLDER).exists():
        try:
            _rows_state["rows"] = worker.call("load_rows", FOLDER)
        except Exception as ex:
            _set_status(f"Initial load failed: {ex}")
    else:
        _set_status(f"Folder not found: {FOLDER}")
    dpg.create_context()
    # Create theme for right-aligned buttons
    with dpg.theme(tag="right_align_theme"):
        with dpg.theme_component(dpg.mvButton):
            dpg.add_theme_style(dpg.mvStyleVar_ButtonTextAlign, 1.0, 0.5)  # Right-align text
            dpg.add_theme_color(dpg.mvThemeCol_Button, (0, 0, 0, 0))       # Transparent background
            dpg.add_theme_color(dpg.mvThemeCol_ButtonActive, (0, 0, 0, 0))
            dpg.add_theme_color(dpg.mvThemeCol_ButtonHovered, (0, 0, 0, 0))
    # Get 80% of screen size and center position
    window_width, window_height = get_screen_size()
    window_x, window_y = center_window(window_width, window_height)
    with dpg.window(label="Inventor Properties Editor", width=window_width, height=window_height, no_collapse=True, no_title_bar=True, no_resize=True, no_move=True) as main_window:
        with dpg.group(horizontal=True):
            dpg.add_text("Folder:")
            dpg.add_input_text(tag=DPG_TAGS["folder_input"], default_value=FOLDER, width=700)
            dpg.add_button(label="Browse", callback=_on_browse_folder)
            dpg.add_button(label="Reload", callback=_on_reload)
        dpg.add_spacer(height=5)
        with dpg.table(tag=DPG_TAGS["table"], header_row=True, borders_innerH=True, borders_innerV=True, borders_outerH=True, borders_outerV=True, resizable=True, policy=dpg.mvTable_SizingFixedFit, scrollY=True, height=600):
            # columns with specific widths
            dpg.add_table_column(label="FileName", init_width_or_weight=150)
            dpg.add_table_column(label="PartNumber", init_width_or_weight=150)
            dpg.add_table_column(label="Title", init_width_or_weight=300)
            dpg.add_table_column(label="Subject", init_width_or_weight=200)
            dpg.add_table_column(label="Comments", init_width_or_weight=100)
            dpg.add_table_column(label="Revision", init_width_or_weight=70)
            dpg.add_table_column(label="DateChecked", init_width_or_weight=130)
        dpg.add_spacer(height=5)
        with dpg.group(horizontal=True):
            dpg.add_button(label="Apply Changes", callback=_on_apply_changes)
            dpg.add_button(label="Close", callback=lambda: dpg.stop_dearpygui())
        dpg.add_spacer(height=5)
        dpg.add_text("", tag=DPG_TAGS["status"])  # status line
    # Folder dialog
    with dpg.file_dialog(directory_selector=True, show=False, callback=_on_folder_chosen, tag=DPG_TAGS["folder_dialog"], width=700, height=400):
        dpg.add_file_extension(".*", color=(255, 255, 255, 255))
    dpg.create_viewport(title="Inventor Properties Editor", width=window_width, height=window_height, x_pos=window_x, y_pos=window_y)
    dpg.setup_dearpygui()
    dpg.set_exit_callback(_on_exit)
    dpg.show_viewport()
    dpg.set_primary_window(main_window, True)
    # Populate table and clear edit fields
    _rebuild_table()
    if _rows_state["rows"]:
        _set_status(f"Loaded {len(_rows_state['rows'])} files from {FOLDER}")
    dpg.start_dearpygui()
    dpg.destroy_context()
if __name__ == "__main__":
    main()