| Current File : //usr/lib/python2.7/site-packages/euca2ools/commands/ec2/resumeimport.pyo |
�
��Xc @ s( d d l Z d d l Z d d l Z d d l m Z d d l m Z m Z d d l m
Z
d d l Z d d l m
Z
d d l m Z d d l m Z d d l m Z m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d
l m Z d d l Z d e
e e
f d � � YZ! d S( i����N( t Arg( t
ArgumentErrort ServerError( t FileTransferProgressBarMixin( t
EC2Request( t DescribeConversionTasks( t
S3AccessMixin( t ImportManifestt ImportImagePart( t DeleteObject( t
HeadObject( t GetObject( t PutObject( t AWSErrort ResumeImportc
B s� e Z d Z e d d d d d �e d d d e d d �e d
d d d d
e d d d d �e d d d d
e d d d e j �e d d
e d e j �e d d d d e j �e d d d d e j �g Z d � Z d � Z
d � Z d � Z d � Z
RS( s) Perform the upload step of an import taskt sourcet metavart FILEt helps3 file containing the disk image to import (required)s -ts --taskt requireds. the ID of the import task to resume (required)s -xs --expirest DAYSt typet defaulti s\ how long the import manifest should remain valid, in
days (default: 30 days)s --part-sizet MiBi
s --user-threadss --dont-verify-formatt actiont
store_trues --dry-runc C s_ t j | � | j � | j j d � s9 d | j d <n | j d d k r[ t d � � n d S( Nt expiresi i s- argument -x/--expires: value must be positive( R t configuret configure_s3_accesst argst getR ( t self( ( sG /usr/lib/python2.7/site-packages/euca2ools/commands/ec2/resumeimport.pyR C s
c C s | j j d � r d S| j j d t � r> d G| j d GHn t j | d | j d g �} | j � d d } | j d � r� | d } n | d d
d } t j j | j d � } | j | | � } | j
d j d
} | | k rt d j
| j d | | � � � n | j d j | d d � \ } } } t j j g | j
D] } t j j | j � ^ qI� }
x� | j
D]� } d j | | j f � } t j | d | j d d | j d d | �} y | j � Wqwt k
r}
|
j d k r | j | | |
� q� qwXqwWd S( Nt dry_runt
show_progresss Uploading image for taskt taskt ConversionTaskIdt conversionTasksi t importVolumet importInstancet volumesR i����i s` file "{0}" is not the same size as the file the import started with (expected: {1}, actual: {2})t
s3_servicet imaget importManifestUrlt /t servicet autht s3_autht pathi� ( R R t FalseR t
from_othert maint euca2oolst utilt get_filesizet% _ResumeImport__get_or_create_manifestt image_partst endR t formatt resolve_url_to_locationt build_progressbar_label_templatet osR/ t basenamet keyt joinR
R
t status_codet _ResumeImport__upload_part( R t desc_convR" t
vol_containert file_sizet manifestt file_size_from_manifestt _t buckett partt pbar_label_templatet part_s3patht head_reqt err( ( sG /usr/lib/python2.7/site-packages/euca2ools/commands/ec2/resumeimport.pyR2 L sB
.c C s� | j d j | d d � \ } } } d j | | f � } y� t j d d � �x } t j | d | j d d | j d d
| d | d t �} | j � | j j
d
� | j d � t j
| � } Wd QXWn� t k
r�}
|
j d k r�| j j
d � | j | | � } t j � } t j j | t j j | � � } t | d � � }
| j |
d t �Wd QXt j | d
| d | d t �} | j � t j | � t j | � q�� n X| S( NR( R) R* R+ t max_sizei � R, R- R. R t destR! s. using existing import manifest from the serveri i� s creating new import manifestt wt pretty_print( R R: R? t tempfilet SpooledTemporaryFileR R1 R0 R2 t logt infot seekR t read_from_fileobjR R@ t _ResumeImport__generate_manifestt mkdtempR< R/ R= t opent dump_to_fileobjt TrueR t removet rmdir( R RC RD RG RH R> t manifest_s3patht manifest_destfilet get_reqRE RM t tempdirt manifest_filenamet
manifest_filet put_req( ( sG /usr/lib/python2.7/site-packages/euca2ools/commands/ec2/resumeimport.pyt __get_or_create_manifestz s@
c C sO | j j d � p d } | d } | j d j | d d � \ } } } | j d d � d } t d
| j j � } | d d | _ t j | d | j d d
| j d d d j
| | f � �}
|
j | � | _ t
| d d � | _ t
| d d � | _ | j j d � pd d } x)t t j j d | | � � D] \ } }
t � } | | _ |
| _ t |
| | � d | _ d j | t j j | j d � | � | _ d j
| | j f � } t j |
d | �} t j |
d | �} t j |
d | �}
| j | � | _ | j | � | _! |
j | � | _" | j# j$ | � q>W| S( NR i i�Q R( R) R* R+ i i t loglevelR9 R, R- R. R/ t sizet volumet part_sizei
i i s {0}/{1}.part.{2}R i (% R R R: t rsplitR RT t levelt file_formatR R1 R? t get_presigned_url2t self_destruct_urlt intt
image_sizet volume_sizet enumeratet sixt movest rangeR t indext startt minR8 R9 R< R/ R= R> R
R t head_urlt get_urlt
delete_urlR7 t append( R RC RD t dayst timeoutRG RH R> t
key_prefixRE t
delete_reqRj Rw t
part_startRI t part_pathRL Ra ( ( sG /usr/lib/python2.7/site-packages/euca2ools/commands/ec2/resumeimport.pyt __generate_manifest� s>
%c C s� | j j d | | j | j � | j d t j j | j � d | j d � } t
| j d � �� } | j | j � t
j | d | j d d | j d d | d
| d | j | j d d | j j d t � d
| �} | j � SWd QXd S( Ns Uploading part %s (bytes %i-%i)t fnameRw i R R, R( R- R. RO Rh R! t progressbar_label( RT RU Rx R8 R9 R< R/ R= R> Rw RZ R RV R R1 R R0 R2 ( R RI RK RJ t part_pbar_labelR Re ( ( sG /usr/lib/python2.7/site-packages/euca2ools/commands/ec2/resumeimport.pyt
__upload_part� s % ( t __name__t
__module__t DESCRIPTIONR R\ Rp t argparset SUPPRESSt ARGSR R2 R6 RX RA ( ( ( sG /usr/lib/python2.7/site-packages/euca2ools/commands/ec2/resumeimport.pyR 0 s$ . # !(" R� t os.pathR< RR t requestbuilderR t requestbuilder.exceptionsR R t requestbuilder.mixinsR Rt t euca2ools.commands.ec2R t. euca2ools.commands.ec2.describeconversiontasksR t euca2ools.commands.ec2.mixinsR t! euca2ools.commands.ec2.structuresR R t" euca2ools.commands.s3.deleteobjectR t euca2ools.commands.s3.headobjectR
t euca2ools.commands.s3.getobjectR t euca2ools.commands.s3.putobjectR t euca2ools.exceptionsR
t euca2ools.utilR3 R ( ( ( sG /usr/lib/python2.7/site-packages/euca2ools/commands/ec2/resumeimport.pyt <module> s"