�
p� h � �l � d Z ddlZddlZddlZddlmZmZmZ ej d� Z G d� d� Z
y)z=
Project-related command implementations for KiCAD interface
� N)�Dict�Any�Optional�kicad_interfacec �� � e Zd ZdZd deej fd�Zdee e
f dee e
f fd�Z dee e
f dee e
f fd�Z dee e
f dee e
f fd �Z
dee e
f dee e
f fd
�Zy) �ProjectCommandsz(Handles project-related KiCAD operationsN�boardc � � || _ y)z'Initialize with optional board instanceN)r )�selfr s �,c:\repo\KiCAD-MCP\python\commands\project.py�__init__zProjectCommands.__init__ s � ���
� �params�returnc �� � |j dd� }|j dt j � � }|j d� }t j j ||� }|j d� s|dz
}t j t j j |� d�� t j � }|j � j |� dd l m } |j � j d
� }|j � j |� |r�t j j! |� } t j j# | � rSt j$ | � }
|j' |
j) � � |j+ |
j- � � |j/ dd � } |j1 | � t j2 | |� t5 |d � 5 } | j7 d
� | j7 d� | j7 dt j j9 | � � d�� | j7 d� | j7 d� ddd� || _ dd|� �||| d�d�S # 1 sw Y � xY w# t< $ r:}
t> jA dtC |
� � �� ddtC |
� d�cY d}
~
S d}
~
ww xY w)zCreate a new KiCAD project� projectName� New_Project�path�template�
.kicad_proT)�exist_okr )�datetimez%Y-%m-%d�
.kicad_pcb�wz{
z
"board": {
z "filename": "z"
z }
z}
NzCreated project: ��namer � boardPath��success�message�projectzError creating project: FzFailed to create project�r r � errorDetails)"�get�os�getcwdr �join�endswith�makedirs�dirname�pcbnew�BOARD�
GetTitleBlock�SetTitler �now�strftime�SetDate�
expanduser�exists� LoadBoard�SetDesignSettings�GetDesignSettings�
SetLayerStack�
GetLayerStack�replace� SetFileName� SaveBoard�open�write�basenamer � Exception�logger�error�str)r r � project_namer r � project_pathr r � current_date�
template_path�template_board�
board_path�f�es r �create_projectz ProjectCommands.create_project sY � �@ �!�:�:�m�]�C�L��:�:�f�b�i�i�k�2�D� �z�z�*�-�H� �7�7�<�<��l�;�L� �(�(� �6� � �,� �
�K�K������ �5�� E� �L�L�N�E�
� � � !� *� *�<� 8�
*�#�<�<�>�2�2�:�>�L� � � � !� )� )�,� 7� � "��� 2� 2�8� <�
��7�7�>�>�-�0�%+�%5�%5�m�%D�N��+�+�N�,L�,L�,N�O��'�'��(D�(D�(F�G� &�-�-�l�L�I�J� � � �j� )� � � �Z�� /� �l�C�(�A���������(�)����+�B�G�G�,<�,<�Z�,H�+I��M�N����� ������ )� �D�J� � .�|�n�=� (� (�!+� ��
� )�(��& � � �L�L�3�C��F�8�<� =� �5� #�A���
�� �s7 �H J! �A7J�9J! �J �J! �! K$�*/K �K$� K$c � � |j d� }|sdddd�S t j j t j j |� � }|j d� r|j
dd� }n|}t j |� }|| _ dd t j j |� � �t j j t j j |� � d
||d �d �S # t $ r:}t j d
t |� � �� ddt |� d�cY d}~S d}~ww xY w)z Open an existing KiCAD project�filenameFzNo filename providedz"The filename parameter is requiredr" r r TzOpened project: r r r zError opening project: zFailed to open projectN)r$ r% r �abspathr2 r( r9 r+ r4 r r> �splitextr? r@ rA rB )r r rM rH r rJ s r � open_projectz ProjectCommands.open_projectW s) � �& � �z�z�*�-�H�� $� 5�$H�� � �w�w���r�w�w�'9�'9�(�'C�D�H� � � � �.� %� -� -�l�L� I�
� %�
� �$�$�Z�0�E� �D�J� � -�b�g�g�.>�.>�z�.J�-K�L� �G�G� ,� ,�R�W�W�-=�-=�j�-I� J�1� M� $�!+� ��
�� � � �L�L�2�3�q�6�(�;� <� �3� #�A���
�� �s# �D �C$D � E� /D>�8E�>Ec � � | j sdddd�S |j d� }|rWt j j t j j |� � }| j j
|� t j | j j � | j � dd| j j � � �t j j t j j | j j � � � d | j j � d �d
�S # t $ r:}t j d t |� � �� dd t |� d�cY d
}~S d
}~ww xY w)z Save the current KiCAD projectF�No board is loaded� Load or create a board firstr" rM TzSaved project to: r )r r r zError saving project: zFailed to save projectN)r r$ r% r rN r2 r: r+ r; � GetFileNamerO r> r? r@ rA rB )r r rM rJ s r � save_projectz ProjectCommands.save_project� s; � � ��:�:� $� 3�$B�� � �z�z�*�-�H�� �7�7�?�?�2�7�7�+=�+=�h�+G�H���
�
�&�&�x�0�
� � �T�Z�Z� 3� 3� 5�t�z�z� B� � /��
�
�0F�0F�0H�/I�J� �G�G� ,� ,�R�W�W�-=�-=�d�j�j�>T�>T�>V�-W� X�YZ� [� �J�J� 2� 2� 4� ��
�� � � �L�L�1�#�a�&��:� ;� �3� #�A���
�� �s# �E �D2E � F
�/F�?F
�F
c
� � | j sdddd�S | j j � }| j j � }dt j j t j j
|� � d ||j � |j � |j � |j � |j d� |j d� |j d� |j d � d
�
d �S # t $ r:}t j d t |� � �� dd
t |� d�cY d}~S d}~ww xY w)z)Get information about the current projectFrR rS r" Tr � � � )
r r �title�date�revision�company�comment1�comment2�comment3�comment4)r r! z Error getting project info: z!Failed to get project informationN)r r- rT r% r rO r> �GetTitle�GetDate� GetRevision�
GetCompany�
GetCommentr? r@ rA rB )r r � title_blockrM rJ s r �get_project_infoz ProjectCommands.get_project_info� s- � �! ��:�:� $� 3�$B�� � �*�*�2�2�4�K��z�z�-�-�/�H� � �G�G� ,� ,�R�W�W�-=�-=�h�-G� H�� K� $� (� 1� 1� 3� '� /� /� 1� +� 7� 7� 9� *� 5� 5� 7� +� 6� 6�q� 9� +� 6� 6�q� 9� +� 6� 6�q� 9� +� 6� 6�q� 9� ��
�� � � �L�L�7��A��x�@� A� �>� #�A���
�� �s# �D �C4D � E �/E�E �E )N)�__name__�
__module__� __qualname__�__doc__r r+ r, r
r rB r rK rP rU rh � r r r r s� � �2��h�v�|�|� 4� �B�T�#�s�(�^� B��S�#�X�� B�H(�4��S��>� (�d�3��8�n� (�T"�4��S��>� "�d�3��8�n� "�H#�t�C��H�~� #�$�s�C�x�.� #r r ) rl r% r+ �logging�typingr r r � getLoggerr@ r rm r r �<module>rq s: ���
�
� � &� &� �� � �,� -��|� |r
