| Current File : //usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyc |
�
��Xc @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z 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 Z d Z d Z d e f d � � YZ d e f d
� � YZ d S( i����N( t open_pipe_fileobjs( t ImagePackMetadatat
ImageMetadatas image.xzs image-md.ymls pack-md.ymlt ImagePackc B s\ e Z d d � Z e d � � Z e d d d � � Z d � Z d � Z d � Z d � Z
RS( c C s( d | _ d | _ | | _ d | _ d S( N( t Nonet pack_mdt image_mdt filenamet _ImagePack__tarball( t selfR ( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyt __init__/ s c C s t d | � �� } | j j t � } t j | j j | � � � } t j | � | _ Wd QX| j j t
� } t j | j j | � � �A } t j | � | _ | j
d � t j | j � � j � } Wd QX| | j j k r� t d | j j | � � n Wd QX| S( NR i sH image metadata appears to be corrupt (expected SHA256: {0}, actual: {1})( R R t getmembert PACK_MD_ARCNAMEt
contextlibt closingt extractfileR t from_fileobjR t IMAGE_MD_ARCNAMER R t seekt hashlibt sha256t readt hexdigestt image_md_sha256sumt RuntimeError( t clsR t packt membert md_fileR ( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyt open5 s
! t c C s} t � } t j | � | _ t � | _ | d k r^ t j j | � r^ t d j
| � � � n t j j | d j
| j j � � � | _
t | � �. } t j | j � � } | j � | j _ Wd QXt j � ��} t j � } d } t | d � �� }
t j d
d t j d | �} | r(| j � n xi t r�|
j t j � } | sMPn | j | � | j j | � | t | � 7} | r+| j | � q+q+W| j j! � | j" � Wd QX| r�| j# � n | j � | j _$ | | j _% t& j' t( j | j
d
d t �� �k }
t j � �- } | j j) | � |
j* | j+ d t, �Wd QX|
j* | d t- �|
j* | j+ d t. �Wd QXWd QX| S( NR s "{0}" is not a directorys {0}.euimagei t rbt xzs -ct stdint stdoutt wt dereferencet arcname( R s -c(/ R R t from_fileR R R t ost patht isdirt
ValueErrort formatt joint get_nvraR R R R R R R t tempfilet NamedTemporaryFilet
subprocesst Popent PIPEt startt Truet euca2oolst BUFSIZEt updateR! t writet lent closet waitt finisht image_sha256sumt
image_sizeR
R t tarfilet dump_to_fileobjt addt nameR R t
IMAGE_ARCNAME( R t image_md_filenamet image_filenamet destdirt progressbarR t
image_md_filet digestt compressed_imaget
bytes_readt original_imaget xz_proct chunkt tarballt pack_md_file( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyt buildI sR
"c C s& | j r | j j � n d | _ d S( N( R R: R ( R ( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyR: | s c C s1 | j s t � t j d | j d d � | _ | S( NRB t modet r( R t AssertionErrorR? R R ( R ( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyt __enter__� s c C s | j � d S( N( R: ( R t type_t valuet tbk( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyt __exit__� s c C sJ | j s t � t j t j d | j d d � � � } t | � SWd QXd S( sW
Return a file-like object that transparently yields the packed image.
RB RR RS N( R RT R
R R? R t _PackedImageWrapper( R RO ( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyt
open_image� s $N( t __name__t
__module__R R
t classmethodR RQ R: RU RY R[ ( ( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyR . s 1 RZ c B s> e Z d Z d � Z d � Z d � Z d � Z d d � Z RS( si
A file-like object that transparently unpacks and decompresses the
image from an image pack
c C s� d | _ d | _ | j t � } | j | � } t � \ } } t j � | _ | j d k r� t j � | j
� t j d d t j
d | d t �| _ | j
� t j | | j j � | j j j
� | j j � t j t j � n | | _ d S( s�
This method takes a tarfile.TarFile object and spawns *two* new
processes: an xz process for decompression and an additional
python process that simply feeds data from the TarFile to it.
The latter is necessary because the file-like object we get from
TarFile.extractfile cannot be passed to a subprocess directly.
For that reason, one is also free to close the tarball after
this object is created.
i R s -dR! R" t close_fdsN( s xzs -d( R t _PackedImageWrapper__subp_pidt _PackedImageWrapper__read_fhR RC R R R' t forkt setpgrpR: R0 R1 R2 R4 t _PackedImageWrapper__xz_proct shutilt copyfileobjR! R; t _exitt EX_OK( R RO R RJ t pipe_rt pipe_w( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyR
� s$
c C sM | j r9 t j t j | j � t j � | j j � n t j t j � d S( N(
R` R' t killt getpgidt signalt SIGTERMRa R: Rg Rh ( R ( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyR: � s c C s | S( N( ( R ( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyRU � s c C s | j � d S( N( R: ( R RV RW RX ( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyRY � s i����c C s | j j | � S( N( Ra R ( R t size( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyR � s ( R\ R] t __doc__R
R: RU RY R ( ( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyRZ � s ( R
R R' Re Rm R0 R? R. R5 t euca2ools.bundle.utilR t( euca2ools.commands.euimage.pack.metadataR R RC R R t objectR RZ ( ( ( sH /usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyt <module> s i