89 lines
3.3 KiB
Python
89 lines
3.3 KiB
Python
from API.bmminer import BMMinerAPI
|
|
from miners import BaseMiner
|
|
import asyncssh
|
|
import logging
|
|
|
|
|
|
class BMMiner(BaseMiner):
|
|
def __init__(self, ip: str) -> None:
|
|
api = BMMinerAPI(ip)
|
|
super().__init__(ip, api)
|
|
self.model = None
|
|
self.config = None
|
|
self.uname = 'root'
|
|
self.pwd = 'admin'
|
|
|
|
def __repr__(self) -> str:
|
|
return f"BMMiner: {str(self.ip)}"
|
|
|
|
async def get_model(self):
|
|
if self.model:
|
|
logging.debug(f"Found model for {self.ip}: {self.model}")
|
|
return self.model
|
|
version_data = await self.api.devdetails()
|
|
if version_data:
|
|
self.model = version_data["DEVDETAILS"][0]["Model"].replace("Antminer ", "")
|
|
logging.debug(f"Found model for {self.ip}: {self.model}")
|
|
return self.model
|
|
logging.warning(f"Failed to get model for miner: {self}")
|
|
return None
|
|
|
|
async def get_hostname(self) -> str:
|
|
try:
|
|
async with (await self._get_ssh_connection()) as conn:
|
|
if conn is not None:
|
|
data = await conn.run('cat /proc/sys/kernel/hostname')
|
|
host = data.stdout.strip()
|
|
logging.debug(f"Found hostname for {self.ip}: {host}")
|
|
return host
|
|
else:
|
|
logging.warning(f"Failed to get hostname for miner: {self}")
|
|
return "?"
|
|
except Exception as e:
|
|
logging.warning(f"Failed to get hostname for miner: {self}")
|
|
return "?"
|
|
|
|
async def _get_ssh_connection(self) -> asyncssh.connect:
|
|
try:
|
|
conn = await asyncssh.connect(str(self.ip),
|
|
known_hosts=None,
|
|
username=self.uname,
|
|
password=self.pwd,
|
|
server_host_key_algs=['ssh-rsa'])
|
|
return conn
|
|
except asyncssh.misc.PermissionDenied:
|
|
try:
|
|
conn = await asyncssh.connect(str(self.ip),
|
|
known_hosts=None,
|
|
username="admin",
|
|
password="admin",
|
|
server_host_key_algs=['ssh-rsa'])
|
|
return conn
|
|
except Exception as e:
|
|
logging.warning(f"{self} raised an exception: {e}")
|
|
raise e
|
|
except OSError:
|
|
logging.warning(f"Connection refused: {self} ")
|
|
return None
|
|
except Exception as e:
|
|
logging.warning(f"{self} raised an exception: {e}")
|
|
raise e
|
|
|
|
async def send_ssh_command(self, cmd):
|
|
result = None
|
|
async with (await self._get_ssh_connection()) as conn:
|
|
for i in range(3):
|
|
try:
|
|
result = await conn.run(cmd)
|
|
except Exception as e:
|
|
logging.warning(f"{self} command {cmd} error: {e}")
|
|
if i == 3:
|
|
return
|
|
continue
|
|
return result
|
|
|
|
async def reboot(self) -> None:
|
|
logging.debug(f"{self}: Sending reboot command.")
|
|
await self.send_ssh_command("reboot")
|
|
logging.debug(f"{self}: Reboot command completed.")
|