q&fndZddlmZddlmZddlmZddlmZddl m Z Gdd e Z Gd d Z y ) zCbase class for store implementations and store-related utility code)_)short)util)urlutil)lfutilc"eZdZdZdZdZdZy) StoreErrorzZRaised when there is a problem getting files from or putting files to a central store.c<||_||_||_||_yN)filenamehashurldetail)selfr rrrs eZdZdZdZdZdZdZd dZdZ dZ y ) basestorec.||_||_||_yr )uirepor)rr#r$rs rrzbasestore.__init__-s rctd)z>Put source file into the store so it can be retrieved by hash.abstract methodNotImplementedError)rsourcers rputz basestore.put2s!"455rctd)zCheck to see if the store contains the given hashes. Given an iterable of hashes it returns a mapping from hash to bool.r&r')rhashess rexistszbasestore.exists6s""455rc g}g}|j}d}|j|Dchc]\}}| c}}}|jtdtdt |5} |D]\} }| j ||dz }|j td| |fz|j|sO|jtd| |tj|jfz|j| |j| |r|j| |f|j|  ddd||fScc}}w#1swY||fSxYw) aGet the specified largefiles from the store and write to local files under repo.root. files is a list of (filename, hash) tuples. Return (success, missing), lists of files successfully downloaded and those not found in the store. success is a list of (filename, hash) tuples; missing is a list of filenames that we could not get. (The detailed error message will already have been presented to the user, so missing is just supplied as a summary.)rsgetting largefilessfiles)unittotalrsgetting %s:%s s'%s: largefile %s not available from %s N)r#r- makeprogressrlenupdatenotegetwarnrrrappend_gethash) rfilessuccessmissingr#at _filenamer availableprogressr s rr5z basestore.get;sS WW KKu E*;9d EF __ # $1X;c%j  - "' -$#a,-40@@A }}T*GGEF#T7+?+?+IJKNN8,==40NNHd#34NN8,! - -*!!-!F -*!!s E &CEEctjtj|jdtj|j|}|dz}tj ||jj j5} |j|||}ddd|k7rE|dk7r*|jjtd|||fztj|ytj||tj |j|y#t$r5}|jj|jd}Yd}~d}~wwxYw#1swYxYw)zGet file with the provided hash and store it in the local repo's store and in the usercache. filename is for informational messages only. rs.tmp) createmodeNs*%s: data corruption (expected %s, got %s) FT)rmakedirsr storepathr$atomictempfilestorerA_getfiler r#r6rrunlinkrenamelinktousercache)rr r storefilenametmpnametmpfilegothasherrs rr8zbasestore._gethashas- f&&tyy#67((D9 ')    : :    --4@   d?#~ DEw/0 KK  G]+tyy$/  S__./    s0E-D,, E*5+E% E-%E**E--E6c>|jjtdt|zt }g}|D]}|j |}d|j t|jfz}|D]p}tj|} | s||} | | jf} | |vs7|j| tj| } |j|| | fr|j||} t|}t|Dchc]\}}| c}}}|r4|jjtd||fzt!| S|jjtd||fzt!| Scc}}w)zVerify the existence (and, optionally, contents) of every big file revision referenced by every changeset in revs. Return 0 if all is well, non-zero on any errors.s'searching %d changesets for largefiles s%d:%ss3verified contents of %d revisions of %d largefiles s4verified existence of %d revisions of %d largefiles )r#statusrr2setr$revrnoder splitstandinfilenodeadd readasstandinr7 _verifyfilesint)rrevscontentsverified filestocheckrRcctxcsetstandinr fctxkey expectedhashfailednumrevsfnamefnode numlfiless rverifyzbasestore.verifys  9 :SY F 5  LC99S>Dtxxz5+=>>D L!..w7=D#T]]_5C(* S)'-';';D'A $++T8\,JK L L""8\:h-X>>E5>?  GGNNIJI&' 6{ GGNNJKI&' 6{?s F ctd)aFetch one revision of one file from the store and write it to tmpfile. Compute the hash of the file on-the-fly as it downloads and return the hash. Close tmpfile. Raise StoreError if unable to download the file (e.g. it does not exist in the store).r&r')rrLr rs rrFzbasestore._getfile ""455rctd)zPerform the actual verification of files in the store. 'contents' controls verification of content hash. 'filestocheck' is list of files to check. Returns _true_ if any problems are found! r&r')rr[r]s rrXzbasestore._verifyfilesrkrN)F) rrrrr*r-r5r8rirFrXrrrr!r!,s, 66 $"L>&P66rr!N)rmercurial.i18nrmercurial.noder mercurialrmercurial.utilsrr Exceptionr r!rrrrss8J II,J6J6r