-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
64325a3
commit 4375cb5
Showing
7 changed files
with
386 additions
and
74 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,142 @@ | ||
| import discord | ||
| from discord.ext import tasks, commands | ||
| import asyncpg | ||
| import json | ||
| import requests | ||
| import imgkit | ||
| import GPUtil | ||
| from tabulate import tabulate | ||
| import numpy as np | ||
| import cv2 | ||
| import pyautogui | ||
| import os | ||
|
|
||
| # insert your bot key here, get one from discord | ||
| key = 'key' | ||
|
|
||
| # channel to send updates and notifications to | ||
| notification_channel = 825054534044090408 | ||
|
|
||
| # pool hash check, reboot if hash lower than desired_hash times_to_check times in a row | ||
| check_enabled = True | ||
| address = "696b18d7e003be5b4d1a66b981313e1959d69066" | ||
| desired_hash = 169000000 | ||
| low_hash = 0 | ||
| times_to_check = 3 | ||
| startup = True | ||
| worker_name = "worker001" # I don't think it's case sensitive | ||
|
|
||
| ping_data = { | ||
| "id": 1, | ||
| "jsonrpc": "2.0", | ||
| "method": "miner_getstatdetail" | ||
| } | ||
|
|
||
| data_json = json.dumps(ping_data) | ||
| payload = {'json_payload': data_json} | ||
|
|
||
| class MyClient(discord.Client): | ||
|
|
||
| async def on_ready(self): | ||
| global low_hash | ||
| low_hash = 0 | ||
| print('Logged on as', self.user) | ||
| channel = client.get_channel(notification_channel) | ||
| await channel.send(f'Reboot Detected!') | ||
|
|
||
| async def on_message(self, message): | ||
| # don't respond to ourselves | ||
| # if message.author == self.user: | ||
| # return | ||
|
|
||
| if message.content == f'{worker_name} reboot': | ||
| print("REBOOTING") | ||
| os.system("shutdown -t 0 -r") | ||
| return | ||
|
|
||
| if message.content == f'{worker_name} force reboot': | ||
| print("REBOOTING") | ||
| os.system("shutdown -t 0 -r -f") | ||
| return | ||
|
|
||
| if message.content == f'{worker_name} screenshot': | ||
| image = pyautogui.screenshot() | ||
| image = cv2.cvtColor(np.array(image),cv2.COLOR_RGB2BGR) | ||
|
|
||
| # writing it to the disk using opencv | ||
| cv2.imwrite("screen.png", image) | ||
| await message.channel.send(file=discord.File('screen.png')) | ||
| return | ||
|
|
||
|
|
||
| if message.content == f'{worker_name} ping': | ||
| r = requests.get('http://127.0.0.1:3333', data=payload) | ||
| html_file = open("temp.html", "w") | ||
| html_file.write(r.text) | ||
| html_file.close() | ||
| imgkit.from_file('temp.html', 'out.jpg') | ||
| await message.channel.send(file=discord.File('out.jpg')) | ||
|
|
||
| # Ethminer API doesn't report temperature for me for some reason | ||
| # this is a backup method (only works for nvidia) | ||
| await message.channel.send("="*40+"GPU Details"+"="*40) | ||
| gpus = GPUtil.getGPUs() | ||
| list_gpus = [] | ||
| for gpu in gpus: | ||
| gpu_id = gpu.id | ||
| gpu_name = gpu.name | ||
| gpu_load = f"{gpu.load*100}%" | ||
| gpu_temperature = f"{gpu.temperature} °C" | ||
| list_gpus.append([ | ||
| gpu_name, gpu_temperature, gpu_load]) | ||
|
|
||
| output = ("```" + "\n\n" + tabulate(list_gpus, tablefmt="plain", headers=["GPU", "Temps","Load"]) + "```") | ||
| await message.channel.send(output) | ||
| return | ||
|
|
||
|
|
||
| @tasks.loop(minutes = 10) | ||
| async def check_hashrate(): | ||
| global low_hash, startup | ||
| channel = client.get_channel(notification_channel) | ||
| try: | ||
| r = requests.get(f'https://api.ethermine.org/miner/:{address}/workers') | ||
| found_worker = False | ||
|
|
||
| for worker in r.json()["data"]: | ||
|
|
||
| if worker["worker"] == worker_name: | ||
|
|
||
| if worker["reportedHashrate"] < desired_hash: | ||
| low_hash+=1 | ||
| current_hash = worker["reportedHashrate"] | ||
| print(f"low hash detected: {current_hash}, {low_hash} times") | ||
| if low_hash >= times_to_check: | ||
| await channel.send(f"{worker_name} hash too low: {current_hash}") | ||
| await channel.send("screenshot") | ||
| await channel.send("ping") | ||
| os.system("shutdown -t 10 -r") | ||
| return | ||
| else: | ||
| low_hash = 0 | ||
|
|
||
| found_worker = True | ||
| startup = False | ||
|
|
||
| if found_worker == False and startup == False: | ||
| await channel.send("worker not found on pool, rebooting") | ||
| await channel.send("screenshot") | ||
| await channel.send("ping") | ||
| os.system("shutdown -t 10 -r") | ||
| elif startup == True: | ||
| startup = False | ||
|
|
||
| except requests.ConnectionError: | ||
| print("error, no internet") | ||
|
|
||
| if check_enabled: | ||
| check_hashrate.add_exception_type(asyncpg.PostgresConnectionError) | ||
| check_hashrate.start() | ||
|
|
||
| client = MyClient() | ||
| client.run(key) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.