增加运维机器人单元

aibox/skills/ssh-exec.py

@@ -0,0 +1,77 @@
+#!/usr/bin/env python3
+"""ssh-exec - 通过SSH远程执行单条命令的工具（支持密码认证）"""
+
+import argparse
+import sys
+import paramiko
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="ssh-exec - 通过SSH远程执行单条命令（支持密码认证）",
+        epilog="""
+参数说明:
+  -H, --host       SSH主机地址（默认: 127.0.0.1）
+  -P, --port       SSH端口号（默认: 22）
+  -u, --user        SSH用户名（必填）
+  -p, --password   SSH密码（必填）
+  -c, --command    要执行的远程命令（必填）
+
+使用示例:
+  %(prog)s -u qing -p 12345678 -c "ls -la"
+  %(prog)s -u root -H 192.168.1.100 -P 22 -p mypass -c "df -h"
+  %(prog)s -u qing -H 127.0.0.1 -p 12345678 -c "uname -a"
+""",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+    parser.add_argument("-H", "--host", default="127.0.0.1", help="SSH主机地址（默认: 127.0.0.1）")
+    parser.add_argument("-P", "--port", type=int, default=22, help="SSH端口号（默认: 22）")
+    parser.add_argument("-u", "--user", required=True, help="SSH用户名（必填）")
+    parser.add_argument("-p", "--password", required=True, help="SSH密码（必填）")
+    parser.add_argument("-c", "--command", required=True, help="要执行的远程命令（必填）")
+
+    args = parser.parse_args()
+
+    client = paramiko.SSHClient()
+    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+
+    try:
+        client.connect(
+            hostname=args.host,
+            port=args.port,
+            username=args.user,
+            password=args.password,
+            timeout=10,
+        )
+
+        stdin, stdout, stderr = client.exec_command(args.command, timeout=30)
+
+        exit_code = stdout.channel.recv_exit_status()
+        out_text = stdout.read().decode("utf-8", errors="replace")
+        err_text = stderr.read().decode("utf-8", errors="replace")
+
+        if out_text:
+            print(out_text, end="")
+        if err_text:
+            print(err_text, end="", file=sys.stderr)
+
+        sys.exit(exit_code)
+
+    except paramiko.AuthenticationException:
+        print(f"错误: SSH认证失败（{args.user}@{args.host}:{args.port}）", file=sys.stderr)
+        sys.exit(1)
+    except paramiko.SSHException as e:
+        print(f"错误: SSH连接失败 - {e}", file=sys.stderr)
+        sys.exit(1)
+    except TimeoutError:
+        print(f"错误: 连接超时（{args.host}:{args.port}）", file=sys.stderr)
+        sys.exit(1)
+    except Exception as e:
+        print(f"错误: {e}", file=sys.stderr)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+if __name__ == "__main__":
+    main()