OcIc@sdZddklZddkZddkZddkZddkZddkZddkZde fdYZ dZ dZ dZ d Zd dd YZeied dd YZeiedddYZdddYZdeifdYZdddYZddkZddkZddkZddkZddkZddkZddkZddkZddkZddk Z ddk!Z!ddk"Z"ddk#Z#dS(s,Read increment files and restore to originali(t generatorsNt RestoreErrorcBseZRS((t__name__t __module__(((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRscCsw|iii}|iii}|i||i|||i|}|i|}|i|||i dS(s?Recursively restore mirror and inc_rpath to target at rest_timeN( tconntrestoret MirrorStructt TargetStructtset_mirror_and_rest_timestinitialize_rf_cachetget_initial_itert get_diffstpatchtclose_rf_cache(t mirror_rpt inc_rpathttargettrestore_to_timetMirrorStTargetSt target_itert diff_iter((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pytRestores c Cs|i\}}tioti|}n|i|i|d}|ipgSn|i}g}xo|i D]a}t i |}|oE|d|jo4|i |}|i pt|i |qxqxW|S(s"Returns increments with given basei((tdirsplittGlobalstchars_to_quotetFilenameMappingtunquotet __class__Rtisdirtindextlistdirtrpathtget_incfile_infotappendt isincfiletAssertionError( Rtdirnametbasenamet parent_dirRtinc_listtfilenametinc_infotinc((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyt get_inclist(s     c cs|itijp tdti|ti||titid}titi d}t i ||}x|D]\}}|p d}n,|p d}n||joq}nd}|o |i p |i } t id|| ffVq}WtidS(s2List the changed files under mirror_rp since rest time Notice the output is an iterator of RORPs. We do this because we want to give the remote connection the data in buffered increments, and this is done automatically for rorp iterators. Encode the lines in the first element of the rorp's index. sRun locally onlyitnewtdeletedtchangeds%-7s %sN(RRtlocal_connectionR$RRR tget_mirror_rorp_itert _rest_timet _mirror_timetrorpitert Collate2Iterst get_indexpathR tRORPathR ( Rtinc_rpRtold_itertcur_itertcollatedtold_rorptcur_rorptchanget path_desc((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pytListChangedSince9s&       ccs`|itijp tdti|ti||ti}x|D] }|VqMWdS(suList the files in archive at the given time Output is a RORP Iterator with info in index. See ListChangedSince. sRun locally onlyN(RRR0R$RRR R1(RR8ttimeR9trorp((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyt ListAtTimeSs  RcBseZdZdZdZdZdZdZdZ ddZ dZ dZ dddZ dZd Zd Zd Zd Zd ZRS(s+Hold functions to be run on the mirror sidecCs%|it_|i|t_dS(s>Set class variabels _mirror_time and _rest_time on mirror connN(tget_mirror_timeRR3t get_rest_timeR2(tclsR((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRiscCshttiid}|ptiidn(t|djotiddn|diS(s)Return time (in seconds) of latest mirrortcurrent_mirrors$Could not get time of current mirroris5Warning, two different times for current mirror foundii( R,RtrbdirR"tlogtLogt FatalErrortlent getinctime(RFtcur_mirror_incs((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRDns csG|i}tfd|}|ot|Sn t|SdS(sReturn older time, if restore_to_time is in between two inc times There is a slightly tricky reason for doing this: The rest of the code just ignores increments that are older than restore_to_time. But sometimes we want to consider the very next increment older than rest time, because rest_time will be between two increments, and what was actually on the mirror side will correspond to the older one. So if restore_to_time is inbetween two increments, return the older one. cs |jS(((RA(R(s9/var/lib/python-support/python2.5/rdiff_backup/restore.pytsN(tget_increment_timestfiltertmaxtmin(RFRtinctimest older_times((Rs9/var/lib/python-support/python2.5/rdiff_backup/restore.pyREws  cCs|iphd|i<}nhd|i<}| p |i otiid}nx$t|D]}d||i.dir file and the mirror_metadata file. Sorted ascending. t incrementstmirror_metadataN( R3tNoneRDRRRHR"R,RMtkeystsort(RFtrptdR+t return_list((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRPs    cCsSt|}t||t|}|||_|_||_t||_dS(s#Set cls.rf_cache to CachedRF objectN(R,t RestoreFilet mirror_basetinc_basetroot_rftCachedRFtrf_cache(RFR_R`R(trf((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR s   cCs|iidS(s)Run anything remaining on CachedRF objectN(Rctclose(RF((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR scCs|djo |i}ntitii||ii}|pA|oti i dnti dd|i |i }n|i oti|i |}n|S(sReturn iter of mirror rps at given restore time Usually we can use the metadata file, but if this is unavailable, we may have to build it from scratch. If the cls._select object is set, use it to filter out the unwanted files from the metadata_iter. sMirror metadata not founds:Warning: Mirror metadata not found, reading from directoryiN(RXR2tmetadatat SetManagert ManagerObjt GetAtTimeR_RRIRJRKtget_rorp_iter_from_rfRat_selectt selectiont FilterIter(RFt rest_timetrequire_metadatat rorp_iter((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR1s     cGs:|p tdti||_|ii||dS(s&Initialize the mirror selection objects+If no selection options, don't use selectorN(R$RltSelectRkt ParseArgs(RFt target_rpt select_optst filelists((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pytset_mirror_selectsccs\|i}|V|io:x7|iD]%}x|i|D] }|VqAWq+WndS(s&Recursively yield mirror rorps from rfN(t get_attribsRt yield_sub_rfsRj(RFRdRBtsub_rftattribs((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRjs   cs.djoSnfd}|S(sSubtract index from index of each rorp in rorp_iter subtract_indicies and add_indicies are necessary because we may not be restoring from the root index. c3s]xVD]N}|it jpt|if|it|_|VqWdS(N(RRLR$(RB(RRp(s9/var/lib/python-support/python2.5/rdiff_backup/restore.pytget_iters -(((RFRRpR{((RRps9/var/lib/python-support/python2.5/rdiff_backup/restore.pytsubtract_indiciess cCs=|i|ii|i}ti||}|i|S(sGiven rorp iter of target files, return diffs Here the target_iter doesn't contain any actual data, just attribute listings. Thus any diffs we generate will be snapshots. (R|R_RR1R4R5tget_diffs_from_collated(RFRtmir_iterR;((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR sccsx|D]\}}tio|oti||n| p4| p,||j ptio*ti|| o|i||}nd}tio|oti|n|o |VqqWdS(sGet diff iterator from collatedN(Rtpreserve_hardlinkstHardlinktadd_rorptrorp_eqtget_diffRXtdel_rorp(RFR;tmir_rorpt target_rorptdiff((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR}s  cCs|pti|i}ntio*ti|o|iti|nP|i oB|i i|i}|i i ||}|i ti|n|id|S(sGet a diff for mir_rorp at timetsnapshot(R R7RRRRtislinkedt flaglinkedtget_link_indextisregR_Rctget_fptsetfilethasht FileWrappertset_attached_filetype(RFRRtexpanded_indextfile_fp((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRs  N(RRt__doc__RXRkR3R2RRDRERPR R R1RvRjR|R R}R(((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR`s"        RcBs/eZdZdZdZdZdZRS(s:Hold functions to be run on the target side when restoringcGs6ti||_|ii|||iidS(s:Return a selection object iterating the rorpaths in targetN(RlRqRkRrtset_iter(RFRRtRu((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pytset_target_selectscCs|ipti|iS(s4Return selector previously set with set_initial_iter(RkRlRqR(RFR((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR scCsttit|g}xDti||D]0}tid|id||i|q(W|i|i dS(sPatch target with the diffs from the mirror side This function and the associated ITRB is similar to the patching code in backup.py, but they have different error correction requirements, so it seemed easier to just repeat it all in this module. sProcessing changed file iN( R4tIterTreeReducert PatchITRBt FillInIterRIRJR6RtFinishtsetdata(RFRRtITRR((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR s  N(RRRRXRkRR R (((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR s   RbcBsJeZdZdZdZddZdZddZdZ RS(sStore RestoreFile objects until they are needed The code above would like to pretend it has random access to RFs, making one for a particular index at will. However, in general this involves listing and filtering a directory, which can get expensive. Thus, when a CachedRF retrieves an RestoreFile, it creates all the RFs of that directory at the same time, and doesn't have to recalculate. It assumes the indicies will be in order, so the cache is deleted if a later index is requested. cCs<||_g|_tidjot|i|_ndS(s*Initialize CachedRF, self.rf_list variableiN(Ratrf_listRt process_uidtPermissionChangerRt perm_changer(tselfRa((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyt__init__:s  cCs_d|f}dig}|iD]}|t|iq!~}d}di|||fS(s=Used for debugging, return indicies of cache rfs for printings!-------- Cached RF for %s -------t s--------------------------s (tjoinRtstrR(RRts1t_[1]Rdts2ts3((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pytlist_rfs_in_cacheAs 6cCsx|ip|i||pdSq,n|id}|i|jo,tidjo|i||n|Sq|i|jo8|d |id jp|i|| odSqq|id=qdS(s*Get a RestoreFile for given index, or NoneiiN(Rtadd_rfsRXRRRR(RRRRd((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pytget_rfHs    cCswti|i||||ii}|p?tid|odi|pdfdti dSn|i S(s3Return the file object (for reading) of given indexsgError: Unable to retrieve data for file %s! The cause is probably data loss from the backup repository.t/t.it( tlongnamet update_rfRRaRRIRJRt cStringIOtStringIOtget_restore_fp(RRRRd((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRYs %cCs|p |iSn|iodSn|d }tidjo|i|nt|iii||iii|g}t |i }|pdSn||i dd+dS(sGiven index, add the rfs in that same directory Returns false if no rfs are available, which usually indicates an error. Niii( Rathas_alt_mirror_nameRRRR^Rt new_indexR8tlistRxR(RRRt parent_indexttemp_rftnew_rfs((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRds   cCs%tidjo|iindS(s)Finish remaining rps in PermissionChangeriN(RRRtfinish(R((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRevsN( RRRRRRXRRRRe(((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRb,s     R^cBsheZdZdZdZdZdZdZdZdZ dZ d Z d Z RS( sHold data about a single mirror file and its related increments self.relevant_incs will be set to a list of increments that matter for restoring a regular file. If the patches are to mirror_rp, it will be the first element in self.relevant.incs cCs6|i|_||_|||_|_|idS(N(RRR8R(tset_relevant_incs(RRR8R(((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRs  cCs}ddi|ifg}|ig}|iD])}|d|i|i|ifq0~|iddi|S(s<Return printable string of relevant incs, used for debuggings---- Relevant incs for %sRs%s %s %ss --------------------------------s (RRtextendt relevant_incst getinctypetlstattpathR"(RtlRR+((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pytrelevant_incs_strings  : cCsd|i_d|i_|i ptitijo|ig|_dSn|i}d}x>|t |jo*||i djoPn|d}qbW||d |_|i p|idi djo|ii |in|ii dS(sSet self.relevant_incs to increments that matter for restoring relevant_incs is sorted newest first. If mirror_rp matters, it will be (first) in relevant_incs. RiNRii( Rtinc_typetinc_compressedR(RR2R3Rtget_newer_incsRLRR"treverse(Rt newer_incsti((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRs$     cCszg}xD|iD]9}|i}|tijo|i||fqqW|ig}|D]}||dqb~S(sReturn list of newer incs sorted by time (increasing) Also discard increments older than rest_time (rest_time we are assuming is the exact time rdiff-backup was run, so no need to consider the next oldest increment or any of that) i(R(RMRR2R"RZ(RtincpairsR+RARtpair((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRs   cCst|id}|idjoti|iSn|i}|i|_|idjod|id|i p|ii|jo|ii||_n|iS(s2Return RPath by adding index to self.basis_root_rp(RRRR(RR((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pytget_rp_from_rootRscCs(|i|}|i o |i S(s0True if diff_rorp and mirror are not directories(RR(RRt diff_rorpR[((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pytcan_fast_processXscCsE|i|}ti|}|i|||ti||dS(s>Patch base_rp with diff_rorp (case where neither is directory)N(RtTempFileR-t patch_to_tempR trename(RRRR[ttf((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyt fast_process]scCs|ipdSn|iptid|idnm|i|ijo*tid|i|ifdn-tid|i|i|ifddS(s8Check the hash in the copy_report with hash in diff_rorpNs!Hash for %s missing, cannot checkisHash %s of %s verifiedis6Warning: Hash %s of %s doesn't match recorded hash %s!(Rthas_sha1RIRJR6t sha1_digesttget_sha1(Rt copy_reportR((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyt check_hashds    ! cCs|ioti|||idSn|idjoti||}n0|idjptti |||}|i |||i oti ||ndS(s>Patch basis_rp, writing output in new, which doesn't exist yetNRR( t isflaglinkedRtlink_rpRtget_attached_filetypeR tcopyR$Rt patch_localR Rt copy_attribs(Rtbasis_rpRR-R ((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRrs  cCsw|i|}|_|ip|ip|i pt|io|i||n|i||dS(s9Start processing directory - record information for laterN(Rtbase_rpRRR$t prepare_dirtset_dir_replacement(RRRR((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyt start_processs , cCsa|idjptti||_ti||i|io|idndS(sSet self.dir_replacement, which holds data until done with dir This is used when base_rp is a dir, and diff_rorp is not. RiN( RR$RR-RR tcopy_with_attribsRtchmod(RRR((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRs  cCsV|i|_|ip)|io|in|in|iddS(s(Prepare base_rp to turn into a directoryiN(RRRRtdeletetmkdirR(RRR((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRs   cCs|io1|iiptti|i|inI|ipt|ii|iioti |i|indS(sFinish processing directoryN( RRRR$R RRtrmdirRR(R((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyt end_processs  ( RRRRRRRR RRRRR(((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR>s        RcBsGeZdZdZddZdZdZdZdZ RS(s"Change the permission of mirror files and directories The problem is that mirror files and directories may need their permissions changed in order to be read and listed, and then changed back when we are done. This class hooks into the CachedRF object to know when an rp is needed. cCs||_d|_g|_dS(N((troot_rpt current_indextopen_index_list(RR((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRs  cCsl|o|iodSn|i}||_| p ||jodSn|i||i||dS(s9Given rpath, change permissions up to and including indexN(RR t restore_oldtadd_new(RRRt old_index((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyt__call__s   cCs_xX|ioM|id\}}}|t| |jo|i|nP|id=qWdS(s1Restore permissions for indicies we are done withiN(R!RLR(RRR$told_rpt old_perms((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR"s cCsx|i||D]}|io|i p(|io}|io |i ob|i}|iid|i||f|io|i d|Bq|i d|BqqWdS(s=Change permissions of directories between old_index and indexiiiN( tget_new_rp_listRtreadableRt executabletgetpermsR!tinsertRR(RR$RR[R'((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR#s(  ccsxRtt|dddD]&}|| || jo |}PqqWdptx:t|dt|dD]}|ii|| VqsWdS(sReturn list of new rp's between old_index and index Do this lazily so that the permissions on the outer directories are fixed before we need the inner dirs. iiiN(trangeRLR$RR(RR$RRtcommon_prefix_lent total_len((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyR(s cCs.x'|iD]\}}}|i|q WdS(sRestore any remaining rpsN(R!R(RRR[tperms((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRs N( RRRRRXR%R"R#R(R(((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyRs   (((((($Rt __future__RRtosRtstaticR4Rt ExceptionRRR,R@RCRt MakeClassRRbR^t ITRBranchRRRtTimeRRRlR RIRRft statisticsRRR(((s9/var/lib/python-support/python2.5/rdiff_backup/restore.pyss"$$     Og@