python+deepseek api连接ssh linux服务器命令行ai助手工具代码

代码语言:python

所属分类:人工智能

代码描述:python+deepseek api连接ssh linux服务器命令行ai助手工具代码,可将你文字描述转换成当前系统的cmd命令进行执行并返回结果。当前deepseek api也是调用openai的库,也就说适合所有兼容openai的大模型api。

代码标签: python deepseek api ssh linux 命令行 ai 助手 工具 代码

下面为部分代码预览,完整代码请点击下载或在bfwstudio webide中打开

#!/usr/local/python3/bin/python3
# -*- coding: utf-8 -*
import paramiko
from openai import OpenAI

class SSHAssistant:
    def __init__(self):
        self.ssh = paramiko.SSHClient()
        self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        self.system_info = {}

    def connect(self, hostname, username, password=None, key_filename=None):
        try:
            self.ssh.connect(hostname, username=username, password=password, key_filename=key_filename)
            self.system_info = self.get_remote_system_info()
            print(self.system_info)
            print("SSH连接成功")
            return True
        except Exception as e:
            print(f"SSH连接失败: {str(e)}")
            return False

    def get_remote_system_info(self):
        """获取远程服务器系统信息"""
        info = {}
        commands = {
            'os': 'cat /etc/os-release | grep PRETTY_NAME',
            'kernel': 'uname -r',
            'hostname': 'hostname',
            'cpu': 'cat /proc/cpuinfo | grep "model name" | head -n1',
            'memory': 'free -h | head -n2 | tail -n1'
        }
        
        for key, cmd in commands.items():
            try:
                stdin, stdout, stderr = self.ssh.exec_command(cmd, timeout=10)
                output = stdout.read().decode().strip()
                error = stderr.read().decode().strip()
                
                if error:
                    print(f"命令 '{cmd}' 执行出错: {error}")
                    info[key] = "N/A"
  .........完整代码请登录后点击上方下载按钮下载查看

网友评论0