【python】查看v2ray流量统计脚本

1
#!/bin/python3
2

3
# Created by Rehtt on 2020/12/11
4
# 查看v2ray多用户流量统计
5
# 使用前需要开启v2ray自带的统计工具，配置方法：https://guide.v2fly.org/advanced/traffic.html#%E9%85%8D%E7%BD%AE%E7%BB%9F%E8%AE%A1%E5%8A%9F%E8%83%BD
6
# 本脚本使用email作为用户标识，使用tag作为组标识
7

8
import re
9
import os
10
import json
11
import getopt
12
import sys
13

14
v2ConfigFile = '/etc/v2ray/config.json'  # v2ray配置文件位置
15
v2ctlServer = '127.0.0.1:53844'  # v2ctl api配置信息
16

17

18
def dataSize(size, i=0):
19
    if size > 1e3:
20
        size /= 1e3
21
        s = dataSize(size, i+1)
22
    else:
23
        size = round(size, 2)
24
        if i == 1:
25
            s = str(size)+'KB'
26
        elif i == 2:
27
            s = str(size)+'MB'
28
        elif i >= 3:
29
            s = str(size)+'GB'
30
    return s
31

32

33
def red(f):
34
    return '\033[31m'+f+'\033[0m'
35

36

37
def help():
38
    print("v2ray-flow.py [-h] [-g <tag>] [-c <file>] [-s <server>]")
39
    print("默认显示全部用户\n")
40
    print("-h,--help\t显示帮助")
41
    print("-g <tag>\t显示组的流量")
42
    print("\t-g A\t\t(显示tag:A组的流量)\n")
43
    # print("-u <email>\t显示单个用户流量")
44
    # print("\t-u rehtt@qq.com\t(显示email为rehtt@qq.com的流量)\n")
45
    print("-c <file>\t输入v2ray配置文件")
46
    print("\t-c /etc/v2ray/config.json\n")
47
    print("-s <server>\t输入v2ctl api server地址")
48
    print("\t-s 127.0.0.1:53844")
49

50

51
def getDownUp(name, res):
52
    i = 0
53
    z = 0
54
    down = 0
55
    up = 0
56
    try:
57
        for r in res:
58
            i += 1
59
            if name in r:
60
                z += 1
61
                if 'downlink' in r:
62
                    down = dataSize(float(re.sub('\D', '', res[i])))
63
                else:
64
                    up = dataSize(float(re.sub('\D', '', res[i])))
65
                if z == 2:
66
                    break
67
    except:
68
        pass
69
    return down, up
70

71

72
def main(argv):
73
    global v2ctlServer, v2ConfigFile
74
    c = '/usr/bin/v2ray/v2ctl api --server={server} StatsService.QueryStats \'pattern: "{q}" reset: false\''
75
    try:
76
        opts, args = getopt.getopt(argv, "hg:c:s:u:", ["help"])
77
    except getopt.GetoptError:
78
        help()
79
        sys.exit(2)
80
    q = 'user'
81
    G = 0
82
    for opt, arg in opts:
83
        if opt in ('-h', '--help'):
84
            help()
85
            sys.exit()
86
        elif opt == '-g':
87
            q = arg
88
            G = 1
89
        # elif opt == '-u':
90
        #     q = arg
91
        #     G = 2
92
        if opt == '-c':
93
            v2ConfigFile = arg
94
        if opt == '-s':
95
            v2ctlServer = arg
96
    tags = {}
97

98
    c = c.format(server=v2ctlServer, q=q)
99
    res = os.popen(c).readlines()
100
    if G == 0:
101
        with open(v2ConfigFile, 'r') as f:
102
            data = json.load(f)
103
            for i in data['inbounds']:
104
                if i['tag'] != 'api':
105
                    emails = []
106
                    for email in i['settings']['clients']:
107
                        emails.append(email['email'])
108
                    tags[i['tag']] = emails
109
        for tag in tags:
110
            for email in tags[tag]:
111
                down, up = getDownUp(email, res)
112
                print('Email:', red(email), '\tTag:'+red(tag))
113
                print('上传:', up, '\t下载:', down, '\n')
114
    else:
115
        if G==1:
116
            print('Tag:', red(q))
117
            down, up = getDownUp(q, res)
118
        # else:
119
        #     print('Email:', red(q))
120
        print('上传:', up, '\t下载:', down, '\n')
121

122

123

124
if __name__ == "__main__":
125
    main(sys.argv[1:])