U j@sdZddlZddlZddlZddlmZmZddlmZddl m Z m Z GdddeZ ee Gdd d Z e ZGd d d ejZd d ZddZddZdd d dddgZdS)z` Save and load Small OBjects to and from files, using various formats. Maintainer: Moshe Zadka N) Interface implementer)styles)logruntimec@s"eZdZdZddZdddZdS) IPersistablez9An object which can be saved in several formats to a filecCsdSzWSet desired format. @type style: string (one of 'pickle' or 'source') Nstyler r 9/tmp/pip-unpacked-wheel-j5h_c8_q/twisted/persisted/sob.pysetStyleszIPersistable.setStyleNcCsdS)xSave object to file. @type tag: string @type filename: string @type passphrase: string Nr )tagfilename passphraser r r save!szIPersistable.save)NNN)__name__ __module__ __qualname____doc__r rr r r r rsrc@sBeZdZdZddZddZddZdd Zd d Zdd dZ d S) PersistentpicklecCs||_||_dS)N)originalname)selfrrr r r __init__.szPersistent.__init__cCs ||_dSrr )rr r r r r 2szPersistent.setStylecCsl|r|}|d}nR|rD|jd|d|}|jd|d|}n |jd|}|jd|}||fS)Nz-2-z-2..)r)rrextr finalnamer r r _getFilename9s zPersistent._getFilenamec Cs&t|d}||j|W5QRXdS)Nwb)openr)rrdumpFuncfr r r _saveTempEs zPersistent._saveTempcCs2|jdkrddlm}d}nddd}d}||fS) Nsourcer) jellyToSourceZtascSst||ddS)N)rdump)objfiler r r r$Psz&Persistent._getStyle..dumpFuncZtap)N)r twisted.persisted.aotr()rr$rr r r _getStyleIs    zPersistent._getStyleNcCs|\}}|dk rtd||||\}}td|jd|d|||tjdkrvt j |rvt |t ||tddS)rNzpassphrase must be NonezSaving z application to z...win32zSaved.)r. TypeErrorr!rmsgrr&rZ platformTypeospathisfileremoverename)rrrrrr$r r r r rVs    zPersistent.save)NNN) rrrr rr r!r&r.rr r r r r*s  rc@s eZdZdZddZddZdS)_EverythingEphemeralrcCs ||_dS)zS @param mainMod: The '__main__' module that this class will proxy. N)mainMod)rr8r r r rpsz_EverythingEphemeral.__init__cCsLzt|j|WStk rF|jr(ntd|tYSYnXdS)Nz#Warning! Loading from __main__: %s)getattrr8AttributeErrorinitRunrr1r Ephemeral)rkeyr r r __getattr__vsz _EverythingEphemeral.__getattr__N)rrrr;rr>r r r r r7msr7c Csd}|dkrddlm}n tjd}}t||}ttjd}|tjd<d|_|"z ||}W5|j tjd<XW5QRXt d|_t |d}|dk r| ||S) zLoad an object from a file. Deserialize an object from a file. The file can be encrypted. @param filename: string @param style: string (one of 'pickle' or 'source') rr'r)unjellyFromSourcerb__main__N)r-r@rloadr#r7sysmodulesr;r8rZ doUpgraderr )rr mode_loadfpeevalueZ persistabler r r rDs$      rDc CsHt|}|}W5QRXd|i}t||d}t|||||}|S)zLoad the value of a variable in a Python file. Run the contents of the file in a namespace and return the result of the variable named C{variable}. @param filename: string @param variable: string __file__exec)r#readcompileeval)rvariableZfileObjdatadZcodeObjrKr r r loadValueFromFiles   rTcCs(tj|d}dddddddd|S)NrCpythonrr')z.tacz.etacz.pyz.tapz.etapz.tasz.etas)r2r3splitext)rrr r r guessTypesrW Persistant)rr2rrEZzope.interfacerrZtwisted.persistedrZtwisted.pythonrrrrrXr<r7rDrTrW__all__r r r r s* ?!