HOME

�
��Xc@s�dd
lZdd
l
Z
dd
lZdd
lZdd
lZdd
lZdd
lZdd
lZdd
lZddl	m
Z

ddlmZm
Z

dZdZdZdef
d��YZd	ef
d
��YZd
S(i����N(
topen_pipe_fileobjs(tImagePackMetadatat
ImageMetadatasimage.xzsimage-md.ymlspack-md.ymlt	ImagePackcBs\eZ
dd�
Zed
��
Zeddd��
Zd�Zd�Zd�Z	d�Z
RS(	cCs(d|_
d|_|
|_d|_dS(
N(tNonetpack_mdtimage_mdtfilenamet_ImagePack__tarball(tselfR((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyt__init__/s
	
	
	
cCs

td
|
�
��}|j
jt�
}tj|j
j|�
�
�}tj|�
|_	WdQX|j
jt
�
}tj|j
j|�
�
�A}tj|�
|_|j
d�

tj|j��
j�}WdQX||j	jkr�td|j	j|��
nWdQX|S(NRisHimage metadata appears to be corrupt (expected SHA256: {0}, actual: {1})(RRt	getmembertPACK_MD_ARCNAMEt
contextlibtclosingtextractfileR
tfrom_fileobjRtIMAGE_MD_ARCNAMERRtseekthashlibtsha256treadt	hexdigesttimage_md_sha256sumtRuntimeError(tclsRtpacktmembertmd_fileR((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pytopen5s 










!

	

tcCs}t�}t
j|
�
|_t�|_|d
kr^tjj|�
r^t	dj
|�
�
�
ntjj|dj
|jj��
�|_
t|
�
�.}tj|j��
}|j�|j_WdQXtj���
}tj�}d}	t|d���}
tjd
dtjd	|�
}|r(
|j�
nxitr�
|
jtj�
}|sM
Pn|j|�

|jj|�

|	t |�
7}	|r+
|j|	�

q+
q+
W|jj!�
|j"�
WdQX|r�
|j#�
n|j�|j_$|	|j_%t&j't(j|j
d
dt�
�
�k}
tj��-}|jj)|�

|
j*|j+dt,�


WdQX|
j*|
dt-�


|
j*|j+dt.�


WdQXWdQX|S(NRs"{0}" is not a directorys{0}.euimageitrbtxzs-ctstdintstdoutt
wtdereferencetarcname(R s-c(/RRt	from_fileRR
Rtostpathtisdirt
ValueErrortformattjointget_nvraRRRRRRRttempfiletNamedTemporaryFilet
subprocesstPopentPIPEtstarttTruet	euca2oolstBUFSIZEtupdateR!twritetlentclosetwaittfinishtimage_sha256sumt
image_sizeR
Rttarfiletdump_to_fileobjtaddtnameRRt
IMAGE_ARCNAME(Rtimage_md_filenametimage_filenametdestdirtprogressbarRt
image_md_filetdigesttcompressed_imaget
bytes_readtoriginal_imagetxz_proctchunkttarballtpack_md_file((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pytbuildIsR	












	



	






















"
c

Cs&|jr|jj
�
nd|_dS(
N(RR:R(
R	((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyR:|s
	

c

Cs1|jst
�
tjd
|jdd�|_|S(NRBtmodet
r(RtAssertionErrorR?RR(
R	((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyt	__enter__�s


c
Cs|j�
dS(
N(
R:(R	ttype_tvaluettbk((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyt__exit__�s
c
CsJ|jst
�
tjtjd
|jdd��
�}
t|
�
SWdQXdS(sW
        Return a file-like object that transparently yields the packed image.
        RBRRRSN(RRTR
RR?Rt_PackedImageWrapper(R	RO((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyt
open_image�s
$
N(t__name__t
__module__RR
tclassmethodRRQR:RURYR[(((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyR.s
1			RZcBs>eZ
dZd
�Zd�Zd�Zd�Zdd�
ZRS(si
    A file-like object that transparently unpacks and decompresses the
    image from an image pack
    cCs�d|_
d|_|
jt�
}|
j|�
}t�\}}tj�|_
|j
d
kr�tj	�
|j
�
tjddtj
d|dt�
|_|j
�
tj||jj�
|jjj
�
|jj�
tjtj�

n	||_dS(	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.
        iR s-dR!R"t	close_fdsN(sxzs-d(Rt_PackedImageWrapper__subp_pidt_PackedImageWrapper__read_fhRRCRRR'tforktsetpgrpR:R0R1R2R4t_PackedImageWrapper__xz_proctshutiltcopyfileobjR!R;t_exittEX_OK(R	RORRJtpipe_rtpipe_w((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyR
�s$	
	


















c

CsM|jr9t
jt
j|j�
tj�
|jj�
nt
jt
j	�

dS(
N(
R`R'tkilltgetpgidtsignaltSIGTERMRaR:RgRh(
R	((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyR:�s
	 
c


Cs|S(
N((
R	((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyRU�s
c
Cs|j�
dS(
N(
R:(R	RVRWRX((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyRY�s
i����cCs|jj
|
�
S(
N(RaR(R	tsize((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyR�s
(R\R]t__doc__R
R:RURYR(((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyRZ�s				(R
RR'ReRmR0R?R.R5teuca2ools.bundle.utilRt(euca2ools.commands.euimage.pack.metadataR
RRCRRtobjectRRZ(((sH/usr/lib/python2.7/site-packages/euca2ools/commands/euimage/pack/pack.pyt<module>s










i