machineconfig 1.97__py3-none-any.whl → 2.1__py3-none-any.whl

machineconfig/cluster/sessions_managers/wt_remote.py

@@ -18,20 +18,19 @@ TMP_LAYOUT_DIR = Path.home().joinpath("tmp_results", "wt_layouts", "layout_manag
 
 
 class WTRemoteLayoutGenerator:
-    
     def __init__(self, remote_name: str, session_name_prefix: str):
         self.remote_name = remote_name
         self.session_name = session_name_prefix + "_" + WTLayoutGenerator.generate_random_suffix()
         self.tab_config: Dict[str, Tuple[str, str]] = {}
         self.script_path: Optional[str] = None
-        
+
         # Initialize modular components
         self.remote_executor = WTRemoteExecutor(remote_name)
         self.layout_generator = WTLayoutGenerator()
         self.process_monitor = WTProcessMonitor(self.remote_executor)
         self.session_manager = WTSessionManager(self.remote_executor, self.session_name, TMP_LAYOUT_DIR)
         self.status_reporter = WTStatusReporter(self.process_monitor, self.session_manager)
-    
+
     def copy_script_to_remote(self, local_script_file: Path, random_suffix: str) -> str:
         return self.session_manager.copy_script_to_remote(local_script_file, random_suffix)
 
@@ -42,12 +41,12 @@ class WTRemoteLayoutGenerator:
             output_path = Path(output_dir)
         else:
             output_path = TMP_LAYOUT_DIR
-        self.script_path = self.layout_generator.create_wt_script(tab_config, output_path, self.session_name)        
+        self.script_path = self.layout_generator.create_wt_script(tab_config, output_path, self.session_name)
         return self.script_path
-    
+
     def get_layout_preview(self, tab_config: Dict[str, Tuple[str, str]]) -> str:
         return self.layout_generator.generate_wt_command(tab_config)
-    
+
     def check_command_status(self, tab_name: str, use_verification: bool = True) -> Dict[str, Any]:
         return self.process_monitor.check_command_status(tab_name, self.tab_config, use_verification)
 
@@ -102,14 +101,7 @@ class WTRemoteLayoutGenerator:
         return self.remote_executor.get_remote_windows_info()
 
     def to_dict(self) -> Dict[str, Any]:
-        return {
-            "remote_name": self.remote_name,
-            "session_name": self.session_name,
-            "tab_config": self.tab_config,
-            "script_path": self.script_path,
-            "created_at": datetime.now().isoformat(),
-            "class_name": self.__class__.__name__
-        }
+        return {"remote_name": self.remote_name, "session_name": self.session_name, "tab_config": self.tab_config, "script_path": self.script_path, "created_at": datetime.now().isoformat(), "class_name": self.__class__.__name__}
 
     def to_json(self, file_path: Optional[str] = None) -> str:
         # Generate file path if not provided
@@ -120,57 +112,57 @@ class WTRemoteLayoutGenerator:
             path_obj = default_dir / f"wt_session_{random_id}.json"
         else:
             path_obj = Path(file_path)
-        
+
         # Ensure .json extension
-        if not str(path_obj).endswith('.json'):
-            path_obj = path_obj.with_suffix('.json')
-            
+        if not str(path_obj).endswith(".json"):
+            path_obj = path_obj.with_suffix(".json")
+
         # Ensure parent directory exists
         path_obj.parent.mkdir(parents=True, exist_ok=True)
-        
+
         # Serialize to JSON
         data = self.to_dict()
-        
-        with open(path_obj, 'w', encoding='utf-8') as f:
-            json.dump(data, f, indent=2, ensure_ascii=False)
-        
+
+        text = json.dumps(data, indent=2, ensure_ascii=False)
+        path_obj.write_text(text, encoding="utf-8")
+
         logger.info(f"✅ Serialized WTRemoteLayoutGenerator to: {path_obj}")
         return str(path_obj)
 
     @classmethod
-    def from_json(cls, file_path: str) -> 'WTRemoteLayoutGenerator':
+    def from_json(cls, file_path: str) -> "WTRemoteLayoutGenerator":
         path_obj = Path(file_path)
-        
+
         # Ensure .json extension
-        if not str(path_obj).endswith('.json'):
-            path_obj = path_obj.with_suffix('.json')
-            
+        if not str(path_obj).endswith(".json"):
+            path_obj = path_obj.with_suffix(".json")
+
         if not path_obj.exists():
             raise FileNotFoundError(f"JSON file not found: {path_obj}")
-        
+
         # Load JSON data
-        with open(path_obj, 'r', encoding='utf-8') as f:
+        with open(path_obj, "r", encoding="utf-8") as f:
             data = json.load(f)
-        
+
         # Validate that it's the correct class
-        if data.get('class_name') != cls.__name__:
+        if data.get("class_name") != cls.__name__:
             logger.warning(f"Class name mismatch: expected {cls.__name__}, got {data.get('class_name')}")
-        
+
         # Create new instance
         # Extract session name prefix by removing the suffix
-        session_name = data['session_name']
-        if '_' in session_name:
-            session_name_prefix = '_'.join(session_name.split('_')[:-1])
+        session_name = data["session_name"]
+        if "_" in session_name:
+            session_name_prefix = "_".join(session_name.split("_")[:-1])
         else:
             session_name_prefix = session_name
-            
-        instance = cls(remote_name=data['remote_name'], session_name_prefix=session_name_prefix)
-        
+
+        instance = cls(remote_name=data["remote_name"], session_name_prefix=session_name_prefix)
+
         # Restore state
-        instance.session_name = data['session_name']
-        instance.tab_config = data['tab_config']
-        instance.script_path = data['script_path']
-        
+        instance.session_name = data["session_name"]
+        instance.tab_config = data["tab_config"]
+        instance.script_path = data["script_path"]
+
         logger.info(f"✅ Loaded WTRemoteLayoutGenerator from: {file_path}")
         return instance
 
@@ -180,10 +172,10 @@ class WTRemoteLayoutGenerator:
             dir_path = Path.home() / "tmp_results" / "wt_sessions" / "serialized"
         else:
             dir_path = Path(directory_path)
-            
+
         if not dir_path.exists():
             return []
-        
+
         json_files = [f.name for f in dir_path.glob("*.json")]
         return sorted(json_files)
 
@@ -220,69 +212,70 @@ if __name__ == "__main__":
     # Example usage
     sample_tabs = {
         "🤖Bot1": ("~/code/bytesense/bithence", "python bot1.py --create_new_bot True"),
-        "🤖Bot2": ("~/code/bytesense/bithence", "python bot2.py --create_new_bot True"), 
+        "🤖Bot2": ("~/code/bytesense/bithence", "python bot2.py --create_new_bot True"),
         "📊Monitor": ("~", "Get-Process | Sort-Object CPU -Descending | Select-Object -First 10"),
-        "📝Logs": ("C:/logs", "Get-Content app.log -Wait")
+        "📝Logs": ("C:/logs", "Get-Content app.log -Wait"),
     }
-    
+
     # Replace 'myserver' with an actual SSH config alias for a Windows machine
     remote_name = "myserver"  # This should be in ~/.ssh/config
     session_name = "test_remote_session"
-    
+
     try:
         # Create layout using the remote generator
         generator = WTRemoteLayoutGenerator(remote_name=remote_name, session_name_prefix=session_name)
         script_path = generator.create_wt_layout(sample_tabs)
         print(f"✅ Remote layout created successfully: {script_path}")
-        
+
         # Check if Windows Terminal is available on remote
         wt_available = generator.check_wt_available()
         print(f"🖥️  Windows Terminal available on {remote_name}: {'✅' if wt_available else '❌'}")
-        
+
         # Get remote Windows info
         windows_info = generator.get_remote_windows_info()
         if windows_info.get("wt_available"):
             print(f"📦 Remote system info: {windows_info.get('windows_info', 'Unknown')}")
-        
+
         # Demonstrate serialization
         print("\n💾 Demonstrating serialization...")
         saved_path = generator.to_json()
         print(f"✅ Session saved to: {saved_path}")
-        
+
         # List all saved sessions
         saved_sessions = WTRemoteLayoutGenerator.list_saved_sessions()
         print(f"📋 Available saved sessions: {saved_sessions}")
-        
+
         # Demonstrate loading (using the full path)
         loaded_generator = WTRemoteLayoutGenerator.from_json(saved_path)
         print(f"✅ Session loaded successfully: {loaded_generator.session_name}")
         print(f"📊 Loaded tabs: {list(loaded_generator.tab_config.keys())}")
-        
+
         # Show command preview
         preview = generator.get_layout_preview(sample_tabs)
         print(f"\n📋 Command Preview:\n{preview}")
-        
+
         # Demonstrate status checking
         print(f"\n🔍 Checking command status on remote '{remote_name}':")
         generator.print_status_report()
-        
+
         # Show Windows Terminal overview
-        print(f"\n🖥️  Windows Terminal Overview:")
+        print("\n🖥️  Windows Terminal Overview:")
         generator.print_windows_terminal_overview()
-        
+
         # Start the session (uncomment to actually start)
         # start_result = generator.start_wt_session()
         # print(f"Session start result: {start_result}")
-        
+
         # Attach to session (uncomment to attach)
         # generator.attach_to_session()
-        
-        print(f"\n▶️  To start this session, run:")
-        print(f"   generator.start_wt_session()")
-        print(f"\n📎 To attach to this session, run:")
-        print(f"   generator.attach_to_session()")
-        
+
+        print("\n▶️  To start this session, run:")
+        print("   generator.start_wt_session()")
+        print("\n📎 To attach to this session, run:")
+        print("   generator.attach_to_session()")
+
     except Exception as e:
         print(f"❌ Error: {e}")
         import traceback
-        traceback.print_exc() 
+
+        traceback.print_exc()