³ò OCÖIc @sÕdZddkZddkZddkZddkZddkZddkZddkZddkZe dƒ\Z Z Z Z Zdadd d„ƒYZdd d„ƒYZd„Zd „Zd „Zd „ZdS(s9archive.py - Archive handling (extract/create) for Comix.iÿÿÿÿNit ExtractorcBsqeZdZd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z RS( sÂExtractor is a threaded class for extracting different archive formats. The Extractor can be loaded with paths to archives (currently ZIP, tar, or RAR archives) and a path to a destination directory. Once an archive has been set it is possible to filter out the files to be extracted and set the order in which they should be extracted. The extraction can then be started in a new thread in which files are extracted one by one, and a signal is sent on a condition after each extraction, so that it is possible for other threads to wait on specific files to be ready. Note: Support for gzip/bzip2 compressed tar archives is limited, see set_files() for more info. cCs t|_dS(N(tFalset _setupped(tself((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyt__init__#scCs÷||_||_t|ƒ|_g|_h|_t|_d |_ t i ƒ|_ |it jo+ti|dƒ|_|iiƒ|_nY|itttfjo+ti|dƒ|_|iiƒ|_n|itjo÷td joutƒatd jo[dGHtid dtitit dƒƒ}|i!t dƒƒ|i"ƒ|i#ƒd Sqent$i%tdd|gƒ}|i&ƒ}g}|i'ƒD]}||i(t)i*ƒq~|_|i+ƒ|i,ƒndG|GHd St-|_.|i S( sÉSetup the extractor with archive and destination dir . Return a threading.Condition related to the is_ready() method, or None if the format of isn't supported. trs$! Could not find RAR file extractor.is"Could not find RAR file extractor!sgYou need either the rar or the unrar program installed in order to read RAR (.cbr) files.tvbs--s! Non-supported archive format:N(/t_srct_dsttarchive_mime_typet_typet_filest _extractedRt_stoptNonet_extract_threadt threadingt Conditiont _conditiontZIPtzipfiletZipFilet_zfiletnamelisttTARtGZIPtBZIP2ttarfiletopent_tfiletgetnamestRARt _rar_exect _get_rar_exectgtkt MessageDialogtMESSAGE_WARNINGt BUTTONS_CLOSEt_tformat_secondary_markuptruntdestroytprocesstProcesstspawnt readlinestrstriptostlineseptclosetwaittTrueR(Rtsrctdsttdialogtproctfdt_[1]tname((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pytsetup&sH              6   cCs|iS(s,Return a list of names of all the files the extractor is currently set for extracting. After a call to setup() this is by default all files found in the archive. The paths in the list are relative to the archive root and are not absolute for the files once extracted. (R (R((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyt get_filesTscCs_|ittfjo<g}|iD]}||jo ||q$q$~|_n ||_dS(sySet the files that the extractor should extract from the archive in the order of extraction. Normally one would get the list of all files in the archive using get_files(), then filter and/or permute this list before sending it back using set_files(). Note: Random access on gzip or bzip2 compressed tar archives is no good idea. These formats are supported *only* for backwards compability. They are fine formats for some purposes, but should not be used for scanned comic books. So, we cheat and ignore the ordering applied with this method on such archives. N(R RRR (RtfilesR9tx((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyt set_files\s <cCs|ii|tƒS(sxReturn True if the file in the extractor's file list (as set by set_files()) is fully extracted. (R tgetR(RR:((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pytis_readymscCs|iS(s=Return the mime type name of the extractor's current archive.(R (R((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyt get_mime_typesscCs1t|_|io|iiƒt|_ndS(s‹Signal the extractor to stop extracting and kill the extracting thread. Blocks until the extracting thread has terminated. N(R3R RRtjoinRtsetupped(R((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pytstopws   cCs9tid|iƒ|_|iitƒ|iiƒdS(sÒStart extracting the files in the file list one by one using a new thread. Every time a new file is extracted a notify() will be signalled on the Condition that was returned by setup(). ttargetN(RtThreadt_thread_extractRt setDaemonRtstart(R((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pytextract€scCsO|itjo|iiƒn+|itttfjo|iiƒndS(spClose any open file objects, need only be called manually if the extract() method isn't called. N(R RRR1RRRR(R((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyR1‰scCs/x|iD]}|i|ƒq W|iƒdS(s.Extract the files in the file list one by one.N(R t _extract_fileR1(RR:((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyRH’s cCs÷|io|iƒtidƒny„|itjoŠtii|i |ƒ}tii tii |ƒƒpti tii |ƒƒnt |dƒ}|i|ii|ƒƒ|iƒnç|itttfjo\tiitii|i |ƒƒi|i ƒo|ii||i ƒq¨dG|GdGHnr|itjoatd j oKtitddddd d |i||i g ƒ}|iƒ|iƒq¨d GHnWnt j onX|i!i"ƒt#|i$|<|i!i%ƒ|i!i&ƒd S( s«Extract the file named to the destination directory, mark the file as "ready", then signal a notify() on the Condition returned by setup(). itwbs! Non-local tar member:s R>s-kbs-p-s-o-s-inuls--s$! Could not find RAR file extractor.N('R R1tsystexitR RR/tpathRCRtexiststdirnametmakedirsRtwriteRtreadRRRtnormpatht startswithRRKRR RR*R+RR,R2t ExceptionRtacquireR3R tnotifytrelease(RR:tdst_pathtnewR7((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyRL˜s:  $  !     (t__name__t __module__t__doc__RR;R<R?RARBRERKR1RHRL(((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyRs  .     tPackercBs2eZdZd„Zd„Zd„Zd„ZRS(s Packer is a threaded class for packing files into ZIP archives. It would be straight-forward to add support for more archive types, but basically all other types are less well fitted for this particular task than ZIP archives are (yes, really). cCs:||_||_||_||_d|_t|_dS(smSetup a Packer object to create a ZIP archive at . All files pointed to by paths in the sequences and will be included in the archive when packed. The files in will be renamed on the form "NN - .ext", so that the lexical ordering of their filenames match that of their order in the list. The files in will be included as they are, assuming their filenames does not clash with other filenames in the archive. All files are placed in the archive root. N(t _image_filest _other_filest _archive_patht _base_nameRt _pack_threadRt_packing_successful(Rt image_filest other_filest archive_patht base_name((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyRËs     cCs9tid|iƒ|_|iitƒ|iiƒdS(s6Pack all the files in the file lists into the archive.RFN(RRGt _thread_packRfRIRRJ(R((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pytpackßscCs(|idjo|iiƒn|iS(svBlock until the packer thread has finished. Return True if the packer finished its work successfully. N(RfRRCRg(R((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyR2åsc Csyti|idƒ}Wn#tj odG|iGHdSnXg}dttt|iƒƒƒ|if}x¼t|iƒD]«\}}||dt i i |ƒdf}y|i ||ti ƒWnQtj oEd||ifGH|iƒyt i|iƒWnnXdSnX|i|ƒq}Wx½|iD]²}t i i|ƒ}x||jod|}qQWy|i ||tiƒWnQtj oEd||ifGH|iƒyt i|iƒWnnXdSnX|i|ƒq6W|iƒt|_dS(Ntws! Could not create archives%%0%dd - %s%%sis1! Could not add file %s to add to %s, aborting...s_%s(RRRdRXtlentstrRbRet enumerateR/RPtsplitextRTt ZIP_STOREDR1tremovetappendRctbasenamet ZIP_DEFLATEDR3Rg(Rtzfilet used_namestpatterntiRPtfilename((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyRlísV    $       (R^R_R`RRmR2Rl(((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyRaÂs    cCsyïtii|ƒoØti|tiƒpdSnti|ƒotSnt |dƒ}|i dƒ}|i ƒt i |ƒoQtii|ƒdjo8|idƒotSn|idƒotSntSn|djotSqînWntj odG|GHnXdS( s;Return the archive type of or None for non-archives.trbiitBZhs‹sRar!s! Error while readingN(R/RPtisfiletaccesstR_OKRRt is_zipfileRRRUR1Rt is_tarfiletgetsizeRWRRRRRX(RPR8tmagic((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyR s( ) cCsShtdƒt<tdƒt<tdƒt<tdƒt<tdƒt<|S(s0Return a text representation of an archive type.s ZIP archives Tar archivesGzip compressed tar archivesBzip2 compressed tar archives RAR archive(R&RRRRR(t archive_type((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pytget_name0s cCsœtidtiƒ}tƒ}|i|dƒ|iƒ}|djodSn|iƒ}|iƒt t |i |ƒƒ}t i |ƒi}|||fS(sŽReturn a tuple (mime, num_pages, size) with info about the archive at , or None if doesn't point to a supported archive. s!\.(jpg|jpeg|png|gif|tif|tiff)\s*$N(tretcompiletIRR;RRBR<R1RotfiltertsearchR/tstattst_size(RPtimage_ret extractortmimeR=t num_pagestsize((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pytget_archive_info9s     cCs<x5dD]-}ti|gƒiƒdj o|SqqWdS(sfReturn the name of the RAR file extractor executable, or None if no such executable is found. tunrartrar(R•R–N(R*R+R,R(tcommand((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyR!Js  (((R`RNR/RˆRRRR"R*trangeRRRRRRR RRaR R‡R”R!(((sG/home/sork/Pontus/Programmering/Python/Comix/comix-4.0.4/src/archive.pyss         ¯V