eswyft/msshpd
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

support json

Browse Source
This commit is contained in:
Evgeny Zinoviev 2021-08-08 21:33:52 +03:00
parent dbc9e6b36b
commit dc0b151491
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
msshpd
View File

@ -7,6 +7,7 @@ from argparse import ArgumentParser
routes = web.RouteTableDef() routes = web.RouteTableDef()
mss_host = None mss_host = None
mss_port = None mss_port = None
mss_json_response = False
class MySimpleSocketClient: class MySimpleSocketClient:
@ -41,7 +42,11 @@ async def variable_handler(request):
client.write(command) client.write(command)
try: try:
response = {'data': client.read()} raw = client.read()
if mss_json_response:
response = json.loads(raw)
else:
response = {'data': raw}
except Exception as e: except Exception as e:
response = {'error': str(e)} response = {'error': str(e)}
@ -52,12 +57,14 @@ if __name__ == '__main__':
parser = ArgumentParser() parser = ArgumentParser()
parser.add_argument('--mss-host', required=True, type=str) parser.add_argument('--mss-host', required=True, type=str)
parser.add_argument('--mss-port', required=True, type=int) parser.add_argument('--mss-port', required=True, type=int)
parser.add_argument('--mss-json-response', action='store_true')
parser.add_argument('--host', default='0.0.0.0', type=str) parser.add_argument('--host', default='0.0.0.0', type=str)
parser.add_argument('--port', default=8080, type=int) parser.add_argument('--port', default=8080, type=int)
args = parser.parse_args() args = parser.parse_args()
mss_host = args.mss_host mss_host = args.mss_host
mss_port = args.mss_port mss_port = args.mss_port
mss_json_response = args.mss_json_response
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
level=logging.INFO) level=logging.INFO)