:vj!vdZddlmZddlmZGddeZGddZGdd eZGd d eZy ) 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) ctzCreate a checkpoint.NotImplementedErrorself contents_mgrpaths }/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/jupyter_server/services/contents/checkpoints.pycreate_checkpointzCheckpoints.create_checkpoint!!ctzRestore a checkpointr r r checkpoint_idrs rrestore_checkpointzCheckpoints.restore_checkpointrrctz5Rename a single checkpoint from old_path to new_path.r r rold_pathnew_paths rrename_checkpointzCheckpoints.rename_checkpoint rrctzdelete a checkpoint for a filer r rrs rdelete_checkpointzCheckpoints.delete_checkpoint$rrctz-Return a list of checkpoints for a given filer r rs rlist_checkpointszCheckpoints.list_checkpoints(rrc\|j|D]}|j|d||y)0Rename all checkpoints for old_path to new_path.idNr%rr rrcps rrename_all_checkpointsz"Checkpoints.rename_all_checkpoints,s-''1B  " "2d8Xx @2rcZ|j|D]}|j|d|y)*Delete all checkpoints for the given path.r(Nr%r!r r checkpoints rdelete_all_checkpointsz"Checkpoints.delete_all_checkpoints1s,//5J  " ":d#3T :6rN __name__ __module__ __qualname____doc__rrrr!r%r,r2rrrr s+ """""A ;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 tz NTcontenttypenotebookr>fileformatUnexpected type %s)getcreate_notebook_checkpointcreate_file_checkpointrr?r r rmodeltype_s rrz)GenericCheckpointsMixin.create_checkpointMs  t 4f  J 22i  f_..i h  C!5!<= =rc|j|dd}|dk(r|j||}n'|dk(r|j||}ntdd|z|j ||y) Restore a checkpoint.Fr=r?r@rArCrDNrEget_notebook_checkpointget_file_checkpointrsave)r r rrrJrIs rrz*GenericCheckpointsMixin.restore_checkpoint^sr  u 5f= J 00EE f_,,]DAEC!5!=> >%&rctzwCreate a checkpoint of the current state of a file Returns a checkpoint model for the new checkpoint. r r r>rBrs rrGz.GenericCheckpointsMixin.create_file_checkpointj "!rctrRr r nbrs rrFz2GenericCheckpointsMixin.create_notebook_checkpointqrTrctzGet the content of a checkpoint for a non-notebook file. Returns a dict of the form:: { 'type': 'file', 'content': , 'format': {'text','base64'}, } r r s rrOz+GenericCheckpointsMixin.get_file_checkpointxs "!rctzGet the content of a checkpoint for a notebook. Returns a dict of the form:: { 'type': 'notebook', 'content': , } r r s rrNz/GenericCheckpointsMixin.get_notebook_checkpoints "!rN r4r5r6r7rrrGrFrOrNr8rrr:r:7s%*>" '"" " "rr:c:eZdZdZdZdZdZdZdZdZ dZ y ) AsyncCheckpointszS Base class for managing checkpoints for a ContentsManager asynchronously. cKtwrr r s rrz"AsyncCheckpoints.create_checkpoint !! cKtwrr rs rrz#AsyncCheckpoints.restore_checkpointr`racKtwrr rs rrz"AsyncCheckpoints.rename_checkpointr`racKtwrr r s rr!z"AsyncCheckpoints.delete_checkpointr`racKtwr#r r$s rr%z!AsyncCheckpoints.list_checkpointsr`racK|j|d{D] }|j|d||d{"y7(7 w)r'Nr(r)r*s rr,z'AsyncCheckpoints.rename_all_checkpointssA--h777B((D8XF F F8 FsAA AAAAcK|j|d{D]}|j|d|d{!y7'7 w)r.Nr(r/r0s rr2z'AsyncCheckpoints.delete_all_checkpointss@ $ 5 5d ;;;J((D)94@ @ @< @sA?AAAANr3r8rrr^r^s,"""""G Arr^c4eZdZdZdZdZdZdZdZdZ y) AsyncGenericCheckpointsMixinzp Helper for creating Asynchronous Checkpoints subclasses that can be used with any ContentsManager. cK|j|dd{}|d}|dk(r|j|d|d{S|dk(r!|j|d|d|d{Stdd |z7`7;7wr<)rErFrGrrHs rrz.AsyncGenericCheckpointsMixin.create_checkpoints"&&tT&::f  J 88i  f_44i h  C!5!=> >; s3BA:&BA<%B&A>'B<B>Bc.K|j|dd{}|d}|dk(r|j||d{}n/|dk(r|j||d{}ntdd|z|j ||d{y7s7Q737 w) rLFr=Nr?r@rArCrDrM)r r rr content_modelrJrIs rrz/AsyncGenericCheckpointsMixin.restore_checkpoints*..tU.CC f% J 66}dKKE f_22=$GGEC!5!=> >t,,,DLG -sCBB #BBBB)BBBBBBcKtwrRr rSs rrGz3AsyncGenericCheckpointsMixin.create_file_checkpoint "!racKtwrRr rVs rrFz7AsyncGenericCheckpointsMixin.create_notebook_checkpointrnracKtwrYr r s rrOz0AsyncGenericCheckpointsMixin.get_file_checkpoints"!racKtwr[r r s rrNz4AsyncGenericCheckpointsMixin.get_notebook_checkpoints"!raNr\r8rrriris% ?" -"" " "rriN) r7 tornado.webrtraitlets.config.configurablerrr:r^rir8rrrtsK "=);%);XX"X"v!A{!AHI"#:I"r