machineconfig 3.0__py3-none-any.whl → 3.2__py3-none-any.whl

machineconfig/scripts/python/fire_agents_help_search.py

@@ -1,4 +1,3 @@
-
 from machineconfig.utils.source_of_truth import EXCLUDE_DIRS
 import fnmatch
 from pathlib import Path
@@ -79,4 +78,4 @@ def search_python_files(repo_root: Path, keyword: str) -> list[Path]:
             return
 
     _walk_and_filter(repo_root)
-    return matches
+    return matches

machineconfig/scripts/python/fire_agents_load_balancer.py

@@ -1,5 +1,3 @@
-
-
 from typing import Literal, TypeAlias
 from math import ceil
 from pathlib import Path
@@ -7,36 +5,36 @@ from pathlib import Path
 
 SPLITTING_STRATEGY: TypeAlias = Literal[
     "agent_cap",  # User decides number of agents, rows/tasks determined automatically
-    "task_rows"   # User decides number of rows/tasks, number of agents determined automatically
+    "task_rows",  # User decides number of rows/tasks, number of agents determined automatically
 ]
 DEFAULT_AGENT_CAP = 6
 
 
 def chunk_prompts(prompt_material_path: Path, strategy: SPLITTING_STRATEGY, joiner: str, *, agent_cap: int | None, task_rows: int | None) -> list[str]:
     """Chunk prompts based on splitting strategy.
-    
+
     Args:
         prompts: List of prompts to chunk
-        strategy: Either 'agent_cap' or 'task_rows'  
+        strategy: Either 'agent_cap' or 'task_rows'
         agent_cap: Maximum number of agents (used with 'agent_cap' strategy)
         task_rows: Number of rows/tasks per agent (used with 'task_rows' strategy)
     """
     prompts = [p for p in prompt_material_path.read_text(encoding="utf-8", errors="ignore").split(joiner) if p.strip() != ""]  # drop blank entries
-    
+
     if strategy == "agent_cap":
         if agent_cap is None:
             raise ValueError("agent_cap must be provided when using 'agent_cap' strategy")
-        
+
         if len(prompts) <= agent_cap:
             return prompts
-        
+
         print(f"Chunking {len(prompts)} prompts into groups for up to {agent_cap} agents because it exceeds the cap.")
         chunk_size = ceil(len(prompts) / agent_cap)
         grouped: list[str] = []
         for i in range(0, len(prompts), chunk_size):
             grouped.append(joiner.join(prompts[i : i + chunk_size]))
         return grouped
-    
+
     elif strategy == "task_rows":
         if task_rows is None:
             raise ValueError("task_rows must be provided when using 'task_rows' strategy")
