a D b@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.!must be implemented in a subclassNNotImplementedError)self contents_mgrpathr Elib/python3.9/site-packages/notebook/services/contents/checkpoints.pycreate_checkpointszCheckpoints.create_checkpointcCs tddS)zRestore a checkpointrNr)rr checkpoint_idr r r r restore_checkpointszCheckpoints.restore_checkpointcCs tddS)z5Rename a single checkpoint from old_path to new_path.rNr)rrold_pathnew_pathr r r rename_checkpoint!szCheckpoints.rename_checkpointcCs tddS)zdelete a checkpoint for a filerNrrrr r r r delete_checkpoint%szCheckpoints.delete_checkpointcCs tddS)z-Return a list of checkpoints for a given filerNr)rr r r r list_checkpoints)szCheckpoints.list_checkpointscCs&||D]}||d||q dS)z0Rename all checkpoints for old_path to new_path.idN)rr)rrrZcpr r r rename_all_checkpoints-sz"Checkpoints.rename_all_checkpointscCs$||D]}||d|q dS)z*Delete all checkpoints for the given path.rN)rr)rr Z checkpointr r r delete_all_checkpoints2sz"Checkpoints.delete_all_checkpointsN) __name__ __module__ __qualname____doc__r rrrrrrr 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) NTcontenttypenotebookrfileformatUnexpected type %s)getcreate_notebook_checkpointcreate_file_checkpointr)rr r modelr r r r r Nsz)GenericCheckpointsMixin.create_checkpointcCs\|j|ddd}|dkr(|||}n$|dkr>|||}ntdd||||dS) zRestore a checkpoint.Frr r!r"r$r%N)r&get_notebook_checkpointget_file_checkpointrZsave)rr rr r r)r r r r_sz*GenericCheckpointsMixin.restore_checkpointcCs tddSzwCreate a checkpoint of the current state of a file Returns a checkpoint model for the new checkpoint. rNr)rrr#r r r r r(ksz.GenericCheckpointsMixin.create_file_checkpointcCs tddSr,r)rZnbr r 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'}, } rNrrr 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': , } rNrrr r r r*s z/GenericCheckpointsMixin.get_notebook_checkpointN) rrrrr rr(r'r+r*r r r r r8s  rN)rZ tornado.webrZtraitlets.config.configurablerrobjectrr r r r s  +