o
    !d                     @   s   d dl Z G dd deZdS )    Nc                   @   s$   e Zd Zdd Zdd Zdd ZdS )IAMc                 C   s
   || _ d S )N)
iam_client)selfr    r   Kusr/lib/python3.10/site-packages/awscli/customizations/emrcontainers/iam.py__init__   s   
zIAM.__init__c                 C   s   | j j|d}|ddS )z2Method to retrieve trust policy of given role name)RoleNameZRoleZAssumeRolePolicyDocument)r   Zget_roleget)r   	role_nameZroler   r   r   get_assume_role_policy   s   zIAM.get_assume_role_policyc                 C   s   | j j|t|dS )z0Method to update trust policy of given role name)r   ZPolicyDocument)r   update_assume_role_policyjsondumps)r   r
   Zassume_role_policyr   r   r   r      s   zIAM.update_assume_role_policyN)__name__
__module____qualname__r   r   r   r   r   r   r   r      s    r   )r   objectr   r   r   r   r   <module>   s   