o
!dY @ s d dl Z d dlmZ d dlZd dlZd dlZd dlmZ d dlm Z e
dZ dZ dZ
dZd Zd
Zd d Zd
d ZdZdd Zdd Zdd ZdefdefdefgZdd ZG dd d e ZdS ) N)sha1)six)CustomArgumentZec2bundleinstancezThe bucket in which to store the AMI. You can specify a bucket that you already own or a new bucket that Amazon EC2 creates on your behalf. If you specify a bucket that belongs to someone else, Amazon EC2 returns an error.zCThe prefix for the image component names being stored in Amazon S3.z7The access key ID of the owner of the Amazon S3 bucket.ac An Amazon S3 upload policy that gives Amazon EC2 permission to upload items into Amazon S3 on the user's behalf. If you provide this parameter, you must also provide your secret access key, so we can create a policy signature for you (the secret access key is not passed to Amazon EC2). If you do not provide this parameter, we generate an upload policy for you automatically. For more information about upload policies see the sections about policy construction and signatures in the Amazon Simple Storage Service Developer Guide.zThe AWS secret access key for the owner of the Amazon S3 bucket specified in the --bucket parameter. This parameter is required so that a signature can be computed for the policy.c K s | d }d|_ tddtd}|| d< tddtd}|| d< tdd td}|| d < td
d td}|| d < td d
td}|| d
< d S )NstorageFBucketbucket)
storage_paramnameZ help_textPrefixprefixZAWSAccessKeyIdz
owner-akid_SAKz owner-sak UploadPolicypolicy)requiredBundleArgument BUCKET_DOCS PREFIX_DOCSOWNER_AKID_DOCSOWNER_SAK_DOCS POLICY_DOCS)Zargument_tablekwargsZ storage_argarg r Lusr/lib/python3.10/site-packages/awscli/customizations/ec2/bundleinstance.py _add_paramsB s6 r c K s@ t | t| }|d r dD ]}|| rd}t|qd S d S )Nr )r r Z
owner_akidZ owner_sakr zOMixing the --storage option with the simple, scalar options is not recommended.)loggerdebugvars
ValueError)Z parsed_argsr Zarg_dictkeymsgr r r _check_args^ s
r! z{{"expiration": "{expires}","conditions": [{{"bucket": "{bucket}"}},{{"acl": "ec2-bundle-read"}},["starts-with", "$key", "{prefix}"]]}}c C sH t jdd}t j | }|d}tj|| d | d d}|| d< d S )N )hoursz%Y-%m-%dT%H:%M:%S.%fZr r
)expiresr r r
)datetime timedeltautcnowstrftimePOLICYformat)paramsdeltar$ Z expires_isor r r r _generate_policyv s
r- c C s | d}| d}|r@|rBtt|d}tj|dt d}|
t| t |
d}|| d< | d= d S d S d S )Nr
r zutf-8)Z digestmodUploadPolicySignature)getbase64 b64encoder bdecodehmacnewencoder update encodebytesdigeststrip)r+ r ZsakZnew_hmacZpsr r r _generate_signature s
r; c K s4 | d d }d|vrt | d|vrt| d S d S )NStorageS3r
r. )r- r; )r+ r r r r r
_check_params s r> z+building-argument-table.ec2.bundle-instancez)operation-args-parsed.ec2.bundle-instancez)before-parameter-build.ec2.BundleInstancec C s t D ]
\}}| || qd S N)EVENTSregister)Z
event_handlereventhandlerr r r register_bundleinstance s rD c s, e Zd Z fddZdd Zdd Z ZS )r c s t t| j|i | || _d S r? )superr __init___storage_param)selfr argsr __class__r r rF s
zBundleArgument.__init__c C s* d|vr
di i|d< ||d d | j < d S )Nr< r= )rG )rH r+ valuer r r _build_storage s z BundleArgument._build_storagec C s |r
| || d S d S r? )rM )rH
parametersrL r r r
add_to_params s z BundleArgument.add_to_params)__name__
__module__ __qualname__rF rM rO
__classcell__r r rJ r r s r )loggingZhashlibr r4 r0 r% Z
awscli.compatr Zawscli.argumentsr getLoggerr r r r r r r r! r) r- r; r> r@ rD r r r r r s4