o i@s>dZddlmZddlmZGdddeZGdddZdS) 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) cCtd)zCreate a checkpoint.!must be implemented in a subclassNotImplementedError)self contents_mgrpathr a/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/notebook/services/contents/checkpoints.pycreate_checkpointzCheckpoints.create_checkpointcCr)zRestore a checkpointrr)r r checkpoint_idr r r r restore_checkpointrzCheckpoints.restore_checkpointcCr)z5Rename a single checkpoint from old_path to new_path.rr)r rold_pathnew_pathr r r rename_checkpoint!rzCheckpoints.rename_checkpointcCr)zdelete a checkpoint for a filerrr rr r r r delete_checkpoint%rzCheckpoints.delete_checkpointcCr)z-Return a list of checkpoints for a given filerr)r r r r r list_checkpoints)rzCheckpoints.list_checkpointscCs&||D] }||d||qdS)z0Rename all checkpoints for old_path to new_path.idN)rr)r rrcpr r r rename_all_checkpoints-sz"Checkpoints.rename_all_checkpointscCs$||D] }||d|qdS)z*Delete all checkpoints for the given path.rN)rr)r r checkpointr r r delete_all_checkpoints2sz"Checkpoints.delete_all_checkpointsN) __name__ __module__ __qualname____doc__rrrrrrrr 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|dkr&||d|d|Stdd |) NTcontenttypenotebookr#fileformatUnexpected type )getcreate_notebook_checkpointcreate_file_checkpointr)r r r modelr$r r r rNsz)GenericCheckpointsMixin.create_checkpointcCs^|j|ddd}|dkr|||}n|dkr|||}ntdd||||dS) zRestore a checkpoint.Fr"r$r%r&r(r)N)r*get_notebook_checkpointget_file_checkpointrsave)r r rr r$r-r r r r_sz*GenericCheckpointsMixin.restore_checkpointcCrzwCreate a checkpoint of the current state of a file Returns a checkpoint model for the new checkpoint. rr)r r#r'r r r r r,kz.GenericCheckpointsMixin.create_file_checkpointcCrr1r)r nbr r r r r+rr2z2GenericCheckpointsMixin.create_notebook_checkpointcCr)zGet the content of a checkpoint for a non-notebook file. Returns a dict of the form: { 'type': 'file', 'content': , 'format': {'text','base64'}, } rrrr r r r/ys z+GenericCheckpointsMixin.get_file_checkpointcCr)zGet the content of a checkpoint for a notebook. Returns a dict of the form: { 'type': 'notebook', 'content': , } rrrr r r r.s z/GenericCheckpointsMixin.get_notebook_checkpointN) rrrr rrr,r+r/r.r r r r r!8s  r!N)r tornado.webrtraitlets.config.configurablerrr!r r r r s   +