websocket+Django+python+paramiko实现web页面执行命令并实时输出

websocket+Django+python+paramiko实现web页面执行命令并实时输出

一、概述

WebSocket

WebSocket的工作流程:浏览器通过JavaScript向服务端发出建立WebSocket连接的请求,在WebSocket连接建立成功后,客户端和服务端就可以通过 TCP连接传输数据。因为WebSocket连接本质上是TCP连接,不需要每次传输都带上重复的头部数据,所以它的数据传输量比轮询和Comet技术小很多。

paramiko

paramiko模块,基于SSH用于连接远程服务器并执行相关操作。

 

shell脚本

/opt/test.sh

#!/bin/bash

for i in {1..10}
do
    sleep 0.5
    echo 母鸡生了$i个鸡蛋;
done

 

网页执行脚本,效果如下:

 websocket+Django+python+paramiko实现web页面执行命令并实时输出_第1张图片

 

怎么样,是不是很nb!下面会详细介绍如何具体实现!

 

二、详细操作

django版本

最新版本 2.1.5有问题,使用websocket,谷歌浏览器会报错

WebSocket connection to 'ws://127.0.01:8000/echo_once/' failed: Error during WebSocket handshake: Unexpected response code: 400

 

所以不能使用最新版本,必须使用 2.1.4以及2.x系列都可以!

 

安装模块

pip3 install paramiko dwebsocket django==2.1.4

 

创建项目

使用Pycharm创建一个项目 wdpy ,这个是测试的,名字无所谓!

websocket+Django+python+paramiko实现web页面执行命令并实时输出_第2张图片

 

添加路由,修改文件 urls.py

from django.contrib import admin
from django.urls import path
from websocket import views

urlpatterns = [
    path('admin/', admin.site.urls),
    path('echo_once/', views.echo_once),
]

 

修改views.py,增加视图函数

from django.shortcuts import render
from dwebsocket.decorators import accept_websocket, require_websocket
from django.http import HttpResponse
import paramiko


# def exec_command(comm):
#     hostname = '192.168.0.162'
#     username = 'root'
#     password = 'root'
#
#     ssh = paramiko.SSHClient()
#     ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
#     ssh.connect(hostname=hostname, username=username, password=password)
#     stdin, stdout, stderr = ssh.exec_command(comm,get_pty=True)
#     result = stdout.read()
#     ssh.close()
#     yield result


@accept_websocket
def echo_once(request):
    if not request.is_websocket():  # 判断是不是websocket连接
        try:  # 如果是普通的http方法
            message = request.GET['message']
            return HttpResponse(message)
        except:
            return render(request, 'index.html')
    else:
        for message in request.websocket:
            message = message.decode('utf-8')  # 接收前端发来的数据
            print(message)
            if message == 'backup_all':#这里根据web页面获取的值进行对应的操作
                command = 'bash /opt/test.sh'#这里是要执行的命令或者脚本
                
                # 远程连接服务器
                hostname = '192.168.0.162'
                username = 'root'
                password = 'root'

                ssh = paramiko.SSHClient()
                ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
                ssh.connect(hostname=hostname, username=username, password=password)
                # 务必要加上get_pty=True,否则执行命令会没有权限
                stdin, stdout, stderr = ssh.exec_command(command, get_pty=True)
                # result = stdout.read()
                # 循环发送消息给前端页面
                while True:
                    nextline = stdout.readline().strip()  # 读取脚本输出内容
                    # print(nextline.strip())
                    request.websocket.send(nextline) # 发送消息到客户端
                    # 判断消息为空时,退出循环
                    if not nextline:
                        break

                ssh.close()  # 关闭ssh连接
            else:
                request.websocket.send('小样儿,没权限!!!'.encode('utf-8'))
View Code

 

 在 templates 目录下新建文件 index.html




    "Content-Type" content="text/html; charset=utf-8"/>
    CMDB系统
    
    


{#
#}

"margin: 20px;">脚本执行结果:

"messagecontainer" style="margin: 20px;">

View Code

 

和manage.py 同级目录,新建目录static,在里面放 jquery-1.12.4.js 文件

从百度搜索下载即可!

 

修改 settings.py ,设置satic路径

STATIC_URL = '/static/'
STATICFILES_DIRS = (
    os.path.join(BASE_DIR,"static"),
)

 

requirements.txt  这个是项目依赖文件,内容如下:

Django==2.1.4
dwebsocket==0.5.10
paramiko==2.4.2

 

此时,目录结果如下:

./
├── db.sqlite3
├── manage.py
├── requirements.txt
├── static
│   └── jquery-1.12.4.js
├── templates
│   └── index.html
├── untitled2
│   ├── __init__.py
│   ├── settings.py
│   ├── urls.py
│   └── wsgi.py
└── websocket
    ├── admin.py
    ├── apps.py
    ├── __init__.py
    ├── migrations
    │   └── __init__.py
    ├── models.py
    ├── tests.py
    └── views.py

 

启动linux服务器

这里使用的是linux服务器,系统是 ubuntu-16.04.4-server-amd64

脚本 /opt/test.sh 就是上面的内容,已经设置了755权限

 

启动项目

使用Pycharm启动,访问网页:

http://127.0.0.1:8000/echo_once/

 

效果就是上面演示的!

 

测试命令

除了执行脚本,还可以执行其他命令,比如安装ntpdate

修改views.py,将 command 修改一下

command = 'apt-get install -y ntpdate'#这里是要执行的命令或者脚本

 

再次执行,效果如下:

websocket+Django+python+paramiko实现web页面执行命令并实时输出_第3张图片

 

完整项目

如需完整项目代码,请访问:

https://github.com/py3study/wdpy

 

 

 

本文参考链接:
https://blog.csdn.net/linxi7/article/details/76161584

 

注意:这篇文章的效果并不是实时输出,它是命令执行完成之后,才显示在网页上面的!

那么因此,我在他的代码基础上,做了一些改进!才实现 实时输出的效果!

 

posted @ 2019-01-18 18:45 肖祥 阅读( ...) 评论( ...) 编辑 收藏

你可能感兴趣的:(websocket+Django+python+paramiko实现web页面执行命令并实时输出)