�
�Уh-
� � � S r SSKJr SSKJrJr SSKJr SSKJ r SSK
J r SSK J
r
SS KJr SS
KJr " S S 5 rg
)zsSimplified AuthFramework client using service composition.
Copyright (c) 2025 AuthFramework. All rights reserved.
� )� annotations)�Any�Self� )� AdminService)� AuthService)�
BaseClient)�
MFAService)� OAuthService)� UserServicec � � \ rS rSrSrSSSS. SS jjrSS jr SS jrSS
jrSS jr SS jr
SS
jr SS jr SS jr
Srg)�AuthFrameworkClient� z:Simplified AuthFramework client using service composition.g >@� N)�timeout�retries�api_keyc �* � [ UUUUS9U l [ U R 5 U l [ U R 5 U l [
U R 5 U l [ U R 5 U l [ U R 5 U l g)z�Initialize AuthFramework client.
Args:
base_url: Base URL of the AuthFramework server
timeout: Request timeout in seconds
retries: Number of retry attempts for failed requests
api_key: Optional API key for authentication
)�base_urlr r r N) r �_clientr �authr �userr
�mfar �oauthr �admin)�selfr r r r s �`C:\Users\cires\OneDrive\Documents\projects\AuthFramework\sdks\python\src\authframework\client.py�__init__� AuthFrameworkClient.__init__ sn � �" "� ����
�� � �� � �-�� � �� � �-�� � �d�l�l�+���!�$�,�,�/��
�!�$�,�,�/��
� c � �V # � U R R 5 I Sh v�N U $ N7f)zAAsync context manager entry.
Returns:
The client instance.
N)r �
__aenter__�r s r r" � AuthFrameworkClient.__aenter__4 s' � � � �l�l�%�%�'�'�'�� � (�s � )�'�)c � �X # � U R R XU5 I Sh v�N g N7f)z�Async context manager exit.
Args:
exc_type: Exception type if an exception occurred
exc_val: Exception value if an exception occurred
exc_tb: Exception traceback if an exception occurred
N)r � __aexit__)r �exc_type�exc_val�exc_tbs r r& � AuthFrameworkClient.__aexit__> s! � � � �l�l�$�$�X��?�?�?�s � *�(�*c � �T # � U R R 5 I Sh v�N g N7f)z(Close the client and clean up resources.N)r �closer# s r r, �AuthFrameworkClient.closeN s � � ��l�l� � �"�"�"�s � (�&�(c �: � U R R U5 g)zTSet access token for authenticated requests.
Args:
token: Access token to set
N)r �set_access_token)r �tokens r r/ �$AuthFrameworkClient.set_access_tokenR s � �
� � �%�%�e�,r c �8 � U R R 5 g)z Clear the stored access token.N)r �clear_access_tokenr# s r r3 �&AuthFrameworkClient.clear_access_token[ s � � � � �'�'�)r c �6 � U R R 5 $ )zVGet the current access token.
Returns:
Current access token or None if not set.
)r �get_access_tokenr# s r r6 �$AuthFrameworkClient.get_access_token_ s � � �|�|�,�,�.�.r c � �V # � U R R SS5 I Sh v�N $ N7f)zFCheck server health status.
Returns:
Server health information.
�GETz/healthN�r � make_requestr# s r � health_check� AuthFrameworkClient.health_checkh s$ � � � �\�\�.�.�u�i�@�@�@�@�� � )�'�)c � �V # � U R R SS5 I Sh v�N $ N7f)zcGet server information and capabilities.
Returns:
Server information and supported features.
r9 z/infoNr: r# s r �get_server_info�#AuthFrameworkClient.get_server_infoq s$ � � � �\�\�.�.�u�g�>�>�>�>�r> )r r r r r r )
r �strr �floatr �intr �
str | None�return�None)rF r )r' ztype[BaseException] | Noner( zBaseException | Noner) �objectrF rG )rF rG )r0 rB rF rG )rF rE )rF zdict[str, Any])�__name__�
__module__� __qualname__�__firstlineno__�__doc__r r" r& r, r/ r3 r6 r< r@ �__static_attributes__� r r r r s� � �D� �� "�
0�� 0� � 0�
� 0� �
0�
� 0�>�@�,�@� &�@� � @�
� @� #�-�*�/�A�?r r N)rM �
__future__r �typingr r �_adminr �_authr �_baser �_mfar
�_oauthr �_userr r rO r r �<module>rX s0 ���
#� � � � � � � �f?� f?r