加入参数方便客户端一键执行

This commit is contained in:
ubuntu 2018-07-27 11:56:13 +08:00
parent b71ac73d75
commit 76d0851fba
2 changed files with 23 additions and 0 deletions

@ -18,6 +18,7 @@ import string
import math import math
import re import re
import os import os
import sys
import json import json
import subprocess import subprocess
import collections import collections
@ -159,6 +160,17 @@ def get_network(ip_version):
return False return False
if __name__ == '__main__': if __name__ == '__main__':
for argc in sys.argv:
if 'SERVER' in argc:
SERVER = argc.split('SERVER=')[-1]
elif 'PORT' in argc:
PORT = int(argc.split('PORT=')[-1])
elif 'USER' in argc:
USER = argc.split('USER=')[-1]
elif 'PASSWORD' in argc:
PASSWORD = argc.split('PASSWORD=')[-1]
elif 'INTERVAL' in argc:
INTERVAL = int(argc.split('INTERVAL=')[-1])
socket.setdefaulttimeout(30) socket.setdefaulttimeout(30)
while 1: while 1:
try: try:

@ -127,6 +127,17 @@ def get_network(ip_version):
return False return False
if __name__ == '__main__': if __name__ == '__main__':
for argc in sys.argv:
if 'SERVER' in argc:
SERVER = argc.split('SERVER=')[-1]
elif 'PORT' in argc:
PORT = int(argc.split('PORT=')[-1])
elif 'USER' in argc:
USER = argc.split('USER=')[-1]
elif 'PASSWORD' in argc:
PASSWORD = argc.split('PASSWORD=')[-1]
elif 'INTERVAL' in argc:
INTERVAL = int(argc.split('INTERVAL=')[-1])
socket.setdefaulttimeout(30) socket.setdefaulttimeout(30)
while 1: while 1:
try: try: