a mb#@s`dZddlmZddlmZGdddeZGdddeZGdd d eZGd d d eZ d S) z# Classes for managing Checkpoints. ) HTTPError)LoggingConfigurablec@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS) Checkpointsaq Base class for managing checkpoints for a ContentsManager. Subclasses are required to implement: create_checkpoint(self, contents_mgr, path) restore_checkpoint(self, contents_mgr, checkpoint_id, path) rename_checkpoint(self, checkpoint_id, old_path, new_path) delete_checkpoint(self, checkpoint_id, path) list_checkpoints(self, path) cCs tddSzCreate a checkpoint.!must be implemented in a subclassNNotImplementedErrorself contents_mgrpathr Klib/python3.9/site-packages/jupyter_server/services/contents/checkpoints.pycreate_checkpointszCheckpoints.create_checkpointcCs tddSzRestore a checkpointrNrr r checkpoint_idr r r rrestore_checkpointszCheckpoints.restore_checkpointcCs tddSz5Rename a single checkpoint from old_path to new_path.rNrr rold_pathnew_pathr r rrename_checkpointszCheckpoints.rename_checkpointcCs tddSzdelete a checkpoint for a filerNrr rr r r rdelete_checkpoint#szCheckpoints.delete_checkpointcCs tddSz-Return a list of checkpoints for a given filerNrr r r r rlist_checkpoints'szCheckpoints.list_checkpointscCs&||D]}||d||q dS)0Rename all checkpoints for old_path to new_path.idNrrr rrZcpr r rrename_all_checkpoints+sz"Checkpoints.rename_all_checkpointscCs$||D]}||d|q dS)*Delete all checkpoints for the given path.r Nrrr r Z checkpointr r rdelete_all_checkpoints0sz"Checkpoints.delete_all_checkpointsN __name__ __module__ __qualname____doc__rrrrrr#r'r r r rr s rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)GenericCheckpointsMixina Helper for creating Checkpoints subclasses that can be used with any ContentsManager. Provides a ContentsManager-agnostic implementation of `create_checkpoint` and `restore_checkpoint` in terms of the following operations: - create_file_checkpoint(self, content, format, path) - create_notebook_checkpoint(self, nb, path) - get_file_checkpoint(self, checkpoint_id, path) - get_notebook_checkpoint(self, checkpoint_id, path) To create a generic CheckpointManager, add this mixin to a class that implement the above four methods plus the remaining Checkpoints API methods: - delete_checkpoint(self, checkpoint_id, path) - list_checkpoints(self, path) - rename_checkpoint(self, checkpoint_id, old_path, new_path) cCs^|j|dd}|d}|dkr.||d|S|dkrL||d|d|Stdd |dS NTcontenttypenotebookr0fileformatUnexpected type %sgetcreate_notebook_checkpointcreate_file_checkpointrr r r modelr1r r rrLsz)GenericCheckpointsMixin.create_checkpointcCs\|j|ddd}|dkr(|||}n$|dkr>|||}ntdd||||dS) Restore a checkpoint.Fr/r1r2r3r5r6Nr8get_notebook_checkpointget_file_checkpointrZsaver r rr r1r<r r rr]sz*GenericCheckpointsMixin.restore_checkpointcCs tddSzwCreate a checkpoint of the current state of a file Returns a checkpoint model for the new checkpoint. rNrr r0r4r r r rr:isz.GenericCheckpointsMixin.create_file_checkpointcCs tddSrBrr nbr r r rr9psz2GenericCheckpointsMixin.create_notebook_checkpointcCs tddSzGet the content of a checkpoint for a non-notebook file. Returns a dict of the form: { 'type': 'file', 'content': , 'format': {'text','base64'}, } rNrrr r rr@ws z+GenericCheckpointsMixin.get_file_checkpointcCs tddSzGet the content of a checkpoint for a notebook. Returns a dict of the form: { 'type': 'notebook', 'content': , } rNrrr r rr?s z/GenericCheckpointsMixin.get_notebook_checkpointN r)r*r+r,rrr:r9r@r?r r r rr-6s  r-c@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)AsyncCheckpointszS Base class for managing checkpoints for a ContentsManager asynchronously. cs tddSrrr r r rrsz"AsyncCheckpoints.create_checkpointcs tddSrrrr r rrsz#AsyncCheckpoints.restore_checkpointcs tddSrrrr r rrsz"AsyncCheckpoints.rename_checkpointcs tddSrrrr r rrsz"AsyncCheckpoints.delete_checkpointcs tddSrrrr r rrsz!AsyncCheckpoints.list_checkpointscs2||IdHD]}||d||IdHqdS)rNr r!r"r r rr#sz'AsyncCheckpoints.rename_all_checkpointscs0||IdHD]}||d|IdHqdS)r$Nr r%r&r r rr'sz'AsyncCheckpoints.delete_all_checkpointsNr(r r r rrIsrIc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)AsyncGenericCheckpointsMixinzp Helper for creating Asynchronous Checkpoints subclasses that can be used with any ContentsManager. csp|j|ddIdH}|d}|dkr:||d|IdHS|dkr^||d|d|IdHStdd |dSr.r7r;r r rrs  z.AsyncGenericCheckpointsMixin.create_checkpointcst|j|dddIdH}|dkr4|||IdH}n*|dkrP|||IdH}ntdd||||IdHdS) r=Fr/r1Nr2r3r5r6r>rAr r rrsz/AsyncGenericCheckpointsMixin.restore_checkpointcs tddSrBrrCr r rr:sz3AsyncGenericCheckpointsMixin.create_file_checkpointcs tddSrBrrDr r rr9sz7AsyncGenericCheckpointsMixin.create_notebook_checkpointcs tddSrFrrr r rr@s z0AsyncGenericCheckpointsMixin.get_file_checkpointcs tddSrGrrr r rr?s z4AsyncGenericCheckpointsMixin.get_notebook_checkpointNrHr r r rrJs  rJN) r,Z tornado.webrZtraitlets.config.configurablerrobjectr-rIrJr r r rs   ,Y$