B 3\@s@dZddlmZddlmZGdddeZGdddeZdS) 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 tddS)zCreate a checkpoint.z!must be implemented in a subclassN)NotImplementedError)self contents_mgrpathr Elib/python3.7/site-packages/notebook/services/contents/checkpoints.pycreate_checkpointszCheckpoints.create_checkpointcCs tddS)zRestore a checkpointz!must be implemented in a subclassN)r)rr checkpoint_idrr r r restore_checkpointszCheckpoints.restore_checkpointcCs tddS)z5Rename a single checkpoint from old_path to new_path.z!must be implemented in a subclassN)r)rr old_pathnew_pathr r r rename_checkpoint!szCheckpoints.rename_checkpointcCs tddS)zdelete a checkpoint for a filez!must be implemented in a subclassN)r)rr rr r r delete_checkpoint%szCheckpoints.delete_checkpointcCs tddS)z-Return a list of checkpoints for a given filez!must be implemented in a subclassN)r)rrr r r list_checkpoints)szCheckpoints.list_checkpointscCs*x$||D]}||d||q WdS)z0Rename all checkpoints for old_path to new_path.idN)rr)rrrZcpr r r rename_all_checkpoints-sz"Checkpoints.rename_all_checkpointscCs(x"||D]}||d|q WdS)z*Delete all checkpoints for the given path.rN)rr)rrZ checkpointr r r delete_all_checkpoints2sz"Checkpoints.delete_all_checkpointsN) __name__ __module__ __qualname____doc__r r rrrrrr r r r r 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) NT)contenttypenotebookrfileformatizUnexpected type %s)getcreate_notebook_checkpointcreate_file_checkpointr)rrrmodelrr r r r Nsz)GenericCheckpointsMixin.create_checkpointcCs\|j|ddd}|dkr(|||}n$|dkr>|||}ntdd||||dS) zRestore a checkpoint.F)rrrrizUnexpected type %sN)r get_notebook_checkpointget_file_checkpointrZsave)rrr rrr#r r r r _sz*GenericCheckpointsMixin.restore_checkpointcCs tddS)zwCreate a checkpoint of the current state of a file Returns a checkpoint model for the new checkpoint. z!must be implemented in a subclassN)r)rrrrr r r r"ksz.GenericCheckpointsMixin.create_file_checkpointcCs tddS)zwCreate a checkpoint of the current state of a file Returns a checkpoint model for the new checkpoint. z!must be implemented in a subclassN)r)rZnbrr r r r!rsz2GenericCheckpointsMixin.create_notebook_checkpointcCs tddS)zGet the content of a checkpoint for a non-notebook file. Returns a dict of the form: { 'type': 'file', 'content': , 'format': {'text','base64'}, } z!must be implemented in a subclassN)r)rr rr r r r%ys z+GenericCheckpointsMixin.get_file_checkpointcCs tddS)zGet the content of a checkpoint for a notebook. Returns a dict of the form: { 'type': 'notebook', 'content': , } z!must be implemented in a subclassN)r)rr rr r r r$s z/GenericCheckpointsMixin.get_notebook_checkpointN) rrrrr r r"r!r%r$r r r r r8s  rN)rZ tornado.webrZtraitlets.config.configurablerrobjectrr r r r s  +