mirror of
https://github.com/selkies-project/selkies.git
synced 2025-09-04 10:54:06 +00:00
31 lines
946 B
Python
Executable File
31 lines
946 B
Python
Executable File
from aiohttp import web
|
|
import pathlib
|
|
import argparse
|
|
|
|
async def websocket_handler(request):
|
|
ws = web.WebSocketResponse()
|
|
await ws.prepare(request)
|
|
async for msg in ws:
|
|
if msg.type == web.WSMsgType.TEXT:
|
|
await ws.send_str("Echo: " + msg.data)
|
|
elif msg.type == web.WSMsgType.CLOSED:
|
|
break
|
|
return ws
|
|
|
|
app = web.Application()
|
|
app.router.add_get("/ws", websocket_handler)
|
|
|
|
# Serve static files
|
|
current_path = pathlib.Path(__file__).parent
|
|
static_path = current_path / "static"
|
|
print(static_path)
|
|
app.router.add_static("/", path=static_path, name="static")
|
|
|
|
if __name__ == '__main__':
|
|
# Parse command-line arguments
|
|
parser = argparse.ArgumentParser(description='Run WebSocket server')
|
|
parser.add_argument('--port', type=int, default=8080,
|
|
help='Port to run the server on (default: 8080)')
|
|
args = parser.parse_args()
|
|
web.run_app(app, port=args.port)
|