88from .._utils import AsyncCallbacks , CancelCallback , wrap_async
99from ..types import MISSING , MISSING_TYPE
1010from ._button import BOOKMARK_ID
11- from ._restore_state import RestoreContextState
12- from ._save_state import ShinySaveState
11+ from ._restore_state import RestoreState
12+ from ._save_state import BookmarkState
1313from ._types import BookmarkStore
1414
1515# TODO: Barret - Bookmark state
@@ -192,8 +192,7 @@ def _get_bookmark_exclude(self) -> list[str]:
192192 def on_bookmark (
193193 self ,
194194 callback : (
195- Callable [[ShinySaveState ], None ]
196- | Callable [[ShinySaveState ], Awaitable [None ]]
195+ Callable [[BookmarkState ], None ] | Callable [[BookmarkState ], Awaitable [None ]]
197196 ),
198197 / ,
199198 ) -> CancelCallback :
@@ -266,8 +265,7 @@ async def do_bookmark(self) -> None:
266265 def on_restore (
267266 self ,
268267 callback : (
269- Callable [[RestoreContextState ], None ]
270- | Callable [[RestoreContextState ], Awaitable [None ]]
268+ Callable [[RestoreState ], None ] | Callable [[RestoreState ], Awaitable [None ]]
271269 ),
272270 ) -> CancelCallback :
273271 """
@@ -281,8 +279,7 @@ def on_restore(
281279 def on_restored (
282280 self ,
283281 callback : (
284- Callable [[RestoreContextState ], None ]
285- | Callable [[RestoreContextState ], Awaitable [None ]]
282+ Callable [[RestoreState ], None ] | Callable [[RestoreState ], Awaitable [None ]]
286283 ),
287284 ) -> CancelCallback :
288285 """
@@ -402,8 +399,7 @@ def _get_bookmark_exclude(self) -> list[str]:
402399 def on_bookmark (
403400 self ,
404401 callback : (
405- Callable [[ShinySaveState ], None ]
406- | Callable [[ShinySaveState ], Awaitable [None ]]
402+ Callable [[BookmarkState ], None ] | Callable [[BookmarkState ], Awaitable [None ]]
407403 ),
408404 / ,
409405 ) -> CancelCallback :
@@ -419,17 +415,15 @@ def on_bookmarked(
419415 def on_restore (
420416 self ,
421417 callback : (
422- Callable [[RestoreContextState ], None ]
423- | Callable [[RestoreContextState ], Awaitable [None ]]
418+ Callable [[RestoreState ], None ] | Callable [[RestoreState ], Awaitable [None ]]
424419 ),
425420 ) -> CancelCallback :
426421 return self ._on_restore_callbacks .register (wrap_async (callback ))
427422
428423 def on_restored (
429424 self ,
430425 callback : (
431- Callable [[RestoreContextState ], None ]
432- | Callable [[RestoreContextState ], Awaitable [None ]]
426+ Callable [[RestoreState ], None ] | Callable [[RestoreState ], Awaitable [None ]]
433427 ),
434428 ) -> CancelCallback :
435429 return self ._on_restored_callbacks .register (wrap_async (callback ))
@@ -457,14 +451,14 @@ async def do_bookmark(self) -> None:
457451
458452 try :
459453 # ?withLogErrors
460- from ..bookmark ._bookmark import ShinySaveState
454+ from ..bookmark ._bookmark import BookmarkState
461455 from ..session import session_context
462456
463- async def root_state_on_save (state : ShinySaveState ) -> None :
457+ async def root_state_on_save (state : BookmarkState ) -> None :
464458 with session_context (self ._session_root ):
465459 await self ._on_bookmark_callbacks .invoke (state )
466460
467- root_state = ShinySaveState (
461+ root_state = BookmarkState (
468462 input = self ._session_root .input ,
469463 exclude = self ._get_bookmark_exclude (),
470464 on_save = root_state_on_save ,
@@ -541,7 +535,7 @@ def __init__(self, session_proxy: SessionProxy):
541535 # The goal of this method is to save the scope's values. All namespaced inputs
542536 # will already exist within the `root_state`.
543537 @self ._root_bookmark .on_bookmark
544- async def scoped_on_bookmark (root_state : ShinySaveState ) -> None :
538+ async def scoped_on_bookmark (root_state : BookmarkState ) -> None :
545539 return await self ._scoped_on_bookmark (root_state )
546540
547541 from ..session import session_context
@@ -557,7 +551,7 @@ async def scoped_on_bookmarked(url: str) -> None:
557551 ns_prefix = str (self ._ns + self ._ns ._sep )
558552
559553 @self ._root_bookmark .on_restore
560- async def scoped_on_restore (restore_state : RestoreContextState ) -> None :
554+ async def scoped_on_restore (restore_state : RestoreState ) -> None :
561555 if self ._on_restore_callbacks .count () == 0 :
562556 return
563557
@@ -567,22 +561,22 @@ async def scoped_on_restore(restore_state: RestoreContextState) -> None:
567561 await self ._on_restore_callbacks .invoke (scoped_restore_state )
568562
569563 @self ._root_bookmark .on_restored
570- async def scoped_on_restored (restore_state : RestoreContextState ) -> None :
564+ async def scoped_on_restored (restore_state : RestoreState ) -> None :
571565 if self ._on_restored_callbacks .count () == 0 :
572566 return
573567
574568 scoped_restore_state = restore_state ._state_within_namespace (ns_prefix )
575569 with session_context (self ._session_proxy ):
576570 await self ._on_restored_callbacks .invoke (scoped_restore_state )
577571
578- async def _scoped_on_bookmark (self , root_state : ShinySaveState ) -> None :
572+ async def _scoped_on_bookmark (self , root_state : BookmarkState ) -> None :
579573 # Exit if no user-defined callbacks.
580574 if self ._on_bookmark_callbacks .count () == 0 :
581575 return
582576
583- from ..bookmark ._bookmark import ShinySaveState
577+ from ..bookmark ._bookmark import BookmarkState
584578
585- scoped_state = ShinySaveState (
579+ scoped_state = BookmarkState (
586580 input = self ._session_root .input ,
587581 exclude = self ._root_bookmark .exclude ,
588582 on_save = None ,
@@ -621,8 +615,7 @@ def _create_effects(self) -> NoReturn:
621615 def on_bookmark (
622616 self ,
623617 callback : (
624- Callable [[ShinySaveState ], None ]
625- | Callable [[ShinySaveState ], Awaitable [None ]]
618+ Callable [[BookmarkState ], None ] | Callable [[BookmarkState ], Awaitable [None ]]
626619 ),
627620 / ,
628621 ) -> CancelCallback :
@@ -658,17 +651,15 @@ def store(self) -> BookmarkStore:
658651 def on_restore (
659652 self ,
660653 callback : (
661- Callable [[RestoreContextState ], None ]
662- | Callable [[RestoreContextState ], Awaitable [None ]]
654+ Callable [[RestoreState ], None ] | Callable [[RestoreState ], Awaitable [None ]]
663655 ),
664656 ) -> CancelCallback :
665657 return self ._on_restore_callbacks .register (wrap_async (callback ))
666658
667659 def on_restored (
668660 self ,
669661 callback : (
670- Callable [[RestoreContextState ], None ]
671- | Callable [[RestoreContextState ], Awaitable [None ]]
662+ Callable [[RestoreState ], None ] | Callable [[RestoreState ], Awaitable [None ]]
672663 ),
673664 ) -> CancelCallback :
674665 return self ._on_restored_callbacks .register (wrap_async (callback ))
@@ -697,8 +688,7 @@ def _get_bookmark_exclude(self) -> NoReturn:
697688 def on_bookmark (
698689 self ,
699690 callback : (
700- Callable [[ShinySaveState ], None ]
701- | Callable [[ShinySaveState ], Awaitable [None ]]
691+ Callable [[BookmarkState ], None ] | Callable [[BookmarkState ], Awaitable [None ]]
702692 ),
703693 ) -> NoReturn :
704694 raise NotImplementedError (
@@ -728,8 +718,7 @@ async def do_bookmark(self) -> NoReturn:
728718 def on_restore (
729719 self ,
730720 callback : (
731- Callable [[RestoreContextState ], None ]
732- | Callable [[RestoreContextState ], Awaitable [None ]]
721+ Callable [[RestoreState ], None ] | Callable [[RestoreState ], Awaitable [None ]]
733722 ),
734723 ) -> NoReturn :
735724 raise NotImplementedError (
@@ -739,8 +728,7 @@ def on_restore(
739728 def on_restored (
740729 self ,
741730 callback : (
742- Callable [[RestoreContextState ], None ]
743- | Callable [[RestoreContextState ], Awaitable [None ]]
731+ Callable [[RestoreState ], None ] | Callable [[RestoreState ], Awaitable [None ]]
744732 ),
745733 ) -> NoReturn :
746734 raise NotImplementedError (
0 commit comments