84 lines
2.3 KiB
Python
84 lines
2.3 KiB
Python
from API.bmminer import BMMinerAPI
|
|
from API.bosminer import BOSMinerAPI
|
|
from API.cgminer import CGMinerAPI
|
|
from API.btminer import BTMinerAPI
|
|
from API.unknown import UnknownAPI
|
|
import ipaddress
|
|
import asyncssh
|
|
import logging
|
|
|
|
|
|
class BaseMiner:
|
|
def __init__(
|
|
self,
|
|
ip: str,
|
|
api: BMMinerAPI or BOSMinerAPI or CGMinerAPI or BTMinerAPI or UnknownAPI,
|
|
) -> None:
|
|
self.ip = ipaddress.ip_address(ip)
|
|
self.uname = None
|
|
self.pwd = None
|
|
self.api = api
|
|
self.api_type = None
|
|
self.model = None
|
|
self.light = None
|
|
self.nominal_chips = 1
|
|
|
|
async def _get_ssh_connection(self) -> asyncssh.connect:
|
|
"""Create a new asyncssh connection"""
|
|
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_file(self, src, dest):
|
|
async with (await self._get_ssh_connection()) as conn:
|
|
await asyncssh.scp(src, (conn, dest))
|
|
|
|
async def check_light(self):
|
|
return self.light
|
|
|
|
async def get_board_info(self):
|
|
return None
|
|
|
|
async def get_config(self):
|
|
return None
|
|
|
|
async def get_hostname(self):
|
|
return None
|
|
|
|
async def get_model(self):
|
|
return None
|
|
|
|
async def reboot(self):
|
|
return None
|
|
|
|
async def restart_backend(self):
|
|
return None
|
|
|
|
async def send_config(self, yaml_config):
|
|
return None
|