@@ -47,6 +45,6 @@ def chunk_prompts(prompt_material_path: Path, strategy: SPLITTING_STRATEGY, join
         for i in range(0, len(prompts), task_rows):
             grouped.append(joiner.join(prompts[i : i + task_rows]))
         return grouped
-    
+
     else:
         raise ValueError(f"Unknown splitting strategy: {strategy}")

machineconfig/scripts/python/fire_jobs.py

@@ -324,6 +324,7 @@ python -m machineconfig.cluster.templates.cli_click --file {choice_file} """
     # program_path.parent.mkdir(parents=True, exist_ok=True)
     # program_path.write_text(command, encoding="utf-8")
     import subprocess
+
     subprocess.run(command, shell=True, check=True)
 
 

machineconfig/scripts/python/helpers/cloud_helpers.py

@@ -24,6 +24,7 @@ class ArgsDefaults:
     key = None
     pwd = None
 
+
 @dataclass
 class Args:
     cloud: Optional[str] = None

machineconfig/scripts/python/mount_nfs.py

@@ -50,6 +50,7 @@ local_mount_point={local_mount_point}
         """
         # PROGRAM_PATH.write_text(txt)
         import subprocess
+
         subprocess.run(txt, shell=True, check=True)
 
         print("✅ Mount paths prepared successfully!\n")
@@ -65,6 +66,7 @@ $driveLetter = "{driver_letter}"
 """
         # PROGRAM_PATH.write_text(txt)
         import subprocess
+
         subprocess.run(txt, shell=True, check=True)
         print("✅ Drive letter selected and configuration saved!\n")
 

machineconfig/scripts/python/mount_nw_drive.py

@@ -1,4 +1,3 @@
-
 from pathlib import Path
 import platform
 
@@ -34,6 +33,7 @@ password='{password}'
 """
         # PROGRAM_PATH.write_text(txt, encoding="utf-8",)
         import subprocess
+
         subprocess.run(txt, shell=True, check=True)
         print("✅ Configuration saved successfully!\n")
 

machineconfig/scripts/python/mount_ssh.py

@@ -54,6 +54,7 @@ fusermount -u /mnt/dbhdd
 
     # PROGRAM_PATH.write_text(txt, encoding="utf-8")
     import subprocess
+
     subprocess.run(txt, shell=True, check=True)
     print("✅ Configuration saved successfully!\n")
 

machineconfig/scripts/python/repos.py

@@ -116,7 +116,6 @@ def main():
         if args.cloud is not None:
             PathExtended(save_path).to_cloud(rel2home=True, cloud=args.cloud)
 
-
     elif args.clone or args.checkout or args.checkout_to_branch:
         print("\n📥 Cloning or checking out repositories...")
         print(">>>>>>>>> Cloning Repos")
@@ -154,6 +153,5 @@ def main():
         print("❌ No action specified. Try passing --push, --pull, --commit, or --all.")
 
 
-
 if __name__ == "__main__":
     main()

machineconfig/scripts/python/repos_helper_record.py

@@ -1,4 +1,3 @@
-
 from machineconfig.utils.path_reduced import PathExtended as PathExtended
 from machineconfig.utils.schemas.repos.repos_types import GitVersionInfo, RepoRecordDict, RepoRemote
 
@@ -16,11 +15,11 @@ def build_tree_structure(repos: list[RepoRecordDict], repos_root: PathExtended)
     """Build a tree structure representation of all repositories."""
     if not repos:
         return "No repositories found."
-    
+
     # Group repos by their parent directories relative to repos_root
     tree_dict: dict[str, list[RepoRecordDict]] = {}
     repos_root_abs = repos_root.expanduser().absolute()
-    
+
     for repo in repos:
         parent_path = PathExtended(repo["parentDir"]).expanduser().absolute()
         try:
@@ -29,35 +28,35 @@ def build_tree_structure(repos: list[RepoRecordDict], repos_root: PathExtended)
         except ValueError:
             # If the path is not relative to repos_root, use the full path
             relative_str = str(parent_path)
-        
+
         if relative_str not in tree_dict:
             tree_dict[relative_str] = []
         tree_dict[relative_str].append(repo)
-    
+
     # Sort directories for consistent output
     sorted_dirs = sorted(tree_dict.keys())
-    
+
     tree_lines: list[str] = []
     tree_lines.append(f"📂 {repos_root.name}/ ({repos_root_abs})")
-    
+
     for i, dir_path in enumerate(sorted_dirs):
         is_last_dir = i == len(sorted_dirs) - 1
         dir_prefix = "└── " if is_last_dir else "├── "
-        
+
         if dir_path:
             tree_lines.append(f"│   {dir_prefix}📁 {dir_path}/")
             repo_prefix_base = "│   │   " if not is_last_dir else "    "
         else:
             repo_prefix_base = "│   "
-        
+
         repos_in_dir = tree_dict[dir_path]
         # Sort repos by name
         repos_in_dir.sort(key=lambda x: x["name"])
-        
+
         for j, repo in enumerate(repos_in_dir):
             is_last_repo = j == len(repos_in_dir) - 1
             repo_prefix = f"{repo_prefix_base}└── " if is_last_repo else f"{repo_prefix_base}├── "
-            
+
             # Create status indicators
             status_indicators = []
             if repo["isDirty"]:
@@ -66,20 +65,20 @@ def build_tree_structure(repos: list[RepoRecordDict], repos_root: PathExtended)
                 status_indicators.append("⚠️ NO_REMOTE")
             if repo["currentBranch"] == "DETACHED":
                 status_indicators.append("🔀 DETACHED")
-            
+
             status_str = f"[{' | '.join(status_indicators)}]" if status_indicators else "[✅ CLEAN]"
-            branch_info = f" ({repo['currentBranch']})" if repo['currentBranch'] != "DETACHED" else ""
-            
+            branch_info = f" ({repo['currentBranch']})" if repo["currentBranch"] != "DETACHED" else ""
+
             # Build the base string without status
             base_str = f"{repo_prefix}📦 {repo['name']}{branch_info}"
-            
+
             # Calculate padding to align status at 75 characters
             target_width = 45
             current_length = len(base_str)
             padding = max(1, target_width - current_length)  # At least 1 space
-            
+
             tree_lines.append(f"{base_str}{' ' * padding}{status_str}")
-    
+
     return "\n".join(tree_lines)
 
 
@@ -111,19 +110,9 @@ def record_a_repo(path: PathExtended, search_parent_directories: bool, preferred
     # Check if repo is dirty (has uncommitted changes)
     is_dirty = repo.is_dirty(untracked_files=True)
 
-    version_info: GitVersionInfo = {
-        "branch": current_branch,
-        "commit": commit
-    }
-
-    res: RepoRecordDict = {
-        "name": repo_root.name,
-        "parentDir": repo_root.parent.collapseuser().as_posix(),
-        "currentBranch": current_branch,
-        "remotes": remotes,
-        "version": version_info,
-        "isDirty": is_dirty
-    }
+    version_info: GitVersionInfo = {"branch": current_branch, "commit": commit}
+
+    res: RepoRecordDict = {"name": repo_root.name, "parentDir": repo_root.parent.collapseuser().as_posix(), "currentBranch": current_branch, "remotes": remotes, "version": version_info, "isDirty": is_dirty}
     return res
 
 
@@ -132,16 +121,16 @@ def count_git_repositories(repos_root: str, r: bool) -> int:
     path_obj = PathExtended(repos_root).expanduser().absolute()
     if path_obj.is_file():
         return 0
-    
+
     search_res = path_obj.search("*", files=False, folders=True)
     count = 0
-    
+
     for a_search_res in search_res:
         if a_search_res.joinpath(".git").exists():
             count += 1
         elif r:
             count += count_git_repositories(str(a_search_res), r=r)
-    
+
     return count
 
 
@@ -150,15 +139,15 @@ def count_total_directories(repos_root: str, r: bool) -> int:
     path_obj = PathExtended(repos_root).expanduser().absolute()
     if path_obj.is_file():
         return 0
-    
+
     search_res = path_obj.search("*", files=False, folders=True)
     count = len(search_res)
-    
+
     if r:
         for a_search_res in search_res:
             if not a_search_res.joinpath(".git").exists():
                 count += count_total_directories(str(a_search_res), r=r)
-    
+
     return count
 
 
@@ -166,22 +155,22 @@ def record_repos_recursively(repos_root: str, r: bool, progress: Progress | None
     path_obj = PathExtended(repos_root).expanduser().absolute()
     if path_obj.is_file():
         return []
-    
+
     search_res = path_obj.search("*", files=False, folders=True)
     res: list[RepoRecordDict] = []
-    
+
     for a_search_res in search_res:
         if progress and scan_task_id:
             progress.update(scan_task_id, description=f"Scanning: {a_search_res.name}")
-        
+
         if a_search_res.joinpath(".git").exists():
             try:
                 if progress and process_task_id:
                     progress.update(process_task_id, description=f"Recording: {a_search_res.name}")
-                    
+
                 repo_record = record_a_repo(a_search_res, search_parent_directories=False, preferred_remote=None)
                 res.append(repo_record)
-                
+
                 if progress and process_task_id:
                     progress.update(process_task_id, advance=1, description=f"Recorded: {repo_record['name']}")
             except Exception as e:
@@ -189,57 +178,45 @@ def record_repos_recursively(repos_root: str, r: bool, progress: Progress | None
         else:
             if r:
                 res += record_repos_recursively(str(a_search_res), r=r, progress=progress, scan_task_id=scan_task_id, process_task_id=process_task_id)
-        
+
         if progress and scan_task_id:
             progress.update(scan_task_id, advance=1)
-    
+
     return res
 
 
 def main(repos_root: PathExtended):
     print("\n📝 Recording repositories...")
-    
+
     # Count total directories and repositories for accurate progress tracking
     print("🔍 Analyzing directory structure...")
     total_dirs = count_total_directories(str(repos_root), r=True)
     total_repos = count_git_repositories(str(repos_root), r=True)
     print(f"📊 Found {total_dirs} directories to scan and {total_repos} git repositories to record")
-    
+
     # Setup progress bars
-    with Progress(
-        SpinnerColumn(),
-        TextColumn("[progress.description]{task.description}"),
-        BarColumn(),
-        MofNCompleteColumn(),
-        TimeElapsedColumn(),
-    ) as progress:
+    with Progress(SpinnerColumn(), TextColumn("[progress.description]{task.description}"), BarColumn(), MofNCompleteColumn(), TimeElapsedColumn()) as progress:
         scan_task = progress.add_task("Scanning directories...", total=total_dirs)
         process_task = progress.add_task("Recording repositories...", total=total_repos)
-        
-        repo_records = record_repos_recursively(
-            repos_root=str(repos_root), 
-            r=True,
-            progress=progress, 
-            scan_task_id=scan_task, 
-            process_task_id=process_task
-        )
-    
+
+        repo_records = record_repos_recursively(repos_root=str(repos_root), r=True, progress=progress, scan_task_id=scan_task, process_task_id=process_task)
+
     res: RepoRecordFile = {"version": "0.1", "repos": repo_records}
-    
+
     # Summary with warnings
     total_repos = len(repo_records)
     repos_with_no_remotes = [repo for repo in repo_records if len(repo["remotes"]) == 0]
     repos_with_remotes = [repo for repo in repo_records if len(repo["remotes"]) > 0]
     dirty_repos = [repo for repo in repo_records if repo["isDirty"]]
     clean_repos = [repo for repo in repo_records if not repo["isDirty"]]
-    
+
     print("\n📊 Repository Summary:")
     print(f"   Total repositories found: {total_repos}")
     print(f"   Repositories with remotes: {len(repos_with_remotes)}")
     print(f"   Repositories without remotes: {len(repos_with_no_remotes)}")
     print(f"   Clean repositories: {len(clean_repos)}")
     print(f"   Dirty repositories: {len(dirty_repos)}")
-    
+
     if repos_with_no_remotes:
         print(f"\n⚠️  WARNING: {len(repos_with_no_remotes)} repositories have no remotes configured:")
         for repo in repos_with_no_remotes:
@@ -257,12 +234,12 @@ def main(repos_root: PathExtended):
         print("   These repositories have uncommitted changes that may need attention.")
     else:
         print("\n✅ All repositories are clean (no uncommitted changes).")
-    
+
     # Display repository tree structure
     print("\n🌳 Repository Tree Structure:")
     tree_structure = build_tree_structure(repo_records, repos_root)
     print(tree_structure)
-    
+
     save_path = CONFIG_PATH.joinpath("repos").joinpath(repos_root.rel2home()).joinpath("repos.json")
     save_json(obj=res, path=save_path, indent=4)
     pprint(f"📁 Result saved at {PathExtended(save_path)}")

machineconfig/scripts/python/repos_helper_update.py

@@ -4,9 +4,9 @@ import subprocess
 import git
 
 
-
 class RepositoryUpdateResult(TypedDict):
     """Result of updating a single repository."""
+
     repo_path: str
     status: str  # "success", "error", "skipped", "auth_failed"
     had_uncommitted_changes: bool
@@ -61,6 +61,7 @@ def get_file_hash(file_path: Path) -> str | None:
     if not file_path.exists():
         return None
     import hashlib
+
     return hashlib.sha256(file_path.read_bytes()).hexdigest()
 
 
@@ -262,4 +263,4 @@ def update_repository(repo: git.Repo, auto_sync: bool, allow_password_prompt: bo
         result["status"] = "error"
         result["error_message"] = str(e)
         print(f"❌ Error updating repository {repo_path}: {e}")
-        return result
+        return result

machineconfig/scripts/python/start_slidev.py

@@ -103,6 +103,7 @@ def main() -> None:
     program = "npm run dev slides.md -- --remote"
     # PROGRAM_PATH.write_text(program, encoding="utf-8")
     import subprocess
+
     subprocess.run(program, shell=True, cwd=SLIDEV_REPO)
     print_code(code=program, lexer="bash", desc="Run the following command to start the presentation")
 

machineconfig/scripts/python/start_terminals.py

@@ -1,6 +1,5 @@
 """Script to start terminals on windows and wsl"""
 
-
 from machineconfig.utils.options import display_options, get_ssh_hosts
 import platform
 from itertools import cycle
@@ -113,6 +112,7 @@ def main():
 
     # PROGRAM_PATH.write_text(cmd, encoding="utf-8")
     import subprocess
+
     subprocess.run(cmd, shell=True)
     print("✅ Command saved successfully!\n")
 

machineconfig/setup_windows/wt_and_pwsh/set_wt_settings.py

@@ -4,6 +4,7 @@ from machineconfig.utils.utils2 import randstr, read_json
 from machineconfig.utils.path_reduced import PathExtended as PathExtended
 from machineconfig.utils.io_save import save_json
 import platform
+
 # from uuid import uuid4
 import os
 from typing import Any

machineconfig/utils/code.py

@@ -10,7 +10,6 @@ from machineconfig.utils.ve import get_ve_activate_line
 from machineconfig.utils.path_reduced import PathExtended as PathExtended
 
 
-
 def get_shell_script_executing_python_file(python_file: str, func: Optional[str], ve_path: str, strict_execution: bool = True):
     if func is None:
         exec_line = f"""python {python_file}"""

machineconfig/utils/notifications.py

@@ -38,7 +38,7 @@ def md2html(body: str) -> str:
     markdown_obj = Markdown(body)
     console.print(markdown_obj)
     html_output = console.export_html(inline_styles=True)
-    
+
     # Try to load GitHub CSS style, fallback to basic style if not found
     gh_style_path = Path(__file__).parent.joinpath("gh_style.css")
     if gh_style_path.exists():
@@ -51,7 +51,7 @@ def md2html(body: str) -> str:
         code { background-color: #f6f8fa; padding: 2px 4px; border-radius: 3px; }
         pre { background-color: #f6f8fa; padding: 16px; border-radius: 6px; overflow: auto; }
         """
-    
+
     return f"""
 <!DOCTYPE html>
 <html>

machineconfig/utils/procs.py

@@ -20,13 +20,10 @@ def get_processes_accessing_file(path: str):
     title = "🔍  SEARCHING FOR PROCESSES ACCESSING FILE"
     console.print(Panel(title, title="[bold blue]Process Info[/bold blue]", border_style="blue"))
     res: dict[int, list[str]] = {}
-    
-    with Progress(
-        SpinnerColumn(),
-        TextColumn("[progress.description]{task.description}"),
-    ) as progress:
+
+    with Progress(SpinnerColumn(), TextColumn("[progress.description]{task.description}")) as progress:
         progress.add_task("🔎 Scanning processes...", total=None)
-        
+
         for proc in psutil.process_iter():
             try:
                 files = proc.open_files()
@@ -35,7 +32,7 @@ def get_processes_accessing_file(path: str):
             tmp = [file.path for file in files if path in file.path]
             if len(tmp) > 0:
                 res[proc.pid] = tmp
-                
+
     # Convert to list of dictionaries for consistent data structure
     result_data = [{"pid": pid, "files": files} for pid, files in res.items()]
     console.print(Panel(f"✅ Found {len(res)} processes accessing the specified file", title="[bold blue]Process Info[/bold blue]", border_style="blue"))
@@ -61,13 +58,10 @@ class ProcessManager:
         title = "📊  INITIALIZING PROCESS MANAGER"
         console.print(Panel(title, title="[bold blue]Process Info[/bold blue]", border_style="blue"))
         process_info = []
-        
-        with Progress(
-            SpinnerColumn(),
-            TextColumn("[progress.description]{task.description}"),
-        ) as progress:
+
+        with Progress(SpinnerColumn(), TextColumn("[progress.description]{task.description}")) as progress:
             progress.add_task("🔍 Reading system processes...", total=None)
-            
+
             for proc in psutil.process_iter():
                 try:
                     mem_usage_mb = proc.memory_info().rss / (1024 * 1024)
@@ -246,8 +240,9 @@ def get_age(create_time: Any) -> str:
 
 
 def main():
-    from machineconfig.utils.procs import ProcessManager; ProcessManager().choose_and_kill()
+    from machineconfig.utils.procs import ProcessManager
+    ProcessManager().choose_and_kill()
 
 
 if __name__ == "__main__":
-    main()
+    pass

machineconfig/utils/schemas/fire_agents/fire_agents_input.py

@@ -14,33 +14,38 @@ SEARCH_STRATEGIES = Literal["file_path", "keyword_search", "filename_pattern"]
 
 class FilePathSearchInput(TypedDict):
     """Input for file_path search strategy."""
+
     file_path: str
     separator: str  # Default: "\n"
 
 
 class KeywordSearchInput(TypedDict):
     """Input for keyword_search strategy."""
+
     keyword: str
 
 
 class FilenamePatternSearchInput(TypedDict):
     """Input for filename_pattern search strategy."""
+
     pattern: str  # e.g., '*.py', '*test*', 'config.*'
 
 
 class AgentCapSplittingInput(TypedDict):
     """Input for agent_cap splitting strategy."""
+
     agent_cap: int  # Default: 6
 
 
 class TaskRowsSplittingInput(TypedDict):
     """Input for task_rows splitting strategy."""
+
     task_rows: int  # Default: 13
 
 
 class FireAgentsMainInput(TypedDict):
     """Complete input structure for fire_agents main function."""
-    
+
     # Core configuration
     repo_root: Path
     search_strategy: SEARCH_STRATEGIES
@@ -50,12 +55,12 @@ class FireAgentsMainInput(TypedDict):
     job_name: str  # Default: "AI_Agents"
     keep_material_in_separate_file: bool  # Default: False
     max_agents: int  # Default: 25
-    
+
     # Search strategy specific inputs (only one will be present based on search_strategy)
     file_path_input: NotRequired[FilePathSearchInput]
     keyword_search_input: NotRequired[KeywordSearchInput]
     filename_pattern_input: NotRequired[FilenamePatternSearchInput]
-    
+
     # Splitting strategy specific inputs (only one will be present based on splitting_strategy)
     agent_cap_input: NotRequired[AgentCapSplittingInput]
     task_rows_input: NotRequired[TaskRowsSplittingInput]
@@ -63,7 +68,7 @@ class FireAgentsMainInput(TypedDict):
 
 class FireAgentsRuntimeData(TypedDict):
     """Runtime data derived from inputs during execution."""
-    
+
     prompt_material: str
     separator: str
     prompt_material_re_splitted: list[str]

machineconfig/utils/schemas/layouts/layout_types.py

@@ -17,11 +17,13 @@ class TabConfig(TypedDict):
 
 class LayoutConfig(TypedDict):
     """Configuration for a complete layout with its tabs."""
+
     layoutName: str
     layoutTabs: List[TabConfig]
 
 
 class LayoutsFile(TypedDict):
     """Complete layout file structure."""
+
     version: str
     layouts: List[LayoutConfig]

machineconfig/utils/schemas/repos/repos_types.py

@@ -1,9 +1,6 @@
-
-
 from typing import TypedDict
 
 
-
 class GitVersionInfo(TypedDict):
     branch: str
     commit: str