1#!/usr/bin/env python3 2# -*- coding: utf-8 -*- 3#*************************************************************************** 4# _ _ ____ _ 5# Project ___| | | | _ \| | 6# / __| | | | |_) | | 7# | (__| |_| | _ <| |___ 8# \___|\___/|_| \_\_____| 9# 10# Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al. 11# 12# This software is licensed as described in the file COPYING, which 13# you should have received as part of this distribution. The terms 14# are also available at https://curl.se/docs/copyright.html. 15# 16# You may opt to use, copy, modify, merge, publish, distribute and/or sell 17# copies of the Software, and permit persons to whom the Software is 18# furnished to do so, under the terms of the COPYING file. 19# 20# This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY 21# KIND, either express or implied. 22# 23# SPDX-License-Identifier: curl 24# 25########################################################################### 26# 27import argparse 28import asyncio 29import logging 30from asyncio import IncompleteReadError 31 32from websockets import server 33from websockets.exceptions import ConnectionClosedError 34 35 36async def echo(websocket): 37 try: 38 async for message in websocket: 39 await websocket.send(message) 40 except ConnectionClosedError: 41 pass 42 43 44async def run_server(port): 45 async with server.serve(echo, "localhost", port): 46 await asyncio.Future() # run forever 47 48 49def main(): 50 parser = argparse.ArgumentParser(prog='scorecard', description=""" 51 Run a websocket echo server. 52 """) 53 parser.add_argument("--port", type=int, 54 default=9876, help="port to listen on") 55 args = parser.parse_args() 56 57 logging.basicConfig( 58 format="%(asctime)s %(message)s", 59 level=logging.DEBUG, 60 ) 61 62 asyncio.run(run_server(args.port)) 63 64 65if __name__ == "__main__": 66 main() 67
