IbJdZddlmZddlmZGddeZGddZy) z# Classes for managing Checkpoints. ) HTTPError)LoggingConfigurablec:eZdZdZdZdZdZdZdZdZ dZ y ) 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) ctd)zCreate a checkpoint.!must be implemented in a subclassNotImplementedError)self contents_mgrpaths H/usr/lib/python3/dist-packages/notebook/services/contents/checkpoints.pycreate_checkpointzCheckpoints.create_checkpoint!"EFFctd)zRestore a checkpointrr )r r checkpoint_idr s rrestore_checkpointzCheckpoints.restore_checkpointrrctd)z5Rename a single checkpoint from old_path to new_path.rr )r rold_pathnew_paths rrename_checkpointzCheckpoints.rename_checkpoint!rrctd)zdelete a checkpoint for a filerr r rr s rdelete_checkpointzCheckpoints.delete_checkpoint%rrctd)z-Return a list of checkpoints for a given filerr )r r s rlist_checkpointszCheckpoints.list_checkpoints)rrc\|j|D]}|j|d||y)z0Rename all checkpoints for old_path to new_path.idN)rr)r rrcps rrename_all_checkpointsz"Checkpoints.rename_all_checkpoints-s4''1 AB  " "2d8Xx @ ArcZ|j|D]}|j|d|y)z*Delete all checkpoints for the given path.rN)rr)r r checkpoints rdelete_all_checkpointsz"Checkpoints.delete_all_checkpoints2s1//5 ;J  " ":d#3T : ;rN) __name__ __module__ __qualname____doc__rrrrrr!r$rrrr s0 GGGGGA ;rrc4eZdZdZdZdZdZdZdZdZ y) 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) c|j|d}|d}|dk(r|j|d|S|dk(r|j|d|d|Stdd |) NTcontenttypenotebookr.fileformatUnexpected type )getcreate_notebook_checkpointcreate_file_checkpointr)r r r modelr/s rrz)GenericCheckpointsMixin.create_checkpointNs  t 4V} : 22i  V^..i h  C#3D6!:; ;rc|j|dd}|dk(r|j||}n'|dk(r|j||}ntdd||j ||y) zRestore a checkpoint.Fr-r/r0r1r3r4N)r5get_notebook_checkpointget_file_checkpointrsave)r r rr r/r8s rrz*GenericCheckpointsMixin.restore_checkpoint_ste4V< : 00EE V^,,]DAEC#3D6!:; ;%&rctdzwCreate a checkpoint of the current state of a file Returns a checkpoint model for the new checkpoint. rr )r r.r2r s rr7z.GenericCheckpointsMixin.create_file_checkpointk ""EFFrctdr>r )r nbr s rr6z2GenericCheckpointsMixin.create_notebook_checkpointrr?rctd)zGet the content of a checkpoint for a non-notebook file. Returns a dict of the form: { 'type': 'file', 'content': , 'format': {'text','base64'}, } rr rs rr;z+GenericCheckpointsMixin.get_file_checkpointys""EFFrctd)zGet the content of a checkpoint for a notebook. Returns a dict of the form: { 'type': 'notebook', 'content': , } rr rs rr:z/GenericCheckpointsMixin.get_notebook_checkpoints""EFFrN) r%r&r'r(rrr7r6r;r:r)rrr+r+8s)*<" 'GG G Grr+N)r( tornado.webrtraitlets.config.configurablerrr+r)rrrFs."=(;%(;VVGVGr