�
+��dc@s�
dZd
dl
Z
d
dlZd
dlZd
dlmZ
d
dlZd
dlmZ
d
dl	m
Z

def
d��YZdeejfd	��YZ
d
ef
d��YZdef
d
��YZdef
d��YZde
f
d��YZde
f
d��YZddd��YZddd��YZdd d��YZd
dlmZ
e
jjd�
s�
e
jje
jjejd��
r�
d
dlmZ
ndS(!s�TurnKey Hub API - Backup

Notes:
    - Default URL: https://hub.turnkeylinux.org/api/backup/
    - REST compliant (GET, POST, PUT)
    - Responses are returned in application/json format
    - API subkey must be sent in the header for all calls (except subkey/)

subkey/
    method: GET
    fields: apikey
    return: subkey

credentials/
    method: GET
    fields:
    return: accesskey, secretkey, usertoken, producttoken

record/create/
    method: POST
    fields: key, turnkey_version, [server_id]
    return: backuprecord

record/update/
    method: PUT
    fields: address
    return: <response_code>

record/<backup_id>/
    method: GET
    fields:
    return: backuprecord

record/<backup_id>/
    method: PUT
    fields: key
    return: backuprecord

records/
    method: GET
    fields:
    return: [ backuprecord, ... ]

archive/
    method: GET
    fields: turnkey_version
    return: archive_content

archive/timestamp/
    method: GET
    fields: turnkey_version
    return: archive_timestamp

Exceptions::

    400 Request.MissingHeader
    400 Request.MissingArgument
    401 HubAccount.Forbidden
    400 HubAccount.InvalidApiKey
    400 BackupAccount.InvalidSubKey
    401 BackupAccount.MalformedSubKey
    404 BackupAccount.NotFound
    401 BackupAccount.NotSubscribed
    404 BackupRecord.NotFound
    401 BackupRecord.LimitExceeded
    400 BackupRecord.ServerIDNotFound
    404 BackupArchive.NotFound
i����N(
tdatetime(
tAPI(
tAttrDicttErrorc
BseZ
d�Zd
�ZRS(cGs tj
||
|�
|
|_dS(
N(t	Exceptiont__init__tdescription(tselfRtargs((s/usr/lib/tklbam/hub.pyR\s

c


Cs|jS(
N(
R(
R((s/usr/lib/tklbam/hub.pyt__str__`s
(t__name__t
__module__RR	(((s/usr/lib/tklbam/hub.pyR[s
	tAPIErrorc
BseZ
d�ZRS(
cCstj
j||
||�
dS(
N(t_APIRR(RtcodetnameR((s/usr/lib/tklbam/hub.pyRds
(R
RR(((s/usr/lib/tklbam/hub.pyRcs
t
NotSubscribedcBseZ
dZed
�
ZRS(s|Backups are not yet enabled for your TurnKey Hub account. Log
into the Hub and go to the "Backups" section for instructions.cCstj
||
�
dS(
N(RR(Rtdesc((s/usr/lib/tklbam/hub.pyRls
(R
RtDESCR(((s/usr/lib/tklbam/hub.pyRgstInvalidBackupErrorc
BseZ
RS((R
R(((s/usr/lib/tklbam/hub.pyRos
R
cBseZ
iid�ZRS(
cCs�ytj
||
|||�SWnm|jk
r�
}
|jd
krSt|j�
�
n|jdkrnt��
nt|j|j|j��
n
XdS(NsBackupRecord.NotFoundsBackupAccount.NotSubscribedsBackupAccount.NotFound(sBackupAccount.NotSubscribedsBackupAccount.NotFound(	R
trequestRRRRRRR(Rtmethodturltattrstheaderst
e((s/usr/lib/tklbam/hub.pyRss





	
(R
RR(((s/usr/lib/tklbam/hub.pyR
rs
tBackupRecordcBs eZ
ed��
Zd
�ZRS(c

Cs|s
dSt
j|d
�S(Ns%Y-%m-%d %H:%M:%S(tNoneRtstrptime(
t
s((s/usr/lib/tklbam/hub.pyt_parse_datetime�s
cCs�|
d
|_|
d|_
|
d|_|
d|_|
d|_|j|
d�
|_|j|
d�
|_t|
d�
|_	|
d	|_
g|_tj
|�

dS(
Ntkeytaddresst	backup_idt	server_idtturnkey_versiontdate_createdtdate_updatedtsizeR(RR R!R"t
profile_idRtcreatedtupdatedtintR&tlabeltsessionsRR(Rtresponse((s/usr/lib/tklbam/hub.pyR�s












	(R
RtstaticmethodRR(((s/usr/lib/tklbam/hub.pyR�s
tBaseCredentialsc
BseZ
d�ZRS(
c

Cs|jj
j�|_dS(
N(t	__class__R
tlowerttype(
R((s/usr/lib/tklbam/hub.pyR�s
(R
RR(((s/usr/lib/tklbam/hub.pyR/�s
tCredentialscBsYeZ
def
d
��YZdef
d��YZdef
d��YZed��
ZRS(tIAMRolec
BseZ
d�ZRS(
cCs5|
|_||_
||_||_tj|�

dS(
N(t	accesskeyt	secretkeytsessiontokent
expirationR/R(RR5R6R7R8((s/usr/lib/tklbam/hub.pyR�s

	
	
	
	(R
RR(((s/usr/lib/tklbam/hub.pyR4�s
tIAMUserc
BseZ
d�ZRS(
cCs,|
|_||_
||_tj|�

dS(
N(R5R6R7R/R(RR5R6R7((s/usr/lib/tklbam/hub.pyR�s
	
	
	(R
RR(((s/usr/lib/tklbam/hub.pyR9�s
tDevPayc
BseZ
d�ZRS(
cCs5|
|_||_
||_||_tj|�

dS(
N(R5R6t	usertokentproducttokenR/R(RR5R6R;R<((s/usr/lib/tklbam/hub.pyR�s

	
	
	
	(R
RR(((s/usr/lib/tklbam/hub.pyR:�s
cCs�td
�|j
j�D�
�
}|
jd�
}|
j�}y|d=Wntk
rX


n
X|sl|j|�S||kr�td|�
�
n|||�S(Nc
ssB|]8}
t|
t
�rt|
t�r|
jj�|
fV
qdS(
N(t
isinstanceR2t
issubclassR/R
R1(t.0tsubcls((s/usr/lib/tklbam/hub.pys	<genexpr>�s

R2sunknown credentials type "%s"(tdictt__dict__tvaluestgettcopytKeyErrorR:R(tclst
dtcreds_typest
creds_typetkwargs((s/usr/lib/tklbam/hub.pyt	from_dict�s	








(R
RR/R4R9R:tclassmethodRL(((s/usr/lib/tklbam/hub.pyR3�s		tBackupscBs�eZ
ejdd
�ZeZdef
d��YZdd�
Zid�
Z	e
d��
Zd�Zd�Z
dd	�
Zd
�Zd�Zd�Zd
�Zd�ZRS(t
TKLBAM_APIURLs(https://hub.turnkeylinux.org/api/backup/tNotInitializedc
BseZ
RS((R
R(((s/usr/lib/tklbam/hub.pyRP�s
cCs7|
dkr|j
d
�
�
n|
|_t�|_dS(Ns(no APIKEY - tklbam not linked to the Hub(RRPtsubkeyR
tapi(RRQ((s/usr/lib/tklbam/hub.pyR�s

	
cCs6i
t|j
�
d
6}|jj|
|j|||�S(NRQ(tstrRQRRRtAPI_URL(RRturiRR((s/usr/lib/tklbam/hub.pyt_api�s

cCs.t�j
d
|jdi
|
d6�}|dS(NtGETssubkey/tapikeyRQ(R
RRT(RGRXR-((s/usr/lib/tklbam/hub.pytget_sub_apikey�s&
c
Cs|jd
d�}
t
j|
�
S(NRWscredentials/(RVR3RL(RR-((s/usr/lib/tklbam/hub.pytget_credentials�s

c
Cs�i
|
d
6}|jdd|�}t
|d�
}|rH||krHd
S|jdd|�}tjt|d�
�
}tjdd�
\}}tj	|d	�}	|	j
|�

|	j�
t|
||�S(s


        Gets a profile for <profile_id> that is newer than <profile_timestamp>.

        If there's a new profile, returns a ProfileArchive instance.
        Otherwise returns None.

        Raises an exception if no profile exists for profile_id.
        R#RWsarchive/timestamp/tarchive_timestampsarchive/tarchive_contenttprefixsarchive.t
wN(
RVR*Rtbase64turlsafe_b64decodeRSttempfiletmkstemptostfdopentwritetclosetProfileArchive(
RR'tprofile_timestampRR-R[tcontenttfdtarchive_pathtfh((s/usr/lib/tklbam/hub.pytget_new_profile�s









cCsFi|
d
6|d6}|r'||d<n|jdd|�}t
|�
S(NRR#R"tPOSTsrecord/create/(RVR(RRR'R"RR-((s/usr/lib/tklbam/hub.pytnew_backup_record	
s





cCs |jd
d|
�}t
|�
S(NRWs
record/%s/(RVR(RR!R-((s/usr/lib/tklbam/hub.pytget_backup_record
s

cCs$|jd
d|
i
|d6�}|S(NtPUTsrecord/%s/inprogress/tbool(
RV(RR!RrR-((s/usr/lib/tklbam/hub.pytset_backup_inprogress
s

cCs*|jd
d|
i
|d6�}t
|�
S(NRqs
record/%s/R(RVR(RR!RR-((s/usr/lib/tklbam/hub.pyt
update_key
s
 
cCs |jd
di
|
d6�}|S(NRqsrecord/update/R (
RV(RR R-((s/usr/lib/tklbam/hub.pytupdated_backup
s

c
Cs"|jd
d�}
t
d�|
�S(NRWsrecords/c

Ss
t|�
S(
N(
R(
t
r((s/usr/lib/tklbam/hub.pyt<lambda>%
t(RVtmap(RR-((s/usr/lib/tklbam/hub.pytlist_backups#
s

N(R
RRctgetenvRTRRPRRRVRMRYRZRmRoRpRsRtRuRz(((s/usr/lib/tklbam/hub.pyRN�s


						Rgc
Bs#eZ
d�Zd
�Zd�ZRS(cCs||_||_
|
|_dS(
N(tpath_archivet	timestampR'(RR'tarchiveR}((s/usr/lib/tklbam/hub.pyR(
s
	
	
cCstj
d
|j|
f�

dS(Nstar -zxf %s -C %s(texecutiltsystemR|(Rtpath((s/usr/lib/tklbam/hub.pytextract-
s
c

Cs,tj
j|j�
r(tj|j�

ndS(
N(RcR�texistsR|tremove(
R((s/usr/lib/tklbam/hub.pyt__del__0
s

(R
RRR�R�(((s/usr/lib/tklbam/hub.pyRg'
s
		(
tConftTKLBAM_DUMMYHUBtdummyhub(
RN((((t__doc__RcR_RaRRtpycurl_wrapperR
R
tutilsRRRRRRRR/R3RNRgtconfR�tenvironRDR�R�tjointDEFAULT_PATHR�(((s/usr/lib/tklbam/hub.pyt<module>Os(



4T

6