| Current File : //usr/lib/python2.7/site-packages/euca2ools/commands/ec2/resumeimport.pyc |
�
��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 s5 | 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 } | d | j d k s� t � | j d � r� | d } n | d
d d } t j j | j d � } | j
| | � } | j d
j d } | | k r,t
d j | j d | | � � � n | j d j | d d � \ } } } t j j g | j D] } t j j | j � ^ qf� }
x� | j D]� } d j | | j f � } t j | d | j d d | j d d | �} y | j � Wq�t k
r,}
|
j d k r&| j | | |
� q-� q�Xq�Wd S( Nt dry_runt
show_progresss Uploading image for taskt taskt ConversionTaskIdt conversionTasksi t conversionTaskIdt 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 AssertionErrort euca2oolst utilt get_filesizet% _ResumeImport__get_or_create_manifestt image_partst endR t formatt resolve_url_to_locationt build_progressbar_label_templatet osR0 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.pyR3 L sD
.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< RA t tempfilet SpooledTemporaryFileR R2 R1 R3 t logt infot seekR t read_from_fileobjR RB t _ResumeImport__generate_manifestt mkdtempR> R0 R? t opent dump_to_fileobjt TrueR t removet rmdir( R RE RF RI RJ R@ t manifest_s3patht manifest_destfilet get_reqRG RO 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 loglevelR; R- R. R/ R0 t sizet volumet part_sizei
i i s {0}/{1}.part.{2}R i (% R R R< t rsplitR RV t levelt file_formatR R2 RA t get_presigned_url2t self_destruct_urlt intt
image_sizet volume_sizet enumeratet sixt movest rangeR t indext startt minR: R; R> R0 R? R@ R
R t head_urlt get_urlt
delete_urlR9 t append( R RE RF t dayst timeoutRI RJ R@ t
key_prefixRG t
delete_reqRl Ry t
part_startRK t part_pathRN Rc ( ( 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 fnameRy i R R- R) R. R/ RQ Rj R! t progressbar_label( RV RW Rz R: R; R> R0 R? R@ Ry R\ R RX R R2 R R1 R3 ( R RK RM RL t part_pbar_labelR Rg ( ( sG /usr/lib/python2.7/site-packages/euca2ools/commands/ec2/resumeimport.pyt
__upload_part� s % ( t __name__t
__module__t DESCRIPTIONR R^ Rr t argparset SUPPRESSt ARGSR R3 R8 RZ RC ( ( ( sG /usr/lib/python2.7/site-packages/euca2ools/commands/ec2/resumeimport.pyR 0 s$ . # !(" R� t os.pathR> RT t requestbuilderR t requestbuilder.exceptionsR R t requestbuilder.mixinsR Rv 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.utilR5 R ( ( ( sG /usr/lib/python2.7/site-packages/euca2ools/commands/ec2/resumeimport.pyt <module> s"