�
��i� � � � d Z ddlmZ ddlZddlZddlZddlZddlmZ ee � j � j j Z e dz Z
dZej j! dd� Zd Zd
Zdd �Zdd �Zed
k( r e� yy)uM
Replace the Mermaid block between COMPOSE_MERMAID_AUTO_START/END in README.md
using docker-compose-viz-mermaid (docker-compose.yml → Mermaid text).
Requires Docker. From repo root:
python scripts/update_compose_mermaid_readme.py
Environment:
COMPOSE_VIZ_IMAGE — override image (default: derlin/docker-compose-viz-mermaid)
� )� annotationsN)�Pathz README.mdzdocker-compose.yml�COMPOSE_VIZ_IMAGEz!derlin/docker-compose-viz-mermaidz#<!-- COMPOSE_MERMAID_AUTO_START -->z!<!-- COMPOSE_MERMAID_AUTO_END -->c �� � t t z } | j � st j d| � �� ddddt � d�ddt
t d d
g }t
j |d d � � }|j d
k7 r_|j j � xs |j j � xs d}t j d|j � d|� �� |j j d� S )NzMissing �docker�runz--rmz-vz:/workz-wz/workz-f�textT)�capture_outputr r z
unknown errorz#docker-compose-viz-mermaid failed (z): �
)
�ROOT� COMPOSE_REL�is_file�sys�exit�IMAGE�
subprocessr �
returncode�stderr�strip�stdout�rstrip)�compose�cmd�r�msgs �(scripts/update_compose_mermaid_readme.py�run_vizr s� � ��[� �G� �?�?� � ���8�G�9�%�&� �
�� � �&��� ��
�� �� �C� ���s�4�d�;�A��|�|�q���h�h�n�n� �E�!�(�(�.�.�"2�E�o�� ���6�q�|�|�n�C��u�M�N� �8�8�?�?�4� � � c � � t j d�� } t | vst | vr)t j
t � dt �dt ��� t
� j � }|st j
d� d|� d�}t j t j t � dz t j t � z t j � }|j t � d |� d t � �| d
� � \ }}|d
k7 rt j
d � || k( r t d
� y t j |dd �� t d� y )Nzutf-8)�encodingz must contain z and z(Visualizer returned empty Mermaid outputz ```mermaid
z
```z \s*.*?\s*r � )�countzRCould not replace exactly one marked region (check markers are unique and paired).z README.md already up to date.)r �newlinez"Updated README.md Mermaid diagram.)�README� read_text�START�ENDr r r r �re�compile�escape�DOTALL�subn�print�
write_text)r � mermaid_body�block�pattern�new_text�ns r �mainr4 6 s
� � � � �W� � -�D� �D��C�t�O� ���F�8�>�%��%��w�?�@��9�?�?�$�L� � ���;�<��<�.�� .�E��j�j����5�)�L�8�2�9�9�S�>�I�2�9�9�U�G��,�,�%���5�'��C�5� 9�4�q�,�I�K�H�a��A�v� ���e�f��4��
�-�.��
���h��$��?� �
.�/r �__main__)�return�str)r6 �None)�__doc__�
__future__r �osr( r r �pathlibr �__file__�resolve�parentr r$ r
�environ�getr r& r' r r4 �__name__� r r �<module>rD s� �� � #� � � �
� � �H�~� � � �&�&�-�-��
� � ��"� �
�
�
���*�,O�P��-��)��!�20�. �z���F� r