a VŸ$c"9ã@sddlmZmZmZmZmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZddlZddlZddlZddlZddlZddlmZmZddlZe ¡ZddlZdeee ¡ƒƒdZ Gdd„de!ƒZ"Gd d „d e!ƒZ#da$d a%e &¡Z'd d „Z(dd„Z)dd„Z*dS)é)ÚdivisionÚabsolute_importÚwith_statementÚprint_functionÚunicode_literals) ÚPY2Ú basestringÚbchrÚbordÚchrÚopenÚpystrÚrangeÚroundÚstrÚtobytesÚunicodeN)Ú clear_slotÚ safe_renameÚ.z.tmpc@s eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&S)'Ú FileLocationz= A location that saves files to a directory on disk. cCsÒ||_zt |j¡Wnty(Yn0tj |j¡zXtj |jd¡}t |dƒ}|  d¡Wdƒn1st0Yt  |¡d|_ Wnty¨d|_ Yn0i|_ tj |jd¡|_d|_d|_dS)Nztext.txtÚwzTest.TFÚ persistentr)Ú directoryÚosÚmakedirsÚ ExceptionÚrenpyÚutilÚexpose_directoryÚpathÚjoinr ÚwriteÚunlinkÚactiveÚmtimesrÚpersistent_mtimeÚpersistent_data)ÚselfrÚfnÚf©r+úrenpy/savelocation.pyÚ__init__7s$  (    zFileLocation.__init__cCstj |jtj |tj¡¡S)z8 Given a slot name, returns a filename. )rr r!rrÚexportsÚfsencodeÚsavegame_suffix©r(Úslotnamer+r+r,Úfilename[szFileLocation.filenamecCstjrddl}| ¡dS)zJ Called to indicate that the HOME filesystem was changed. rN)rÚ emscriptenÚsyncfs)r(r4r+r+r,ÚsyncbszFileLocation.syncc CsV|js dSt0|j}i}tj}t|ƒ}t |j¡D]R}|  |¡sFq6|d| …}z tj   tj   |j|¡¡||<Wq6t y†Yq60q6||_| ¡D] \}}| |d¡|kr˜t|ƒq˜|D]}||vr¾t|ƒq¾|jd|jfD]L}tj  |¡rätj   |¡}||jkrätj |¡} | durä||_| |_q2qäWdƒn1sH0YdS)z; Scan for files that are added or removed. Nú.new)r$Ú disk_lockr%rr0ÚlenrÚlistdirrÚendswithr Úgetmtimer!rÚitemsÚgetrrÚexistsr&Úloadr') r(Ú old_mtimesÚ new_mtimesÚsuffixÚ suffix_lenr)r2ÚmtimeÚpfnÚdatar+r+r,Úscanks>         zFileLocation.scancCsX| |¡}t| |¡Wdƒn1s.0Ytj |¡| ¡| ¡dS)z4 Saves the save record in slotname. N)r3r8Ú write_filerrÚ expose_filer6rH)r(r2Úrecordr3r+r+r,Úsave›s  ( zFileLocation.savecCs t|jƒS)za Returns a list of all slots with savefiles in them, in arbitrary order. )Úlistr%©r(r+r+r,rMªszFileLocation.listcCs|j |d¡S)z… For a slot, returns the time the object was saved in that slot. Returns None if the slot is empty. N)r%r>r1r+r+r,rE²szFileLocation.mtimec Cs6tzæ| |¡}t |d¡¼}z8| d¡}t |¡}|WWdƒWWdƒStylYn0z8| d¡ d¡}d|iWWdƒWWdƒSty¸Yn0iWdƒWWdƒS1sâ0YWn"tyYWdƒdS0Wdƒn1s(0YdS)za Returns the JSON data for slotname. Returns None if the slot is empty. ÚrÚjsonNÚ extra_infozutf-8Ú _save_name) r8r3ÚzipfileÚZipFileÚreadrPÚloadsrÚdecode)r(r2r3ÚzfrGrQr+r+r,rP¼s$   $ ( 6zFileLocation.jsonc Cs tð| |¡}|dur*WdƒdSzn| |¡}t |d¡D}zd}| d¡Wn tyvd}| d¡Yn0Wdƒn1sŒ0YWn ty¸YWdƒdS0|rÒtjj   |d|¡}ntjj   |d|¡}|WdƒS1sþ0YdS)z{ Returns a displayable that show the screenshot for this slot. Returns None if the slot is empty. NrOFzscreenshot.tgaTzscreenshot.png) r8rEr3rSrTÚgetinforrÚdisplayÚimÚ ZipFileImage)r(r2rEr3rXÚpngÚ screenshotr+r+r,r^Ùs&   2 zFileLocation.screenshotc CsltT| |¡}t |d¡}| d¡}Wdƒn1s<0Y|WdƒS1s^0YdS)zi Returns the log component of the file found in `slotname`, so it can be loaded. rOÚlogN)r8r3rSrTrU)r(r2r3rXÚrvr+r+r,r@ús  (zFileLocation.loadcCsXt@| |¡}tj |¡r&t |¡| ¡| ¡Wdƒn1sJ0YdS)z/ Deletes the file in slotname. N)r8r3rr r?r#r6rH)r(r2r3r+r+r,r# s    zFileLocation.unlinkcCstx| |¡}| |¡}tj |¡s6WdƒdS|t}t||ƒt||ƒtj  |¡|  ¡|  ¡Wdƒn1s‚0YdS)z3 If old exists, renames it to new. N) r8r3rr r?ÚtmprrrrJr6rH)r(ÚoldÚnewÚold_tmpr+r+r,Úrenames      zFileLocation.renamecCs€th| |¡}| |¡}tj |¡s6WdƒdSt ||¡tj  |¡|  ¡|  ¡Wdƒn1sr0YdS)z9 Copies `old` to `new`, if `old` exists. N) r8r3rr r?ÚshutilÚcopyfilerrrJr6rH)r(rbrcr+r+r,Úcopy,s     zFileLocation.copycCs|jr|j|jfgSgSdS)z½ Returns a list of (mtime, persistent) tuples loaded from the persistent file. This should return quickly, with the actual load occuring in the scan thread. N)r'r&rNr+r+r,Úload_persistent>szFileLocation.load_persistentc Cs¾t¦|jsWdƒdS|j}|t}|d}t|dƒ}| |¡Wdƒn1s\0Yt||ƒt||ƒtj  |¡|_ t j   |¡| ¡Wdƒn1s°0YdS)zƒ Saves `data` as the persistent data. Data is a binary string giving the persistent data in python format. Nr7Úwb)r8r$rrar r"rrr r<r&rrrJr6)r(rGr)Úfn_tmpÚfn_newr*r+r+r,Úsave_persistentJs (   zFileLocation.save_persistentcCs:|js dSzt |j¡| ¡Wnty4Yn0dS©N)r$rr#rr6rrNr+r+r,Úunlink_persistentfs   zFileLocation.unlink_persistentcCst|tƒsdS|j|jkS©NF)Ú isinstancerr©r(Úotherr+r+r,Ú__eq__rs zFileLocation.__eq__cCs ||k Srnr+rrr+r+r,Ú__ne__xszFileLocation.__ne__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__r-r3r6rHrLrMrErPr^r@r#rerhrirmrortrur+r+r+r,r2s&$ 0 !  rc@s¨eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(S))Ú MultiLocationzš A location that saves in multiple places. When loading or otherwise accessing a file, it loads the newest file found for the given slotname. cCs g|_dSrn©Ú locationsrNr+r+r,r-‚szMultiLocation.__init__cCsdd„|jDƒS)NcSsg|]}|jr|‘qSr+)r$)Ú.0Úir+r+r,Ú †óz2MultiLocation.active_locations..r{rNr+r+r,Úactive_locations…szMultiLocation.active_locationscCsBd}d}|jD].}|jsq| |¡}|dur||kr|}|}q|S)z€ Returns the location containing the slotname with the newest mtime. Returns None if the slot is empty. éÿÿÿÿN)r|r$rE)r(r2rEÚlocationÚlÚ slot_mtimer+r+r,Únewestˆs  zMultiLocation.newestcCs||jvrdS|j |¡dS)z& Adds a new location. N)r|Úappend)r(rƒr+r+r,Úaddžs zMultiLocation.addcCs2d}| ¡D]}| ||¡d}q |s.tdƒ‚dS)NFTz$Not saved - no valid save locations.)rrLr)r(r2rKÚsavedr„r+r+r,rL¨s   zMultiLocation.savecCs*tƒ}| ¡D]}| | ¡¡qt|ƒSrn)ÚsetrÚupdaterM©r(r`r„r+r+r,rM³s zMultiLocation.listcCs | |¡}|durdS| |¡Srn)r†rE©r(r2r„r+r+r,rE»s zMultiLocation.mtimecCs | |¡}|durdS| |¡Srn)r†rPrr+r+r,rPÃs zMultiLocation.jsoncCs | |¡}|durdS| |¡Srn)r†r^rr+r+r,r^Ës zMultiLocation.screenshotcCs| |¡}| |¡Srn)r†r@rr+r+r,r@Ós zMultiLocation.loadcCs| ¡D]}| |¡qdSrn)rr#rr+r+r,r#×s zMultiLocation.unlinkcCs| ¡D]}| ||¡qdSrn)rre©r(rbrcr„r+r+r,reÛs zMultiLocation.renamecCs| ¡D]}| ||¡qdSrn)rrhrŽr+r+r,rhßs zMultiLocation.copycCs$g}| ¡D]}| | ¡¡q |Srn)rÚextendrirŒr+r+r,riãs zMultiLocation.load_persistentcCs| ¡D]}| |¡qdSrn)rrm)r(rGr„r+r+r,rmës zMultiLocation.save_persistentcCs| ¡D] }| ¡qdSrn)rro©r(r„r+r+r,roðs zMultiLocation.unlink_persistentcCs|jD] }| ¡qdSrn)r|rHrr+r+r,rHõs zMultiLocation.scancCst|tƒsdS|j|jkSrp)rqrzr|rrr+r+r,rtüs zMultiLocation.__eq__cCs ||k Srnr+rrr+r+r,ruszMultiLocation.__ne__N)rvrwrxryr-rr†rˆrLrMrErPr^r@r#rerhrirmrorHrtrur+r+r+r,rz|s(  rzFcCs`dats\ztjj ¡Wnty*Yn0tt d¡Wdƒq1sP0YqdS)NFg@)Úquit_scan_threadrÚloadsaverƒrHrÚscan_thread_conditionÚwaitr+r+r+r,Úrun_scan_threads r•cCsDtdat ¡Wdƒn1s&0Ytdur@t ¡dS)NT)r“r‘Ú notify_allÚ scan_threadr!r+r+r+r,Úquit s &r˜cCsltƒ}| ttjjƒ¡tjsDtjsDtj   tjj d¡}| t|ƒ¡|  ¡|tj _tjtdat ¡dS)NÚsaves)Útarget)rzrˆrrÚconfigÚsavedirÚmobileÚmacapprr r!ÚgamedirrHr’rƒÚ threadingÚThreadr•r—Ústart)rƒr r+r+r,Úinit+s  r£)+Ú __future__rrrrrÚ renpy.compatrrr r r r r rrrrrrrSrPrr Úrenpy.loadsaverrrfÚRLockr8ÚtimeÚintraÚobjectrrzr—r‘Ú Conditionr“r•r˜r£r+r+r+r,Ús,8L