machineconfig 1.94__py3-none-any.whl → 1.96__py3-none-any.whl

machineconfig/scripts/python/archive/tmate_start.py

@@ -6,19 +6,19 @@ from pathlib import Path
 import random
 import string
 import os
+from rich.console import Console
+from rich.panel import Panel
 
 
 def main():
-    print(f"""
-╔{'═' * 60}╗
-║ 📡 Tmate Session Launcher
-╚{'═' * 60}╝
-""")
-    
-    print("🔐 Loading credentials...")
+    console = Console()
+
+    console.print(Panel("📡 Tmate Session Launcher", title="[bold blue]Welcome[/bold blue]", subtitle="Manage your tmate sessions effortlessly"))
+
+    console.print("[bold yellow]Loading credentials...[/bold yellow]")
     creds = configparser.ConfigParser()
     creds.read(Path.home().joinpath('dotfiles/creds/tmate/creds.ini'))
-    print("✅ Credentials loaded")
+    console.print("[green]Credentials loaded[/green]")
 
     parser = argparse.ArgumentParser(description='Tmate launcher')
     random_sess = random.choices(list(string.digits + string.ascii_letters), k=20)
@@ -27,21 +27,17 @@ def main():
 
     args = parser.parse_args()
 
-    print(f"🔍 Looking up session configuration: {args.sess_name}")
+    console.print(f"🔍 Looking up session configuration: {args.sess_name}")
     sess_name = creds['sessions_names'][args.sess_name]
     api_key = creds['keys']['api_key']
     
-    print(f"""
-╭{'─' * 60}╮
-│ 🚀 Starting tmate session: {sess_name}
-╰{'─' * 60}╯
-""")
+    console.print(Panel(f"🚀 Starting tmate session: {sess_name}", title="[bold green]Session Info[/bold green]"))
     
     res = f"tmate -a ~/.ssh/authorized_keys -k {api_key} -n {sess_name} -F"
-    print("⚙️  Running: tmate with configured API key and session name")
+    console.print("[bold cyan]Running:[/bold cyan] tmate with configured API key and session name")
     os.system(res)
     
-    print("✅ Tmate session ended")
+    console.print("[green]Tmate session ended[/green]")
 
 
 if __name__ == '__main__':

machineconfig/scripts/python/choose_wezterm_theme.py

@@ -2,9 +2,14 @@
 Choose a theme for Wezterm
 """
 
-from machineconfig.utils.utils import choose_one_option, P
+from machineconfig.utils.utils import choose_one_option, PathExtended
 from typing import Any
 import time
+from rich.panel import Panel
+from rich.console import Console
+import curses
+
+console = Console()
 
 
 schemes_list = [
@@ -38,11 +43,7 @@ schemes_list = [
 
 
 def main2():
-    print(f"""
-╔{'═' * 60}╗
-║ 🎨 WezTerm Theme Selector
-╚{'═' * 60}╝
-""")
+    console.print(Panel("🎨 WezTerm Theme Selector", title_align="left", border_style="green"))
     option = choose_one_option(options=schemes_list, header="Choose a theme for Wezterm", fzf=True)
     set_theme(option)
     print(f"✅ Theme set to: {option}")
@@ -50,36 +51,26 @@ def main2():
 
 def set_theme(theme: str):
     print(f"🔄 Setting WezTerm theme to: {theme}")
-    txt_lines = P("~/.config/wezterm/wezterm.lua").expanduser().read_text().splitlines()
+    txt_lines = PathExtended("~/.config/wezterm/wezterm.lua").expanduser().read_text().splitlines()
     res_lines = []
     for line in txt_lines:
         if 'config.color_scheme = ' in line:
             res_lines.append(f"config.color_scheme = '{theme}'")
         else: res_lines.append(line)
-    P("~/.config/wezterm/wezterm.lua").expanduser().write_text('\n'.join(res_lines))
+    PathExtended("~/.config/wezterm/wezterm.lua").expanduser().write_text('\n'.join(res_lines))
     time.sleep(0.1)
     print("💾 Configuration saved")
 
 
 def main():
-    print(f"""
-╔{'═' * 60}╗
-║ 🎨 WezTerm Theme Selector - Interactive Mode
-╚{'═' * 60}╝
-
+    console.print(Panel("🎨 WezTerm Theme Selector - Interactive Mode", title_align="left", border_style="blue"))
+    print("""
 📝 Use arrow keys to navigate, Enter to select a theme
 """)
