SessionSmith 0.1.3__tar.gz → 0.1.4__tar.gz

{sessionsmith-0.1.3 → sessionsmith-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SessionSmith
-Version: 0.1.3
+Version: 0.1.4
 Summary: Simple session save/load utility for Jupyter notebooks using pickle
 Home-page: https://github.com/yut0takagi/SessionSmith
 Author: YutoTAKAGI

{sessionsmith-0.1.3 → sessionsmith-0.1.4}/SessionSmith/__init__.py

@@ -14,7 +14,7 @@ from .serializers import CustomSerializer
 from .tracer import AlgorithmTracer
 from .visualizer import visualize_algorithm_trace, print_trace_summary
 
-__version__ = "0.1.3"
+__version__ = "0.1.4"
 
 __all__ = [
     "save_session",

{sessionsmith-0.1.3 → sessionsmith-0.1.4}/SessionSmith/core.py

@@ -101,13 +101,14 @@ def _validate_on_error_option(on_error: str) -> str:
     return on_error
 
 
-def _get_globals_dict(globals_dict: Optional[Dict[str, Any]], depth: int = 2) -> Dict[str, Any]:
+def _get_globals_dict(globals_dict: Optional[Dict[str, Any]], depth: int = 2, copy: bool = True) -> Dict[str, Any]:
     """
     グローバル変数辞書を取得します
     
     Args:
         globals_dict: グローバル変数辞書（Noneの場合は自動取得）
         depth: 呼び出し元からのフレーム深度（デフォルトは2：_get_globals_dict -> save/load_session -> ユーザーコード）
+        copy: Trueの場合はコピーを返す（save_session用）、Falseの場合は参照を返す（load_session用）
         
     Returns:
         dict: グローバル変数辞書
@@ -129,7 +130,8 @@ def _get_globals_dict(globals_dict: Optional[Dict[str, Any]], depth: int = 2) ->
                 raise RuntimeError("Cannot access calling frame at specified depth")
             caller_frame = caller_frame.f_back
         
-        result = caller_frame.f_globals.copy()
+        # save_sessionはコピー（安全な読み取り）、load_sessionは参照（変更を反映）
+        result = caller_frame.f_globals.copy() if copy else caller_frame.f_globals
         del frame
         return result
     except Exception as e:
@@ -334,7 +336,8 @@ def load_session(
     if not file_path.is_file():
         raise ValueError(f"'{file_path}' is not a file.")
     
-    globals_dict = _get_globals_dict(globals_dict)
+    # load_sessionでは元のグローバル変数を変更する必要があるため、copy=False
+    globals_dict = _get_globals_dict(globals_dict, copy=False)
     
     if include is not None and not isinstance(include, list):
         raise TypeError(f"include must be a list, got {type(include).__name__}")

{sessionsmith-0.1.3 → sessionsmith-0.1.4}/SessionSmith/manager.py

@@ -59,7 +59,7 @@ class SessionManager:
             depth: 呼び出し元からのフレーム深度（デフォルトは2：_get_globals_dict -> __init__ -> ユーザーコード）
             
         Returns:
-            dict: グローバル変数辞書
+            dict: グローバル変数辞書（参照、コピーではない）
         """
         if globals_dict is not None:
             if not isinstance(globals_dict, dict):
@@ -78,7 +78,8 @@ class SessionManager:
                     raise RuntimeError("Cannot access calling frame at specified depth")
                 caller_frame = caller_frame.f_back
             
-            result = caller_frame.f_globals.copy()
+            # SessionManagerは変数を変更する必要があるため、参照を返す（コピーではない）
+            result = caller_frame.f_globals
             del frame
             return result
         except Exception as e:

{sessionsmith-0.1.3 → sessionsmith-0.1.4}/SessionSmith.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SessionSmith
-Version: 0.1.3
+Version: 0.1.4
 Summary: Simple session save/load utility for Jupyter notebooks using pickle
 Home-page: https://github.com/yut0takagi/SessionSmith
 Author: YutoTAKAGI