vbatts/shadowsocks
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Use space instead of tab for indent

Browse source 
see PEP8
This commit is contained in:
noisyfox 2017-01-07 18:50:12 +11:00
parent b367ad0560
commit fd9214b334
1 changed files with 267 additions and 258 deletions
Download patch file Download diff file Expand all files Collapse all files

29
mujson_mgr.py
View file

@ -10,6 +10,7 @@ import sys
import json
import base64
class MuJsonLoader(object):
def __init__(self):
self.json = None
@ -25,6 +26,7 @@ class MuJsonLoader(object):
f.write(output)
f.truncate()
class MuMgr(object):
def __init__(self):
self.config_path = get_config().MUDB_FILE
@ -37,7 +39,7 @@ class MuMgr(object):
if self.server_addr == '127.0.0.1':
self.server_addr = self.getipaddr()
def getipaddr(self, ifname = 'eth0'):
def getipaddr(self, ifname='eth0'):
import socket
import struct
import fcntl
@ -59,8 +61,11 @@ class MuMgr(object):
obfs = user.get('obfs', '')
protocol = protocol.replace("_compatible", "")
obfs = obfs.replace("_compatible", "")
link = "%s:%s:%s:%s:%s:%s" % (self.server_addr, user['port'], protocol, user['method'], obfs, common.to_str(base64.urlsafe_b64encode(common.to_bytes(user['passwd']))).replace("=", ""))
return "ssr://" + ( encode and common.to_str(base64.urlsafe_b64encode(common.to_bytes(link))).replace("=", "") or link)
link = "%s:%s:%s:%s:%s:%s" % (self.server_addr, user['port'], protocol, user['method'], obfs,
common.to_str(base64.urlsafe_b64encode(common.to_bytes(user['passwd']))).replace(
"=", ""))
return "ssr://" + (
encode and common.to_str(base64.urlsafe_b64encode(common.to_bytes(link))).replace("=", "") or link)
def userinfo(self, user):
ret = ""
@ -68,18 +73,18 @@ class MuMgr(object):
if key in ['enable']:
continue
ret += '\n'
if key in ['transfer_enable', 'u', 'd'] :
if key in ['transfer_enable', 'u', 'd']:
val = user[key]
if val / 1024 < 4:
ret += " %s : %s" % (key, val)
elif val / 1024**2 < 4:
elif val / 1024 ** 2 < 4:
val /= float(1024)
ret += " %s : %s K Bytes" % (key, val)
elif val / 1024**3 < 4:
val /= float(1024**2)
elif val / 1024 ** 3 < 4:
val /= float(1024 ** 2)
ret += " %s : %s M Bytes" % (key, val)
else:
val /= float(1024**3)
val /= float(1024 ** 3)
ret += " %s : %s G Bytes" % (key, val)
else:
ret += " %s : %s" % (key, user[key])
@ -88,7 +93,9 @@ class MuMgr(object):
return ret
def rand_pass(self):
return ''.join([random.choice('''ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789~-_=+(){}[]^&%$@''') for i in range(8)])
return ''.join(
[random.choice('''ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789~-_=+(){}[]^&%$@''') for i
in range(8)])
def add(self, user):
up = {'enable': 1, 'u': 0, 'd': 0, 'method': "aes-128-cfb",
@ -172,6 +179,7 @@ class MuMgr(object):
if match:
print("### user [%s] info %s" % (row['user'], self.userinfo(row)))
def print_server_help():
print('''usage: python mujson_manage.py -a|-d|-e|-c|-l [OPTION]...
@ -198,6 +206,7 @@ General options:
-h, --help show this help message and exit
''')
def main():
shortopts = 'adeclu:p:k:O:o:G:g:m:t:f:h'
longopts = ['help']
@ -311,6 +320,6 @@ def main():
elif action is None:
print_server_help()
if __name__ == '__main__':
main()