fred-dev/comfy_ui_ali
0
1�
2 ���g� � �4 � d dl mZ d dlZd dlZ G d� d� Zy)� )�on_flushNc �0 � e Zd Zd� Zd� Zd� Zd� Zd� Zd� Zy)�TerminalServicec �v � || _ d | _ d | _ t � | _ t | j � y �N)�server�cols�rows�set�
subscriptionsr �
send_messages)�selfr s �o/Users/fredrodrigues/Library/CloudStorage/Dropbox/Code/comfyui_hardware/api_server/services/terminal_service.py�__init__zTerminalService.__init__ s/ � � �� ��� ��� � �U�����#�#�$� c �� � t j � }|j |j fS # t $ r@ t j � }|j |j fcY S # t $ r Y Y yw xY ww xY w)N)�P � )�os�get_terminal_size�columns�lines�OSError�shutil)r �sizes r r z!TerminalService.get_terminal_size so � � ��'�'�)�D��L�L�$�*�*�-� -��� �
� �/�/�1�� � � �d�j�j�1�1���
� �
�� �s, �+. � A7�+A&�#A7�& A3�/A7�2A3�3A7c �� � | j � \ }}d}|| j k7 r || _ d}|| j k7 r || _ d}|r| j | j d�S y )NFT)r r3 )r r r4 )r r r �changeds r � update_sizezTerminalService.update_size sb � � �/�/�1������ �d�i�i� � �D�I��G� �D�I�I� � �D�I��G� � �I�I�t�y�y�9� 9�r c �: � | j j |� y r )r �add�r � client_ids r � subscribezTerminalService.subscribe* s � � ��� � �y�)r c �: � | j j |� y r )r �discardr! s r � unsubscribezTerminalService.unsubscribe- s � � ���"�"�9�-r c �8 � t |� rt | j � sy | j � }| j j � D ]L }|| j j5 vr| j
|� �-| j j d||d�|� �N y )N�logs)�entriesr )�lenr r �copyr �socketsr& � send_sync)r r) �new_sizer" s r r
z TerminalService.send_messages0 s � ��7�|�3�t�'9�'9�#:� ��#�#�%�� �+�+�0�0�2�I��� � � 3� 3�3�� � ��+�� �K�K� !� !�&�g�x�*P�R[� \�
3r N) �__name__�6__module__� __qualname__r r r r# r& r
� r r r r s! � �%� ��"*�.� ]r r )�7app.loggerr r r r r2 r r �<module>r4 s �� � �
�6]� 6]r