hdf5_handler.cpython-313.pyc•4.37 kB
�
�h#
� �� � S SK r S SKrS SKJrJr S SKJr SSKJr \" S5 r S\\
\4 S\\
\4 4S jr S\\
\4 S\\
\4 4S
jr S\\
\4 S\\
\4 4S jr
S\\
\4 S\\
\4 4S jrg)
� N)�Dict�Any)�Path� )�JSONRPCResponse�mock_hdf5_data�params�returnc � � # � SU ; a [ SSS.S9R 5 $ U S n US:X a [ U 5 I S h v�N $ US:X a [ U 5 I S h v�N $ US:X a [ U 5 I S h v�N $ [ S S
U 3S.S9R 5 $ NS N< N%! [
a- n[ S [
U5 S.S9R 5 s S nA$ S nAff = f7f) N�actioni����zMissing 'action' parameter)�code�message)�error�list�read�metadatai����zUnknown action: i ���)r �dict� _handle_list� _handle_read�_handle_metadata� Exception�str)r r �es �G/Users/jafaralzoubi/Desktop/mcp-server/src/capabilities/hdf5_handler.py�handle_requestr s� � � ��v� � �!�.J�K�
�
�$�&� � �H�
�F�� �V� �%�f�-�-� -�
�v�
�%�f�-�-� -�
�z�
!�)�&�1�1� 1�"� %�4D�V�H�2M�N���d�f�
�
.�-�1��
� � �!�c�!�f�5�
�
�$�&� ���s� �&C�B �B�B �C�B �B�B �C�B �/B�0B �3C�4 B �C�B �B �B �
C�!"C �C�C� C�Cc � � # � U R SS5 nU R SS5 n[ U- n[ R SSS9 [ S- [ S- [ S - S
- /nU H, nUR R SSS9 UR SS 9 M. [ X1- 5 n[ R " USS 9nU Vs/ s H* n[ [ U5 R [ 5 5 PM, n nU [ U 5 UUUS
.S.$ s snf 7f)N�patternz*.h5�path� T)�parents�exist_okzsimulation_run_123.h5zexperiment_data_456.h5�resultszanalysis_output.h5)r! )� recursive)r r �
searchPath)�files�count�context)
�get�
MOCK_DATA_DIR�mkdir�parent�touchr �globr � relative_to�len)
r r �subdir� search_path�
mock_files� file_path� full_pattern�
matched_files�f�relative_filess
r r r ! s � � ��j�j��F�+�G�
�Z�Z���
#�F� �&�(�K� ����t��4� �/�/��0�0�� �!�$8�8��J�
� ���� � �t�d� �;������&� �
�{�,�-�L��I�I�l�d�;�M� :G��9F�A��D��G� � �
� .�/� � � �
��^�$�� �&�
� � �� �s �B7D�91C?�*Dc � �� # � U R S5 nU R S5 nU( a U( d [ S5 e[ U- nUR 5 ( d [ SU 35 eUSS/S/ SQS.US S
S .S .$ 7f)
N�filePath�datasetz+Both 'filePath' and 'dataset' are required.�File not found: �d �float32)g�������?g �?g�������?)�shape�dtype� sampleValues�KelvinzSimulated temperature data)�file�units� description)r: �datar' �r( �
ValueErrorr) �exists�FileNotFoundError)r r3 r: � full_paths r r r E s� � � ��
�
�:�&�I��j�j��#�G� �G��F�G�G� � �)�I� � � � � � �"2�9�+� >�?�?� ��3�Z� � +�
� � �7�
� � �s �A:A<c � � # � U R S5 nU( d [ S5 e[ U- nUR 5 ( d [ SU 35 eUSSSSS.S .$ 7f)
Nr9 z Missing 'filePath' parameter.r; �HDF5z1.10z2023-01-01T00:00:00Zz1.2GB)�format�version�created�size)rB r rF )r r3 rJ s r r r a sn � � ��
�
�:�&�I� ��8�9�9� � �)�I� � � � � � �"2�9�+� >�?�?� � � �-��
� � �s �AA )�osr- �typingr r �pathlibr �modelsr r) r r r r r � � r �<module>rW s� �� � � � � #��%�&�
���c�3�h� � �D��c��N� �0"�t�C��H�~� "�$�s�C�x�.� "�H�t�C��H�~� �$�s�C�x�.� �8�4��S��>� �d�3��8�n� rV