-    import curses  # not availble on windows: https://docs.python.org/3/howto/curses.html
-    # from curses import wrapper
     curses.wrapper(accessory)
-    print(f"""
-╔{'═' * 60}╗
-║ ✅ Theme selection completed
-╚{'═' * 60}╝
-""")
+    console.print(Panel("✅ Theme selection completed", title_align="left", border_style="green"))
 
 def accessory(stdscr: Any):
-    import curses
     options = schemes_list
     current_option = 0
     page_size = stdscr.getmaxyx()[0] - 1  # curses.LINES - 1  # Number of lines in the terminal, -1 for status line

machineconfig/scripts/python/cloud_copy.py

@@ -2,35 +2,32 @@
 CC
 """
 
-from crocodile.file_management import P
-from crocodile.core import Struct
+from machineconfig.utils.path_reduced import P as PathExtended
 from crocodile.meta import RepeatUntilNoException
 import getpass
-from machineconfig.scripts.python.helpers.cloud_helpers import Args
 import argparse
 import os
-# from dataclasses import dataclass
-# from pydantic import BaseModel
 from typing import Optional
 
 from machineconfig.scripts.python.helpers.helpers2 import parse_cloud_source_target
-from machineconfig.scripts.python.helpers.cloud_helpers import ArgsDefaults
+from machineconfig.scripts.python.helpers.cloud_helpers import ArgsDefaults, Args
+from rich.console import Console
+from rich.panel import Panel
+from rich.progress import Progress
+from machineconfig.utils.utils2 import pprint
 
+console = Console()
 
 @RepeatUntilNoException(retry=3, sleep=1)
 def get_securely_shared_file(url: Optional[str] = None, folder: Optional[str] = None) -> None:
-    print(f"""
-╔{'═' * 70}╗
-║ 🚀 Secure File Downloader                                                  ║
-╚{'═' * 70}╝
-""")
+    console.print(Panel("🚀 Secure File Downloader", title="[bold blue]Downloader[/bold blue]", border_style="blue"))
     
-    folder_obj = P.cwd() if folder is None else P(folder)
+    folder_obj = PathExtended.cwd() if folder is None else PathExtended(folder)
     print(f"📂 Target folder: {folder_obj}")
     
     if os.environ.get("DECRYPTION_PASSWORD") is not None:
         print("🔑 Using password from environment variables")
-        pwd: str=str(os.environ.get("DECRYPTION_PASSWORD"))
+        pwd = str(os.environ.get("DECRYPTION_PASSWORD"))
     else:
         pwd = getpass.getpass(prompt="🔑 Enter decryption password: ")
         
@@ -42,40 +39,28 @@ def get_securely_shared_file(url: Optional[str] = None, folder: Optional[str] =
         else:
             url = input("🔗 Enter share URL: ")
     
-    print(f"""
-╭{'─' * 70}╮
-│ 📡 Downloading from URL...                                                │
-╰{'─' * 70}╯
-""")
-    from rich.progress import Progress
+    console.print(Panel("📡 Downloading from URL...", title="[bold blue]Download[/bold blue]", border_style="blue"))
     with Progress(transient=True) as progress:
         _task = progress.add_task("Downloading... ", total=None)
-        url_obj = P(url).download(folder=folder_obj)
+        url_obj = PathExtended(url).download(folder=folder_obj)
         
-    print(f"""
-╭{'─' * 70}╮
-│ 📥 Downloaded file: {url_obj}                                
-╰{'─' * 70}╯
-""")
+    console.print(Panel(f"📥 Downloaded file: {url_obj}", title="[bold green]Success[/bold green]", border_style="green"))
     
-    print(f"""
-╭{'─' * 70}╮
-│ 🔐 Decrypting and extracting...                                           │
-╰{'─' * 70}╯
-""")
+    console.print(Panel("🔐 Decrypting and extracting...", title="[bold blue]Processing[/bold blue]", border_style="blue"))
     with Progress(transient=True) as progress:
         _task = progress.add_task("Decrypting... ", total=None)
-        tmp_folder = P.tmpdir(prefix="tmp_unzip")
-        res = url_obj.decrypt(pwd=pwd, inplace=True).unzip(inplace=True, folder=tmp_folder)
-        res.search("*").apply(lambda x: x.move(folder=folder_obj, overwrite=True))
+        tmp_folder = PathExtended.tmpdir(prefix="tmp_unzip")
+        try:
+            res = url_obj.decrypt(pwd=pwd, inplace=True).unzip(inplace=True, folder=tmp_folder)
+            res.search("*").apply(lambda x: x.move(folder=folder_obj, overwrite=True))
+        finally:
+            # Clean up temporary folder
+            if tmp_folder.exists():
+                tmp_folder.delete()
 
 
 def arg_parser() -> None:
-    print(f"""
-╔{'═' * 70}╗
-║ ☁️  Cloud Copy Utility                                                     ║
-╚{'═' * 70}╝
-""")
+    console.print(Panel("☁️  Cloud Copy Utility", title="[bold blue]Cloud Copy[/bold blue]", border_style="blue", width=152))
     
     parser = argparse.ArgumentParser(description='🚀 Cloud CLI. It wraps rclone with sane defaults for optimum type time.')
 
@@ -90,7 +75,7 @@ def arg_parser() -> None:
 
     parser.add_argument("--key", "-k", help="🔑 Key for encryption", type=str, default=ArgsDefaults.key)
     parser.add_argument("--pwd", "-p", help="🔒 Password for encryption", type=str, default=ArgsDefaults.pwd)
-    parser.add_argument("--encrypt", "-e", help="🔐 Decrypt after receiving.", action="store_true", default=ArgsDefaults.encrypt)
+    parser.add_argument("--encrypt", "-e", help="🔐 Encrypt before sending.", action="store_true", default=ArgsDefaults.encrypt)
     parser.add_argument("--zip", "-z", help="📦 unzip after receiving.", action="store_true", default=ArgsDefaults.zip_)
     parser.add_argument("--os_specific", "-o", help="💻 choose path specific for this OS.", action="store_true", default=ArgsDefaults.os_specific)
 
@@ -103,98 +88,58 @@ def arg_parser() -> None:
     args_obj = Args(**args_dict)
 
     if args_obj.config == "ss" and (source.startswith("http") or source.startswith("bit.ly")):
-        print(f"""
-╭{'─' * 70}╮
-│ 🔒 Detected secure share link                                             │
-╰{'─' * 70}╯
-""")
+        console.print(Panel("🔒 Detected secure share link", title="[bold yellow]Warning[/bold yellow]", border_style="yellow"))
         if source.startswith("https://drive.google.com/open?id="):
-            source = "https://drive.google.com/uc?export=download&id=" + source.split("https://drive.google.com/open?id=")[1]
-            print("🔄 Converting Google Drive link to direct download URL")
+            file_id = source.split("https://drive.google.com/open?id=")[1]
+            if file_id:  # Ensure we actually extracted an ID
+                source = f"https://drive.google.com/uc?export=download&id={file_id}"
+                print("🔄 Converting Google Drive link to direct download URL")
+            else:
+                console.print(Panel("❌ Invalid Google Drive link format", title="[bold red]Error[/bold red]", border_style="red"))
+                raise ValueError("Invalid Google Drive link format")
         return get_securely_shared_file(url=source, folder=target)
 
     if args_obj.rel2home is True and args_obj.root is None:
         args_obj.root = "myhome"
         print("🏠 Using 'myhome' as root directory")
 
-    print(f"""
-╭{'─' * 70}╮
-│ 🔍 Parsing source and target paths...                                     │
-╰{'─' * 70}╯
-""")
+    console.print(Panel("🔍 Parsing source and target paths...", title="[bold blue]Info[/bold blue]", border_style="blue"))
     cloud, source, target = parse_cloud_source_target(args=args_obj, source=source, target=target)
     
-    print(f"""
-╭{'─' * 70}╮
-│ ⚙️  Configuration:                                                         │
-╰{'─' * 70}╯
-""")
-    Struct(args_obj.__dict__).print(as_config=True, title="CLI config")
-
-    assert args_obj.key is None, "Key is not supported yet."
+    console.print(Panel("⚙️  Configuration:", title="[bold blue]Config[/bold blue]", border_style="blue"))
+    pprint(args_obj.__dict__, "CLI config")
+
+    if args_obj.key is not None:
+        console.print(Panel("❌ Key-based encryption is not supported yet", title="[bold red]Error[/bold red]", border_style="red"))
+        raise ValueError("Key-based encryption is not supported yet.")
     
     if cloud in source:
-        print(f"""
-╔{'═' * 70}╗
-║ 📥 DOWNLOADING FROM CLOUD                                                 ║
-╠{'═' * 70}╣
-║ ☁️  Cloud: {cloud}                                                        
-║ 📂 Source: {source.replace(cloud + ":", "")}                  
-║ 🎯 Target: {target}                                
-╚{'═' * 70}╝
-""")
+        console.print(Panel(f"📥 DOWNLOADING FROM CLOUD\n☁️  Cloud: {cloud}\n📂 Source: {source.replace(cloud + ':', '')}\n🎯 Target: {target}", title="[bold blue]Download[/bold blue]", border_style="blue", width=152))
         
-        P(target).from_cloud(cloud=cloud, remotepath=source.replace(cloud + ":", ""),
+        PathExtended(target).from_cloud(cloud=cloud, remotepath=source.replace(cloud + ":", ""),
                             unzip=args_obj.zip, decrypt=args_obj.encrypt, pwd=args_obj.pwd,
                             overwrite=args_obj.overwrite,
                             rel2home=args_obj.rel2home, os_specific=args_obj.os_specific, root=args_obj.root, strict=False,
                             )
-        print(f"""
-╔{'═' * 70}╗
-║ ✅ Download completed successfully                                         ║
-╚{'═' * 70}╝
-""")
+        console.print(Panel("✅ Download completed successfully", title="[bold green]Success[/bold green]", border_style="green", width=152))
         
     elif cloud in target:
-        print(f"""
-╔{'═' * 70}╗
-║ 📤 UPLOADING TO CLOUD                                                     ║
-╠{'═' * 70}╣
-║ ☁️  Cloud: {cloud}                                                        
-║ 📂 Source: {source}                                    
-║ 🎯 Target: {target.replace(cloud + ":", "")}                  
-╚{'═' * 70}╝
-""")
+        console.print(Panel(f"📤 UPLOADING TO CLOUD\n☁️  Cloud: {cloud}\n📂 Source: {source}\n🎯 Target: {target.replace(cloud + ':', '')}", title="[bold blue]Upload[/bold blue]", border_style="blue", width=152))
         
-        res = P(source).to_cloud(cloud=cloud, remotepath=target.replace(cloud + ":", ""),
+        res = PathExtended(source).to_cloud(cloud=cloud, remotepath=target.replace(cloud + ":", ""),
                                     zip=args_obj.zip, encrypt=args_obj.encrypt, pwd=args_obj.pwd,
                                     rel2home=args_obj.rel2home, root=args_obj.root, os_specific=args_obj.os_specific, strict=False,
                                     share=args_obj.share)
-        print(f"""
-╔{'═' * 70}╗
-║ ✅ Upload completed successfully                                           ║
-╚{'═' * 70}╝
-""")
+        console.print(Panel("✅ Upload completed successfully", title="[bold green]Success[/bold green]", border_style="green", width=152))
         
         if args_obj.share:
             fname = f".share_url_{cloud}"
-            if P(source).is_dir(): share_url_path = P(source).joinpath(fname)
-            else: share_url_path = P(source).with_suffix(fname)
+            if PathExtended(source).is_dir(): share_url_path = PathExtended(source).joinpath(fname)
+            else: share_url_path = PathExtended(source).with_suffix(fname)
             share_url_path.write_text(res.as_url_str())
-            print(f"""
-╔{'═' * 70}╗
-║ 🔗 SHARE URL GENERATED                                                    ║
-╠{'═' * 70}╣
-║ 📝 URL file: {share_url_path}                            
-║ 🌍 {res.as_url_str()}
-╚{'═' * 70}╝
-""")
+            console.print(Panel(f"🔗 SHARE URL GENERATED\n📝 URL file: {share_url_path}\n🌍 {res.as_url_str()}", title="[bold blue]Share[/bold blue]", border_style="blue", width=152))
     else: 
-        print(f"""
-╔{'═' * 70}╗
-║ ❌ ERROR: Cloud '{cloud}' not found in source or target                   ║
-╚{'═' * 70}╝
-""")
+        console.print(Panel(f"❌ ERROR: Cloud '{cloud}' not found in source or target", title="[bold red]Error[/bold red]", border_style="red", width=152))
         raise ValueError(f"Cloud `{cloud}` not found in source or target.")
 
 

machineconfig/scripts/python/cloud_manager.py

@@ -1,91 +1,77 @@
-"""Run cloud manager.
-"""
-
-from machineconfig.cluster.loader_runner import CloudManager
-import argparse
-
-
-def main():
-    print(f"""
-╔{'═' * 70}╗
-║ ☁️  Cloud Manager                                                         ║
-╚{'═' * 70}╝
-""")
-    
-    parser = argparse.ArgumentParser()
-    parser.add_argument("-c", "--cloud", help="Rclone Config Name", action="store", type=str, default=None)
-    parser.add_argument("-s", "--serve", help="Start job server", action="store_true", default=False)
-    parser.add_argument("-R", "--reset_local", help="Clear local cache", action="store_true", default=False)
-    parser.add_argument("-r", "--rerun_jobs", help="Re-run jobs by bringing them back from wherever to the queue.", action="store_true", default=False)
-    parser.add_argument("-L", "--release_lock", help="Release lock.", action="store_true", default=False)
-    parser.add_argument("-f", "--queue_failed_jobs", help="Bring failed jobs back to queued jobs for a re-trial.", action="store_true", default=False)
-    parser.add_argument("-m", "--monitor_cloud", help="Monitor workers instead of running a job server.", action="store_true", default=False)
-    parser.add_argument("-j", "--num_jobs", help="Number of jobs the server will run in parallel.", action="store", type=int, default=1)
-    args = parser.parse_args()
-
-    print(f"""
-╭{'─' * 70}╮
-│ 🔧 Initializing Cloud Manager with {args.num_jobs} worker{'s' if args.num_jobs > 1 else ''}    │
-╰{'─' * 70}╯
-""")
-    
-    cm = CloudManager(max_jobs=args.num_jobs, cloud=args.cloud, reset_local=args.reset_local)
-    
-    if args.release_lock:
-        print(f"""
-╭{'─' * 70}╮
-│ 🔓 Releasing lock...                                                      │
-╰{'─' * 70}╯
-""")
-        cm.claim_lock()
-        cm.release_lock()
-        print("✅ Lock successfully released")
-        
-    if args.queue_failed_jobs:
-        print(f"""
-╭{'─' * 70}╮
-│ 🔄 Requeuing failed jobs...                                               │
-╰{'─' * 70}╯
-""")
-        cm.clean_failed_jobs_mess()
-        print("✅ Failed jobs moved to queue")
-        
-    if args.rerun_jobs:
-        print(f"""
-╭{'─' * 70}╮
-│ 🔁 Rerunning jobs...                                                      │
-╰{'─' * 70}╯
-""")
-        cm.rerun_jobs()
-        print("✅ Jobs restarted successfully")
-        
-    if args.monitor_cloud:
-        print(f"""
-╔{'═' * 70}╗
-║ 👁️  STARTING CLOUD MONITOR                                                 ║
-╚{'═' * 70}╝
-""")
-        cm.run_monitor()
-        
-    if args.serve:
-        print(f"""
-╔{'═' * 70}╗
-║ 🚀 STARTING JOB SERVER                                                    ║
-╠{'═' * 70}╣
-║ 💻 Running {args.num_jobs} worker{'s' if args.num_jobs > 1 else ''}                                                   ║
-║ ☁️  Cloud: {args.cloud if args.cloud else 'Default'}                                               
-╚{'═' * 70}╝
-""")
-        cm.serve()
-        
-    print(f"""
-╔{'═' * 70}╗
-║ ✅ Cloud Manager finished successfully                                    ║
-╚{'═' * 70}╝
-""")
-    import sys
-    sys.exit(0)
-
-
-if __name__ == '__main__':
-    main()
+# """Run cloud manager.
+# """
+
+# from machineconfig.cluster.cloud_manager import CloudManager
+# from rich.console import Console  # Add import for Console
+# from rich.panel import Panel  # Add import for Panel
+# import argparse
+
+# console = Console()
+
+# BOX_WIDTH = 150  # width for box drawing
+
+
+# def print_section_title(title: str):
+#     """Prints a section title formatted nicely with a border and padding."""
+#     console = Console()
+#     console.print(Panel(title, title_align="left", expand=False))
+
+
+# def main():
+#     console = Console() # Add console initialization
+#     console.print(Panel("☁️  Cloud Manager", title_align="left", expand=False))
+
+#     parser = argparse.ArgumentParser()
+#     parser.add_argument("-c", "--cloud", help="Rclone Config Name", action="store", type=str, default=None)
+#     parser.add_argument("-s", "--serve", help="Start job server", action="store_true", default=False)
+#     parser.add_argument("-R", "--reset_local", help="Clear local cache", action="store_true", default=False)
+#     parser.add_argument("-r", "--rerun_jobs", help="Re-run jobs by bringing them back from wherever to the queue.", action="store_true", default=False)
+#     parser.add_argument("-L", "--release_lock", help="Release lock.", action="store_true", default=False)
+#     parser.add_argument("-f", "--queue_failed_jobs", help="Bring failed jobs back to queued jobs for a re-trial.", action="store_true", default=False)
+#     parser.add_argument("-m", "--monitor_cloud", help="Monitor workers instead of running a job server.", action="store_true", default=False)
+#     parser.add_argument("-j", "--num_jobs", help="Number of jobs the server will run in parallel.", action="store", type=int, default=1)
+#     args = parser.parse_args()
+
+#     init_line = f"🔧 Initializing Cloud Manager with {args.num_jobs} worker{'s' if args.num_jobs > 1 else ''}"
+#     console.print(Panel(init_line, title_align="left", expand=False))
+
+#     cm = CloudManager(max_jobs=args.num_jobs, cloud=args.cloud, reset_local=args.reset_local)
+
+#     if args.release_lock:
+#         line = "🔓 Releasing lock..."
+#         console.print(Panel(line, title_align="left", expand=False))
+#         cm.claim_lock()
+#         cm.release_lock()
+#         print("✅ Lock successfully released")
+
+#     if args.queue_failed_jobs:
+#         line = "🔄 Requeuing failed jobs..."
+#         console.print(Panel(line, title_align="left", expand=False))
+#         cm.clean_failed_jobs_mess()
+#         print("✅ Failed jobs moved to queue")
+
+#     if args.rerun_jobs:
+#         line = "🔁 Rerunning jobs..."
+#         console.print(Panel(line, title_align="left", expand=False))
+#         cm.rerun_jobs()
+#         print("✅ Jobs restarted successfully")
+
+#     if args.monitor_cloud:
+#         title = "👁️  STARTING CLOUD MONITOR"
+#         console.print(Panel(title, title_align="left", expand=False))
+#         cm.run_monitor()
+
+#     if args.serve:
+#         title1 = "🚀 STARTING JOB SERVER"
+#         run_line = f"💻 Running {args.num_jobs} worker{'s' if args.num_jobs > 1 else ''}"
+#         cloud_line = f"☁️  Cloud: {args.cloud if args.cloud else 'Default'}"
+#         console.print(Panel(f"{title1}\n{run_line}\n{cloud_line}", title_align="left", expand=False))
+
+#     title = "✅ Cloud Manager finished successfully"
+#     console.print(Panel(title, title_align="left", expand=False))
+#     import sys
+#     sys.exit(0)
+
+
+# if __name__ == '__main__':
+#     main()