我有一个名为ipmi_management_webserver.py
的WebSocket python文件,您会看到以下提示:
# coding: utf-8
import socket
import struct
import hashlib, base64
import threading
import time
connectionlist = {}
g_code_length = 0
g_header_length = 0
PRINT_FLAG = True
def get_datalength(msg):
global g_code_length
global g_header_length
g_code_length = msg[1] & 127
if g_code_length == 126:
g_code_length = struct.unpack('>H', msg[2:4])[0]
g_header_length = 8
elif g_code_length == 127:
g_code_length = struct.unpack('>Q', msg[2:10])[0]
g_header_length = 14
else:
g_header_length = 6
g_code_length = int(g_code_length)
return g_code_length
def parse_data(msg):
global g_code_length
g_code_length = msg[1] & 127
if g_code_length == 126:
g_code_length = struct.unpack('>H', msg[2:4])[0]
masks = msg[4:8]
data = msg[8:]
elif g_code_length == 127:
g_code_length = struct.unpack('>Q', msg[2:10])[0]
masks = msg[10:14]
data = msg[14:]
else:
masks = msg[2:6]
data = msg[6:]
en_bytes = b""
cn_bytes = []
for i, d in enumerate(data):
nv = chr(d ^ masks[i % 4])
nv_bytes = nv.encode()
nv_len = len(nv_bytes)
if nv_len == 1:
en_bytes += nv_bytes
else:
en_bytes += b'%s'
cn_bytes.append(ord(nv_bytes.decode()))
if len(cn_bytes) > 2:
cn_str = ""
clen = len(cn_bytes)
count = int(clen / 3)
for x in range(count):
i = x * 3
b = bytes([cn_bytes[i], cn_bytes[i + 1], cn_bytes[i + 2]])
cn_str += b.decode()
new = en_bytes.replace(b'%s%s%s', b'%s')
new = new.decode()
res = (new % tuple(list(cn_str)))
else:
res = en_bytes.decode()
return res
def sendMessage(msg):
global connectionlist
send_msg = b""
send_msg += b"\x81"
back_str = []
back_str.append('\x81')
data_length = len(msg.encode())
if PRINT_FLAG:
print("INFO: send message is %s and len is %d" % (msg, len(msg.encode('utf-8'))))
if data_length <= 125:
send_msg += str.encode(chr(data_length))
elif data_length <= 65535:
send_msg += struct.pack('b', 126)
send_msg += struct.pack('>h', data_length)
elif data_length <= (2 ^ 64 - 1):
send_msg += struct.pack('b', 127)
send_msg += struct.pack('>q', data_length)
else:
print(u'too long')
send_message = send_msg + msg.encode('utf-8')
for connection in connectionlist.values():
if send_message != None and len(send_message) > 0:
connection.send(send_message)
def deleteconnection(item):
global connectionlist
del connectionlist['connection' + item]
class WebSocket(threading.Thread):
def __init__(self, conn, index, name, remote, path=""):
# thread
threading.Thread.__init__(self)
self.conn = conn
self.index = index
self.name = name
self.remote = remote
self.path = path
self.GUID = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"
self.buffer = ""
self.buffer_utf8 = b""
self.length_buffer = 0
def generate_token(self, WebSocketKey):
WebSocketKey = WebSocketKey + self.GUID
Ser_WebSocketKey = hashlib.sha1(WebSocketKey.encode(encoding='utf-8')).digest()
WebSocketToken = base64.b64encode(Ser_WebSocketKey)
return WebSocketToken.decode('utf-8')
def run(self):
if PRINT_FLAG:
print('Socket %s Start!' % self.index)
global g_code_length
global g_header_length
self.handshaken = False
while True:
if self.handshaken == False:
if PRINT_FLAG:
print('INFO: Socket %s Start Handshaken with %s!' % (self.index, self.remote))
self.buffer = self.conn.recv(1024).decode(
'utf-8')
if PRINT_FLAG:
print("INFO: Socket %s self.buffer is {%s}" % (self.index, self.buffer))
if self.buffer.find('\r\n\r\n') != -1:
headers = {}
header, data = self.buffer.split('\r\n\r\n', 1)
for line in header.split("\r\n")[1:]:
key, value = line.split(": ", 1)
headers[key] = value
try:
WebSocketKey = headers["Sec-WebSocket-Key"]
except KeyError:
print("Socket %s Handshaken Failed!" % (self.index))
deleteconnection(str(self.index))
self.conn.close()
break
WebSocketToken = self.generate_token(WebSocketKey)
headers["Location"] = ("ws://%s%s" % (headers["Host"], self.path))
handshake = "HTTP/1.1 101 Switching Protocols\r\n" \
"Connection: Upgrade\r\n" \
"Sec-WebSocket-Accept: " + WebSocketToken + "\r\n" \
"Upgrade: websocket\r\n\r\n"
self.conn.send(handshake.encode(encoding='utf-8'))
self.handshaken = True
sendMessage("Welocomg " + self.name + " !")
g_code_length = 0
else:
print("Socket %s Error2!" % (self.index))
deleteconnection(str(self.index))
self.conn.close()
break
else:
mm = self.conn.recv(128)
if g_code_length == 0:
get_datalength(mm)
self.length_buffer += len(mm)
self.buffer_utf8 += mm
if self.length_buffer - g_header_length < g_code_length:
if PRINT_FLAG:
print("INFO: finish")
continue
else:
if PRINT_FLAG:
print("g_code_length:", g_code_length)
print("INFO Line 204: Recv info %s,len %d:" % (self.buffer_utf8, len(self.buffer_utf8)))
if not self.buffer_utf8:
continue
recv_message = parse_data(self.buffer_utf8)
if recv_message == "quit":
print("Socket %s Logout!" % (self.index))
nowTime = time.strftime('%H:%M:%S', time.localtime(time.time()))
sendMessage("%s %s say: %s" % (nowTime, self.remote, self.name + " Logout"))
deleteconnection(str(self.index))
self.conn.close()
break
else:
nowTime = time.strftime('%H:%M:%S', time.localtime(time.time()))
sendMessage("%s %s say: %s" % (nowTime, self.remote, recv_message))
print('ldl:', recv_message)
g_code_length = 0
self.length_buffer = 0
self.buffer_utf8 = b""
# WebSocket
class WebSocketServer(object):
def __init__(self):
self.socket = None
self.i = 0
# begin
def begin(self):
if PRINT_FLAG:
print('WebSocketServer Start!')
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
ip = '127.0.0.1'
port = 65432
if PRINT_FLAG:
print("WebServer is listening %s,%d" % (ip, port))
self.socket.bind((ip, port))
self.socket.listen(50)
global connectionlist
while True:
connection, address = self.socket.accept()
newSocket = WebSocket(connection, self.i, address[0], address)
newSocket.start()
connectionlist['connection' + str(self.i)] = connection
self.i += 1
if __name__ == "__main__":
server = WebSocketServer()
server.begin()
我使用if __name__ == "__main__"
函数创建WebSocketServer,并且在我的测试环境中,我可以直接运行文件以初始化服务器:
if __name__ == "__main__":
server = WebSocketServer()
server.begin()
但是,在我的Django项目中,在我的分发环境中:
这是我项目的wsgi
python模块:
import os
from django.core.wsgi import get_wsgi_application
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "Qiy.settings")
import django
print("django.setup()")
django.setup()
from socketio import Middleware
from xxx.qiy_admin_website_chat.views import sio
django_app = get_wsgi_application()
application = Middleware(sio, django_app)
import eventlet
import eventlet.wsgi
eventlet.wsgi.server(eventlet.listen(('', 8000)), application)
我使用命令来运行django项目:
SCREEN -d -m python3 -m Qiy.wsgi
现在,如何在我的分布式项目中运行ipmi_management_webserver.py
?