o
    !d_	                     @   s$   d Z ddlmZ dd Zdd ZdS )a  
This customization makes it easier to save various pieces of data
returned from iot commands that would typically need to be saved to a
file. This customization adds the following options:

- aws iot create-certificate-from-csr
  - ``--certificate-pem-outfile``: certificatePem
- aws iot create-keys-and-certificate
  - ``--certificate-pem-outfile``: certificatePem
  - ``--public-key-outfile``: keyPair.PublicKey
  - ``--private-key-outfile``: keyPair.PrivateKey
    QueryOutFileArgumentc                 K   sJ   d}t | dd|dd|d< t | dd|dd|d< t | dd	|dd|d< d
S )zAdd outfile save arguments to create-keys-and-certificate

    - ``--certificate-pem-outfile``
    - ``--public-key-outfile``
    - ``--private-key-outfile``
    z'after-call.iot.CreateKeysAndCertificatecertificate-pem-outfilecertificatePem  sessionnamequeryZafter_call_eventpermzpublic-key-outfilezkeyPair.PublicKeyzprivate-key-outfilezkeyPair.PrivateKeyNr   )r   argument_tablekwargsZafter_event r   =usr/lib/python3.10/site-packages/awscli/customizations/iot.py'register_create_keys_and_cert_arguments   s   

r   c                 K   s   t | ddddd|d< dS )z:Add certificate-pem-outfile to create-certificate-from-csrr   r   z'after-call.iot.CreateCertificateFromCsrr   r   Nr   )r   r   r   r   r   r   'register_create_keys_from_csr_arguments/   s
   r   N)__doc__Zawscli.customizations.argumentsr   r   r   r   r   r   r   <module>   s   