3
b @ s d dl mZmZ d dlZddlmZmZmZ ddlm Z yd dl
j j Z
W n ek
rf d dlj Z
Y nX ejdkZG dd deZG d d
d
eZG d d d eZG d
d deZedkr
dZeeZedej ej edej ej edej ee dS ) )absolute_importprint_functionN )getTreeFromXml encode_to_s3decode_from_s3) deunicodise c @ sD e Zd ZdZdZdd Zdd Zdd Zd d
Zd d Z d
d Z
dS )Granteez/http://acs.amazonaws.com/groups/global/AllUsersz.http://acs.amazonaws.com/groups/s3/LogDeliveryc C s" d | _ d | _d | _d| _d | _d S )N )xsi_typetagname display_name
permission)self r F/oak/stanford/groups/akundaje/marinovg/programs/s3cmd-master/S3/ACL.py__init__ s
zGrantee.__init__c C s t d| j| j| jd S )Nz0Grantee("%(tag)s", "%(name)s", "%(permission)s"))r
r r )reprr
r r )r r r r __repr__! s zGrantee.__repr__c C s | j dko| jtjkS )NURI)r
r r
ALL_USERS_URI)r r r r
isAllUsers( s zGrantee.isAllUsersc C s | j o| jdkp| jdkS )NREAD FULL_CONTROL)r r )r r r r
isAnonRead+ s zGrantee.isAnonReadc C s | j o| jdkp| jdkS )NWRITEr )r r )r r r r isAnonWrite. s zGrantee.isAnonWritec C sN t jd}t j|dd| jd}t j|| j}| j|_t j|d}| j|_|S )NZGrantr
z)http://www.w3.org/2001/XMLSchema-instance)z xmlns:xsizxsi:type
Permission)ETElement
SubElementr r
r textr )r elgranteer r r r r
getElement1 s
zGrantee.getElementN) __name__
__module__ __qualname__r LOG_DELIVERY_URIr r r r r r& r r r r r
s r
c @ s e Zd Zdd ZdS )GranteeAnonReadc C s( t j| d| _d| _t j| _d| _d S )NGroupr r )r
r r r
r r r )r r r r r > s
zGranteeAnonRead.__init__N)r' r( r) r r r r r r+ = s r+ c @ s e Zd Zdd ZdS )GranteeLogDeliveryc C s( t j| d| _d| _t j| _|| _dS )z=
permission must be either READ_ACP or WRITE
r, r N)r
r r r
r* r r )r r r r r r F s
zGranteeLogDelivery.__init__N)r' r( r) r r r r r r- E s r- c @ s e Zd ZdZd%ddZdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ ZdS )&ACLsj Nc C s@ |s
t j}g | _d| _d| _tt|}| j| | j| d S )Nr ) r. EMPTY_ACLgranteesowner_id
owner_nickr r
parseOwner parseGrants)r xmltreer r r r S s
z ACL.__init__c C s |j d| _|j d| _d S )Nz .//Owner//IDz.//Owner//DisplayName)findtextr1 r2 )r r6 r r r r3 _ s zACL.parseOwnerc C s xz|j dD ]l}t }|jd}|jd |_|jdj|_x0|D ](}|jdkrX|j|_q@|j|_|j|_ q@W | j
j | q W d S )Nz.//Grantz
.//Granteez/{http://www.w3.org/2001/XMLSchema-instance}typer Z DisplayName) findallr
findattribr r# r r
r r r0 append)r r6 grantr% gr$ r r r r4 c s
zACL.parseGrantsc C sL g }xB| j D ]8}|jr |j}n|j r,d}n|j}|j||jd q W |S )Nz*anon*)r% r )r0 r r r r; r )r aclr% userr r r getGrantListq s zACL.getGrantListc C s | j | jdS )N)idZnick)r1 r2 )r r r r getOwner} s z ACL.getOwnerc C s x| j D ]}|j rdS qW dS )NTF)r0 r )r r% r r r r s zACL.isAnonReadc C s x| j D ]}|j rdS qW dS )NTF)r0 r )r r% r r r r s zACL.isAnonWritec C s | j s| jt d S )N)r
appendGranteer+ )r r r r
grantAnonRead s zACL.grantAnonReadc C s dd | j D | _ d S )Nc S s g | ]}|j s|qS r )r ).0r= r r r
s z&ACL.revokeAnonRead..)r0 )r r r r revokeAnonRead s zACL.revokeAnonReadc C s dd | j D | _ d S )Nc S s g | ]}|j s|qS r )r )rE r= r r r rF s z'ACL.revokeAnonWrite..)r0 )r r r r revokeAnonWrite s zACL.revokeAnonWritec C s | j j| d S )N)r0 r; )r r% r r r rC s zACL.appendGranteec C sR |j }|j }x<| jD ]2}|jj |kr|jdkr8dS |jj |krdS qW dS )Nr TF)lowerupperr0 r r )r r r r% r r r hasGrant s
z ACL.hasGrantc C s | j ||rd S |j }d|kr$d}d|kr8| j|d t }||_||_d|krl|jj |_d|_d|_n.d|krd|_d|_n|jj |_d |_d
|_| j | d S ) NALLr @ZAmazonCustomerByEmailZ EmailAddressz http://acs.amazonaws.com/groups/r, r Z
CanonicalUserZID)
rK rJ revoker
r r rI r r
rC )r r r r% r r r r< s* z ACL.grantc sL j j dkr0 fdd| jD | _n fdd| jD | _d S )NrL c s6 g | ].}|j j kp,|jd k o,|jj ks|qS )N)r rI r )rE r= )r r r rF s z ACL.revoke..c sD g | ]<}|j d k r |j j ks.|jj ko:|jj ks|qS )N)r rI r r rJ )rE r= )r r r r rF s ()rI rJ r0 )r r r r )r r r rN s
z
ACL.revokec C sR t tj}d|jd< |jd}| j|_|jd}x | jD ]}|j|j q8W |S )Nz'http://s3.amazonaws.com/doc/2006-03-01/xmlnsz .//Owner//IDz.//AccessControlList)
r r. r/ r: r9 r1 r# r0 r; r& )r r6 ownerr> r% r r r get_printable_tree s
zACL.get_printable_treec C s t tj| j S )N)r r tostringrQ )r r r r __unicode__ s zACL.__unicode__c C s( t rtj| j ddS tj| j S d S )Nunicode)encoding)PY3r rR rQ )r r r r __str__ s z ACL.__str__)N)r' r( r) r/ r r3 r4 r@ rB r r rD rG rH rC rK r< rN rQ rS rW r r r r r. P s$
r. __main__s
12345678901234567890
owner-nickname
12345678901234567890
owner-nickname
FULL_CONTROL
http://acs.amazonaws.com/groups/global/AllUsers
READ
zGrants:)r r )
__future__r r sys BaseUtilsr r r Utilsr xml.etree.ElementTreeetree ElementTreer ImportErrorelementtree.ElementTree version_inforV objectr
r+ r- r. r' r5 r> printr@ rG rD r r r r s,
'