bugfix
This commit is contained in:
parent
5b5ac52294
commit
63caea7087
31
ahserver/websocketProcessor.py
Normal file
31
ahserver/websocketProcessor.py
Normal file
@ -0,0 +1,31 @@
|
||||
import aiohttp
|
||||
from aiohttp import web
|
||||
from .baseProcessor import PythonScriptProcessor
|
||||
|
||||
class WebSocketProcessor(PythonScriptProcessor):
|
||||
@classmethod
|
||||
def isMe(self,name):
|
||||
return name=='ws'
|
||||
|
||||
async def path_call(self, request,params={}):
|
||||
await self.set_run_env(request)
|
||||
lenv = self.run_ns.copy()
|
||||
lenv.update(params)
|
||||
del lenv['request']
|
||||
txt = self.loadScript(self.real_path)
|
||||
exec(txt,lenv,lenv)
|
||||
func = lenv['myfunc']
|
||||
ws = web.WebSocketResponse()
|
||||
await ws.prepare(request)
|
||||
async for msg in ws:
|
||||
if msg.type == aiohttp/WSMsgType.TEXT:
|
||||
if msg.data == 'close':
|
||||
await ws.close()
|
||||
else:
|
||||
lenv['ws_data'] = msg.data
|
||||
resp = await func(request,**lenv)
|
||||
await ws.send_str(resp)
|
||||
elif msg.type == aiohttp.WSMsgType.ERROR:
|
||||
print('ws connection closed with exception %s' % ws.exception())
|
||||
return ws
|
||||
|
Loading…
Reference in New Issue
Block a user