Skip to content

GitLab

Open
Created by Dan Jones@danjonOwner
Report abuse

Draft: Resolve "Add websockets"

Viewing commit 7a4fd2dd
2 files
+ 57
- 0
Show latest version
client_socke‎t_receive.py‎ +28 -0
client_soc‎ket_send.py‎ +29 -0
  • 7a4fd2dd
    Dan Jones
    test: scripts for testing ws send receive · 7a4fd2dd
    Dan Jones authored 
    Enables sending and receiving messages as a
websocket client.

- use get_token.py to create SOAR_TOKEN env var
- pipenv run python client_socket_send.py
- pipenv run python client_socket_receive.py
client_socket_receive.py
+ 28
- 0
#!/usr/bin/env python
import asyncio
import os
import websockets
token = os.getenv('SOAR_TOKEN')
hauth = {'Authorization': f'Bearer {token}'}
async def receive():
async with websockets.connect("ws://localhost:8087/ws", extra_headers=hauth) as ws:
print("Connection established")
cid = await ws.recv()
print(f"ClientID: {cid}")
while True:
try:
result = await ws.recv()
if result:
print("Received: {}".format(result))
except KeyboardInterrupt:
ws.close()
print("Connection closed")
exit(0)
if __name__ == "__main__":
asyncio.run(receive())
\ No newline at end of file
client_socket_send.py
+ 29
- 0
#!/usr/bin/env python
import asyncio
import os
import websockets
token = os.getenv('SOAR_TOKEN')
hauth = {'Authorization': f'Bearer {token}'}
async def send():
async with websockets.connect("ws://localhost:8087/ws", extra_headers=hauth) as ws:
print("Connection established")
cid = await ws.recv()
print(f"ClientID: {cid}")
while True:
try:
msg = input("Message: ")
# await ws.send("@a:{}:{}".format(cid, msg))
await ws.send(msg)
except KeyboardInterrupt:
ws.close()
print("Connection closed")
exit(0)
if __name__ == "__main__":
asyncio.run(send())