stl_exporter.cpython-312.pyc•6.45 kB
�
H��g� � � � d dl Z d dlZd dlZd dlZd dlmZmZmZmZ d dl m
Z
ej e � Z
G d� d� Zy)� N)�Dict�Any�Optional�Tuple)� STLValidatorc �� � e Zd ZdZdefd�Zddedeeeef de ee
ee f fd�Z ddedeeeef d eeeef deeef fd
�Z d ed eeef defd �Z
d ed
edefd�Zy)� STLExporterzA
Handles STL file export and validation for 3D printing.
�
output_dirc � � || _ || _ t j j |d� | _ t j | j
d�� y)z�
Initialize the STL exporter.
Args:
openscad_wrapper: Instance of OpenSCADWrapper for generating STL files
output_dir: Directory to store output files
�stlT)�exist_okN)�openscad_wrapperr
�os�path�join�stl_dir�makedirs)�selfr r
s �:/home/ubuntu/openscad-mcp-server/src/utils/stl_exporter.py�__init__zSTLExporter.__init__ s= � � !1�� �$����w�w�|�|�J��6�� � � � �D�L�L�4�0� N� scad_file�
parameters�returnc �� � | j j ||� }t j |� \ }}|sct j d|� �� t j |� \ }}|rt j |� \ }}nt j d|� �� |||fS # t $ r9}t j dt |� � �� ddt |� fcY d}~S d}~ww xY w)a
Export a SCAD file to STL format.
Args:
scad_file: Path to the SCAD file
parameters: Optional parameters to override in the SCAD file
Returns:
Tuple of (stl_file_path, is_valid, error_message)
zSTL validation failed: zSTL repair failed: zError exporting STL: � FN)
r � generate_stlr � validate_stl�logger�warning�
repair_stl�error� Exception�str) r r r �stl_file�is_validr" �repair_success� repair_error�es r �
export_stlzSTLExporter.export_stl s� � � %��,�,�9�9�)�Z�P�H� +� 7� 7�� A�O�H�e�����!8��� @�A� 0<�/F�/F�x�/P�,�� �!�&2�&?�&?��&I�O�H�e��L�L�#6�|�n�!E�F��X�u�,� ,��� %� �L�L�0��Q���9� :��u�c�!�f�$� $�� %�s �B B � C"�).C �C"� C"�metadatac � � | j ||� \ }}}d}|r|r| j ||� }|r1t j j |� j d� d n t
t j � � }||||||d�S )ai
Export a SCAD file to STL format and include metadata.
Args:
scad_file: Path to the SCAD file
parameters: Optional parameters to override in the SCAD file
metadata: Optional metadata to include with the STL file
Returns:
Dictionary with STL file information
N�.r )�model_idr% r& r" �
metadata_filer+ ) r* �_create_metadata_filer r �basename�splitr$ �uuid�uuid4) r r r r+ r% r& r" r/ r. s r �export_stl_with_metadataz$STLExporter.export_stl_with_metadataA s� � � %)�O�O�I�z�$J�!��(�E� �
� �� � 6� 6�x�� J�M� AJ�2�7�7�#�#�I�.�4�4�S�9�!�<�s�SW�S]�S]�S_�O`�� !� � �� *� �
�
r r% c �b � t j j |� d � d�} ddl}t |d� 5 }|j ||d�� ddd� t j d|� �� |S # 1 sw Y �#xY w# t $ r+}t j dt |� � �� Y d}~y d}~ww xY w)
z'Create a metadata file for an STL file.r z.jsonN�w� )�indentzCreated metadata file: z Error creating metadata file: r ) r r �splitext�json�open�dumpr �infor# r" r$ )r r% r+ r/ r; �fr) s r r0 z!STLExporter._create_metadata_fileb s� � � �7�7�+�+�H�5�a�8�9��?�
� � ��m�S�)�Q�� � �(�A�a� �0� *�
�K�K�1�-� �A� B� � � *�)��
� � �L�L�9�#�a�&��B� C��� �s. �A: �A.� !A: �.A7�3A: �: B.�!B)�)B.� destinationc �� � t j j |� st d|� �� �t j j |� r?t j j |� }t j j
||� }t j ||� t j d|� �� |S # t $ r+}t j dt |� � �� Y d}~yd}~ww xY w)z�
Copy an STL file to a specified location.
Args:
stl_file: Path to the STL file
destination: Destination path or directory
Returns:
Path to the copied STL file
zSTL file not found: zCopied STL file to: zError copying STL file: Nr )r r �exists�FileNotFoundError�isdirr1 r �shutil�copy2r r> r# r" r$ )r r% r@ �filenamer) s r �copy_stl_to_locationz STLExporter.copy_stl_to_locationq s� � � ��7�7�>�>�(�+�'�*>�x�j�(I�J�J� �w�w�}�}�[�)� �7�7�+�+�H�5�� �g�g�l�l�;�� A� �
�L�L��;� /� �K�K�.�{�m�<� =� � ��� � �L�L�3�C��F�8�<� =��� �s �B:B= �= C1�!C,�,C1)N)NN)�__name__�
__module__� __qualname__�__doc__r$ r r r r r �boolr* r5 r0 rH � r r r r s� � ��
1�S�
1� %�C� %�X�d�3��8�n�5M� %�Y^�_b�dh�jr�sv�jw�_w�Yx� %�D _c�DH�
�#�
�8�D�QT�VY�QY�N�C[�
�)1�$�s�C�x�.�)A�
�MQ�RU�WZ�RZ�^�
�B
�c�
�T�#�s�(�^�
�PS�
� �S� �s� �s� r r )r �loggingr3 rE �typingr r r r �src.utils.stl_validatorr � getLoggerrI r r rN r r �<module>rS s8 �� � � �
� -� -� 0� �� � �8� $��A� Ar