�
� h1 � �V � d dl mZ d dlmZ dddgd�dg d�d�Z ed � Zej
� d
ed efd �� Zej
� d
ed
ee d efd�� Z ej
� d
ed efd�� Z
ej d� ded efd�� Z e
dk( rej � yy)� )�FastMCP)�List� z
2024-12-25z
2025-01-01)�balance�history� )�E001�E002� LeaveManager� employee_id�returnc �H � t j | � }|r | � d|d � d�S y)z3Check how many leave days are left for the employeez has r z leave days remaining.�Employee ID not found.)�employee_leaves�get)r �datas �3C:\Users\sister\Desktop\codebasics-mcp\mcp1\main.py�get_leave_balancer s3 � � � � �{� +�D� � � �e�D��O�#4�4J�K�K� #� � leave_datesc �� � | t vryt |� }t | d }||k r d|� d|� d�S t | dxx |z cc< t | d j |� d|� dt | d � d�S ) zM
Apply leave for specific dates (e.g., ["2025-04-17", "2025-05-01"])
r r z*Insufficient leave balance. You requested z day(s) but have only �.r zLeave applied for z day(s). Remaining balance: )r �len�extend)r r �requested_days�available_balances r � apply_leaver s� � �
�/�)�'���%�N�'� �4�Y�?���>�)�;�N�;K�Ka�bs�at�tu�v�v� �K� ��+�~�=�+��K� ��+�2�2�;�?�
�� /�/K�O�\g�Lh�ir�Ls�Kt�tu� v�vr c �x � t j | � }|r#|d rdj |d � nd}d| � d|� �S y)z"Get leave history for the employeer z, zNo leaves taken.zLeave history for z: r )r r �join)r r r s r �get_leave_historyr - sH � � � � �{� +�D� �04�Y��$�)�)�D��O�,�EW��#�K�=��7�)�<�<� #r zgreeting://{name}�namec � � d| � d�S )zGet a personalized greetingzHello, z3! How can I assist you with leave management today?� )r! s r � get_greetingr$ 7 s � � �T�F�M� N�Nr �__main__N)�mcp.server.fastmcpr �typingr r �mcp�tool�strr r r �resourcer$ �__name__�runr# r r �<module>r. s �� &� � � �l�'C� D��r� *��� �n� �� ����$�3� $�3� $� �$� ����w�S� w�t�C�y� w�S� w� �w�* ����$�3� $�3� $� �$� � � �!�"�O�s� O�s� O� #�O� �z���G�G�I� r
