1 parent dc0279c commit 775d642Copy full SHA for 775d642
1 file changed
pproxy/verbose.py
@@ -1,8 +1,12 @@
1
import time, sys, asyncio, functools
2
3
b2s = lambda i: f'{i/2**30:.1f}G' if i>=2**30 else f'{i/2**20:.1f}M' if i>=2**20 else f'{i/1024:.1f}K'
4
-def all_stat(stats):
+
5
+def all_stat_other(stats):
6
cmd = sys.stdin.readline()
7
+ all_stat(stats)
8
9
+def all_stat(stats):
10
if len(stats) <= 1:
11
print('no traffic')
12
return
@@ -52,8 +56,14 @@ def modstat(remote_ip, host_name, stats=args.stats):
52
56
tostat = (stats[0], stats.setdefault(remote_ip, {}).setdefault(host_name_2, [0]*6))
53
57
return lambda i: lambda s: [st.__setitem__(i, st[i] + s) for st in tostat]
54
58
args.modstat = modstat
59
+ def win_readline(handler):
60
+ while True:
61
+ line = sys.stdin.readline()
62
+ handler()
55
63
if args.v >= 2:
64
asyncio.ensure_future(realtime_stat(args.stats[0]))
- loop.add_reader(sys.stdin, functools.partial(all_stat, args.stats))
-
65
+ if sys.platform != 'win32':
66
+ loop.add_reader(sys.stdin, functools.partial(all_stat_other, args.stats))
67
+ else:
68
+ loop.run_in_executor(None, win_readline, functools.partial(all_stat, args.stats))
69
0 commit comments