小編給大家分享一下Django怎么實現(xiàn)WebSSH操作物理機或虛擬機，希望大家閱讀完這篇文章之后都有所收獲，下面讓我們一起去探討吧！
WebSSH操作物理機或虛擬機
上篇文章給大家介紹詳解基于django實現(xiàn)的webssh簡單例子，有小伙伴說咖啡哥，我們現(xiàn)在還沒有用上Kubernetes，但我想通過瀏覽器連接我們的物理機和虛擬機該怎么辦？
這就比較簡單了，既然我們已經(jīng)實現(xiàn)了瀏覽器操作Kubernetes的Pod，那么想想操作Pod和物理機虛擬機有什么區(qū)別呢？
整個數(shù)據(jù)流是一點沒變：用戶打開瀏覽器--》瀏覽器發(fā)送websocket請求給Django建立長連接--》Django與要操作的服務器建立SSH通道，實時的將收到的用戶數(shù)據(jù)發(fā)送給SSH后的主機，并將主機執(zhí)行的結果數(shù)據(jù)返回給瀏覽器
唯一不一樣的地方就是Django與要操作的服務器建立SSH通道的方式，在Kubernetes中是通過Kubernetes提供的API建立的Stream流，而操作物理機或者虛擬機的時候我們可以使用Paramiko模塊來建立SSH長連接隧道，Paramiko模塊建立SSH長連接通道的方法如下：
# 實例化SSHClient ssh = paramiko.SSHClient() # 當遠程服務器沒有本地主機的密鑰時自動添加到本地，這樣不用在建立連接的時候輸入yes或no進行確認 ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) # 連接SSH服務器，這里以賬號密碼的方式進行認證，也可以用key ssh.connect(hostname=host, port=port, username=username, password=password, timeout=8) # 打開ssh通道，建立長連接 transport = ssh.get_transport() self.ssh_channel = transport.open_session() # 獲取ssh通道，并設置term和終端大小 self.ssh_channel.get_pty(term=term, width=cols, height=rows) # 激活終端，這樣就可以正常登陸了 self.ssh_channel.invoke_shell()
連接建立，可以通過如下方法給SSH通道發(fā)送數(shù)據(jù)
self.ssh_channel.send(data)
當然SSH返回的數(shù)據(jù)也可以通過如下方法持續(xù)的輸出給Websocket
重慶分公司
重慶分公司
while not self.ssh_channel.exit_status_ready():
# SSH返回的數(shù)據(jù)需要轉碼為utf-8，否則json序列化會失敗
data = self.ssh_channel.recv(1024).decode('utf-8','ignore')
if len(data) != 0:
message = {'flag': 'success', 'message': data}
self.websocket.send(json.dumps(message))
else:
break有了這些信息，結合詳解基于django實現(xiàn)的webssh簡單例子的文章，實現(xiàn)WebSSH瀏覽器操作物理機或者虛擬機就不算困難了，完整的Consumer代碼如下：
import io
import json
import paramiko
from threading import Thread
from channels.generic.websocket import WebsocketConsumer
from cmdb.backends.sshargs import args
class SSHBridge(object):
def __init__(self, websocket):
self.websocket = websocket
def connect(self, host, port, username, authtype, password=None, pkey=None, term='xterm', cols=80, rows=24):
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try:
if authtype == 2:
pkey = paramiko.RSAKey.from_private_key(io.StringIO(pkey))
ssh.connect(username=username, hostname=host, port=port, pkey=pkey, timeout=8)
else:
ssh.connect(hostname=host, port=port, username=username, password=password, timeout=8)
except Exception as e:
message = json.dumps({'flag': 'error', 'message': str(e)})
self.websocket.send(message)
return False
# 打開一個ssh通道并建立連接
transport = ssh.get_transport()
self.ssh_channel = transport.open_session()
self.ssh_channel.get_pty(term=term, width=cols, height=rows)
self.ssh_channel.invoke_shell()
# 連接建立一次，之后交互數(shù)據(jù)不會再進入該方法
for i in range(2):
recv = self.ssh_channel.recv(1024).decode('utf-8', 'ignore')
message = json.dumps({'flag': 'success', 'message': recv})
self.websocket.send(message)
def close(self):
try:
self.websocket.close()
self.ssh_channel.close()
except BaseException as e:
pass
def _ws_to_ssh(self, data):
try:
self.ssh_channel.send(data)
except OSError as e:
self.close()
def _ssh_to_ws(self):
try:
while not self.ssh_channel.exit_status_ready():
data = self.ssh_channel.recv(1024).decode('utf-8', 'ignore')
if len(data) != 0:
message = {'flag': 'success', 'message': data}
self.websocket.send(json.dumps(message))
else:
break
except Exception as e:
message = {'flag': 'error', 'message': str(e)}
self.websocket.send(json.dumps(message))
self.close()
def shell(self, data):
Thread(target=self._ws_to_ssh, args=(data,)).start()
Thread(target=self._ssh_to_ws).start()
class SSHConsumer(WebsocketConsumer):
def connect(self):
self.pk = self.scope['url_route']['kwargs'].get('id')
self.query = self.scope.get('query_string')
self.user = self.scope['user']
self.accept()
# ssh_connect_args為SSH連接需要的參數(shù)
ssh_connect_args = args(self.pk, self.user, self.query)
self.ssh = SSHBridge(websocket=self)
self.ssh.connect(**ssh_connect_args)
def disconnect(self, close_code):
self.ssh.close()
def receive(self, text_data=None):
text_data = json.loads(text_data)
self.ssh.shell(data=text_data.get('data', ''))動態(tài)調(diào)整終端窗口大小
看了詳解基于django實現(xiàn)的webssh簡單例子文章，小伙伴又說了，你這只能在連接建立時確定終端窗口的大小，如果我中途調(diào)整了瀏覽器的大小，顯示就亂了，這該怎么辦？
不要著急，接下來就讓我們看看怎么讓終端窗口隨著瀏覽器大小的調(diào)整而改變，上邊的文章中已經(jīng)說過，終端窗口的大小需要瀏覽器和后端返回的Terminal大小保持一致，單單調(diào)整頁面窗口大小或者后端返回的Terminal窗口大小都是不行的，那么從這兩個方向來說明該如何動態(tài)調(diào)整窗口的大小
首先Paramiko模塊建立的SSH通道可以通過resize_pty來動態(tài)改變返回Terminal窗口的大小，使用方法如下：
def resize_pty(self, cols, rows): self.ssh_channel.resize_pty(width=cols, height=rows)
然后Django的Channels每次接收到前端發(fā)過來的數(shù)據(jù)時，判斷一下窗口是否有變化，如果有變化則調(diào)用上邊的方法動態(tài)改變Terminal輸出窗口的大小
我在實現(xiàn)時會給傳過來的數(shù)據(jù)加個flag，如果flag是resize，則調(diào)用resize_pty的方法動態(tài)調(diào)整窗口大小，否則就正常調(diào)用執(zhí)行命令的方法，代碼如下：
def receive(self, text_data=None):
text_data = json.loads(text_data)
if text_data.get('flag') == 'resize':
self.ssh.resize_pty(cols=text_data['cols'], rows=text_data['rows'])
else:
self.ssh.shell(data=text_data.get('data', ''))后端都搞定了，那么來看看前端如何處理吧
首先有一個terminal_size的方法根據(jù)瀏覽器窗口大小除以每個字符所占用的大小計算出cols和rows的值，無論是xterm.js還是Paramiko都是根據(jù)這兩個值來調(diào)整窗口大小的
function terminal_size() {
return {
cols: Math.floor($('#terminal').width() / 9),
rows: Math.floor($(window).height() / 17),
}
}然后通過$(window).resize()來檢測瀏覽器窗口的變化，一旦發(fā)生變化，則發(fā)送一個帶resize標記的數(shù)據(jù)給Django，同時傳遞的數(shù)據(jù)還有新的cols和rows
// terminal resize
$(window).resize(function () {
let cols = terminal_size().cols;
let rows = terminal_size().rows;
send_data = JSON.stringify({
'flag': 'resize',
'cols': cols,
'rows': rows
});
socket.send(send_data);
term.resize(cols, rows)
})最后通過term.resize來調(diào)整xterm渲染的窗口的大小
這樣一個完整的動態(tài)調(diào)整窗口大小的方案就完成了
演示與源碼
我寫了個簡單的Demo來實現(xiàn)上邊的功能，Demo寫完發(fā)現(xiàn)還挺好用，我就擴展了一下添加了內(nèi)網(wǎng)的物理機和虛擬機，歷史原因，有些是賬號密碼認證，有些是密鑰認證，我都給兼容了一下，最終實現(xiàn)的效果如上圖所示
項目里邊要記錄主機的密碼，為了安全這個密碼是通過RSA加密存放在數(shù)據(jù)庫的，每次使用的時候進行解密，加解密的實現(xiàn)，可參考這篇文章django 開發(fā)忘記密碼通過郵箱找回功能示例
看完了這篇文章，相信你對“Django怎么實現(xiàn)WebSSH操作物理機或虛擬機”有了一定的了解，如果想了解更多相關知識，歡迎關注創(chuàng)新互聯(lián)成都網(wǎng)站設計公司行業(yè)資訊頻道，感謝各位的閱讀！
另外有需要云服務器可以了解下創(chuàng)新互聯(lián)scvps.cn，海內(nèi)外云服務器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡單易用、服務可用性高、性價比高”等特點與優(yōu)勢，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應用場景需求。
