NX 도면 출도 전 자동검사｜품번·재질·작성자·축척 누락 확인 Journal

# -*- coding: utf-8 -*-
"""NXOpen 2506 reference-oriented READ-ONLY audit prototype.

NOT validated in an actual Siemens NX runtime.
The pure validator is independently tested with fake objects below.
This code NEVER saves, updates, changes attributes or approves a drawing.
"""

import csv
import io
import math

# Sample company mapping ONLY: adapt to your PRT/Teamcenter attributes.
ATTRIBUTE_ALIASES = {
    'part_number': ('PART_NUMBER', 'DB_PART_NO'),
    'material': ('MATERIAL', 'MAT_SPEC'),
    'drafter': ('DRAWN_BY', 'DRAFTER'),
}


def attribute_map(obj):
    result = {}
    for item in obj.GetUserAttributes():
        key = str(item.Title).strip().upper()
        try:
            val = str(item.StringValue).strip()
        except Exception:
            val = ''
        if not val:
            try:
                val = str(item.RealValue).strip()
            except Exception:
                pass
        if key and val:
            result[key] = val
    return result


def first_value(attributes, candidates):
    return next((attributes.get(x.upper(), '') for x in candidates
                 if attributes.get(x.upper(), '').strip()), '')


def inspect_part(part):
    """Returns rows and issues; safe from all NX write calls."""
    attributes = attribute_map(part)
    part_name = str(getattr(part, 'Name', '(unnamed)'))
    values = {k: first_value(attributes, a)
              for k, a in ATTRIBUTE_ALIASES.items()}
    base_issues = [k.upper() + '_MISSING' for k, v in values.items() if not v]
    rows = []
    try:
        sheets = list(part.DrawingSheets.ToArray())
    except Exception as exc:
        return [{**values, 'part': part_name, 'sheet': '-', 'scale': '-',
                 'view_count': '-', 'status': 'API_ERROR',
                 'issues': 'DRAWING_SHEETS_READ_FAILED: ' + type(exc).__name__}]
    if not sheets:
        return [{**values, 'part': part_name, 'sheet': '-', 'scale': '-',
                 'view_count': 0, 'status': 'REVIEW',
                 'issues': ';'.join(base_issues + ['NO_DRAWING_SHEET'])}]
    for sheet in sheets:
        issues = list(base_issues)
        name = str(getattr(sheet, 'Name', '(unnamed)'))
        scale, view_count = '-', '-'
        try:
            ratio = sheet.GetScale()   # Python binding: tuple of out parameters
            if not isinstance(ratio, (tuple, list)) or len(ratio) != 2:
                raise ValueError('unexpected scale result')
            n, d = float(ratio[0]), float(ratio[1])
            if not (math.isfinite(n) and math.isfinite(d) and n > 0 and d > 0):
                issues.append('INVALID_SHEET_SCALE')
            else:
                scale = f'{n:g}:{d:g}'
        except Exception as exc:
            issues.append('SCALE_API_ERROR:' + type(exc).__name__)
        try:
            view_count = len(sheet.GetDraftingViews())
            if view_count == 0:
                issues.append('NO_DRAFTING_VIEW')
        except Exception as exc:
            issues.append('VIEW_API_ERROR:' + type(exc).__name__)
        rows.append({**values, 'part': part_name, 'sheet': name,
                     'scale': scale, 'view_count': view_count,
                     'status': 'REVIEW' if issues else 'ATTRIBUTE_CHECK_OK',
                     'issues': ';'.join(issues) or '-'})
    return rows


def csv_text(rows):
    columns = ['part','sheet','part_number','material','drafter','scale',
               'view_count','status','issues']
    out = io.StringIO()
    writer = csv.DictWriter(out, fieldnames=columns, lineterminator='\n')
    writer.writeheader()
    for row in rows:
        safe = {k: str(row.get(k, '')) for k in columns}
        # When opening the listing as a spreadsheet, avoid CSV formula injection.
        for k, value in safe.items():
            if value.lstrip().startswith(('=', '+', '-', '@')) and value != '-':
                safe[k] = "'" + value
        writer.writerow(safe)
    return out.getvalue()


def main():
    import NXOpen
    session = NXOpen.Session.GetSession()
    listing = session.ListingWindow
    listing.Open()
    part = session.Parts.Work
    if part is None:
        listing.WriteLine('NO_WORK_PART: open a drawing part first')
        return
    try:
        rows = inspect_part(part)
        for line in csv_text(rows).splitlines():
            listing.WriteLine(line[:128])
        listing.WriteLine('READ-ONLY: result is NOT approval to release drawing.')
    except Exception as exc:
        listing.WriteLine('AUDIT_FAILED: ' + type(exc).__name__ + ': ' + str(exc)[:80])


if __name__ == '__main__':
    main()