You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
4.4 KiB
122 lines
4.4 KiB
#! /usr/bin/python3
|
|
# coding=utf-8
|
|
|
|
import requests
|
|
import json
|
|
import sys
|
|
from pyzabbix import ZabbixMetric, ZabbixSender
|
|
import os
|
|
import re
|
|
|
|
|
|
class AwZabbix:
|
|
def __init__(self, server, url, host):
|
|
self.url = url
|
|
self.server = server
|
|
self.host = host
|
|
self.data = ''
|
|
self.mult_dict = {'G': 1000000000, 'M': 1000000, 'T': 1000000000000, 'K': 1000, 'N': 0, }
|
|
self.packet = []
|
|
self.lld_data = []
|
|
|
|
def get_data(self):
|
|
if len(self.data) == 0:
|
|
self.data = requests.get(self.url).json()
|
|
|
|
def hashrate_str_to_int(self, hashrate_raw):
|
|
if "H" in hashrate_raw:
|
|
m = hashrate_raw[-4:-3]
|
|
if m[0] == ' ':
|
|
hashrate_mult = 1
|
|
else:
|
|
hashrate_mult = self.mult_dict[hashrate_raw[-4:-3]]
|
|
hashrate_raw = "".join(hashrate_raw.split())
|
|
a = float(hashrate_raw[:-5].replace(',', '.')) * hashrate_mult
|
|
else:
|
|
a = 0
|
|
return int(a)
|
|
|
|
def make_packet(self):
|
|
self.get_data()
|
|
if len(self.packet) != 0:
|
|
return self.packet
|
|
total_hashrate_raw = self.data['totalHashrate5s']
|
|
if total_hashrate_raw:
|
|
total_hashrate = self.hashrate_str_to_int(total_hashrate_raw)
|
|
self.packet.append(ZabbixMetric(self.host, 'total[hashrate]', total_hashrate))
|
|
for group in self.data['groupList']:
|
|
for miner in group['minerList']:
|
|
hashrate_raw = miner['speedInfo']['hashrate']
|
|
if hashrate_raw:
|
|
hashrate = self.hashrate_str_to_int(hashrate_raw)
|
|
else:
|
|
hashrate = 0
|
|
hashrate_avg_raw = miner['speedInfo']['avgHashrate']
|
|
if hashrate_avg_raw:
|
|
hashrate_avg = self.hashrate_str_to_int(hashrate_avg_raw)
|
|
else:
|
|
hashrate_avg = 0
|
|
minername = str(miner['name'])
|
|
self.packet.append(ZabbixMetric(self.host, "{}[{}]".format('hashrate', minername), hashrate))
|
|
self.packet.append(ZabbixMetric(self.host, "{}[{}]".format('hashrate_avg', minername), hashrate_avg))
|
|
self.packet.append(ZabbixMetric(self.host, "{}[{}]".format('hashrate_avg_min', minername), int(hashrate_avg * 0.8)))
|
|
self.packet.append(ZabbixMetric(self.host, "{}[{}]".format('ip', minername), miner['hostname']))
|
|
self.packet.append(ZabbixMetric(self.host, "{}[{}]".format('group', minername), miner['groupId']))
|
|
self.packet.append(ZabbixMetric(self.host, "{}[{}]".format('name', minername), miner['name']))
|
|
self.packet.append(ZabbixMetric(self.host, "{}[{}]".format('temp', minername),
|
|
miner['temperature'][:-3]))
|
|
self.packet.append(ZabbixMetric(self.host, "{}[{}]".format('status', minername),
|
|
miner['statusInfo']['statusDisplay']))
|
|
return self.packet
|
|
|
|
def send_packet(self):
|
|
if len(self.packet) == 0:
|
|
self.make_packet()
|
|
return ZabbixSender(self.server).send(self.packet)
|
|
|
|
def lld(self):
|
|
self.get_data()
|
|
if len(self.lld_data) != 0:
|
|
return self.lld_data
|
|
self.lld_data = {}
|
|
jdat = []
|
|
for group in self.data['groupList']:
|
|
for miner in group['minerList']:
|
|
jdat.append({"{#MINERID}": str(miner['id']),
|
|
"{#IP}": str(miner['hostname']),
|
|
"{#MINERNAME}": str(miner['name'])})
|
|
self.lld_data = {'data': jdat}
|
|
return self.lld_data
|
|
|
|
|
|
def read_conf():
|
|
path = os.path.dirname(os.path.realpath(__file__))
|
|
conf_file = os.path.join(path, 'config.json')
|
|
with open(conf_file) as f:
|
|
conf = json.load(f)
|
|
return conf
|
|
|
|
|
|
def main():
|
|
conf = read_conf()
|
|
zabbix_server = conf["zabbix_server"]
|
|
aw_url = conf["aw_url"]
|
|
hostname = conf["hostname"]
|
|
|
|
z = AwZabbix(zabbix_server, aw_url, hostname)
|
|
|
|
if len(sys.argv) == 1:
|
|
print(json.dumps(z.lld(), indent=4, sort_keys=True))
|
|
elif len(sys.argv) == 2 and sys.argv[1] == 'send':
|
|
z.send_packet()
|
|
print(1)
|
|
elif len(sys.argv) == 2 and sys.argv[1] == 'send_debug':
|
|
print(z.send_packet())
|
|
for i in z.packet:
|
|
print(i)
|
|
else:
|
|
print('Wrong args.')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|