used web attribute and removed some redundant changes from the firmware.py

pyasic/miners/backends/auradine.py

@@ -15,7 +15,7 @@
 # ------------------------------------------------------------------------------
 import logging
 from enum import Enum
-import httpx
+import base64
 import aiofiles
 from typing import List, Optional
 
@@ -195,7 +195,7 @@ class Auradine(StockFirmware):
         for key in conf.keys():
             await self.web.send_command(command=key, **conf[key])
 
-    async def upgrade_firmware(self, url: str = None, file_path: str = None, version: str = "latest") -> bool:
+    async def upgrade_firmware(self, *, url: str = None, file_path: str = None, version: str = "latest") -> bool:
         """
         Upgrade the firmware of the Auradine device.
 
@@ -215,16 +215,23 @@ class Auradine(StockFirmware):
 
             if url:
                 # Download the firmware file from the URL
-                async with httpx.AsyncClient() as client:
+                response = await self.web.get(url)
-                    response = await client.get(url)
+                if response.status_code != 200:
-                    if response.status_code != 200:
+                    raise ValueError(f"Failed to download firmware from URL: {url}")
-                        raise ValueError(f"Failed to download firmware from URL: {url}")
+                upgrade_contents = response.content
-                    upgrade_contents = response.content
             else:
-                # read the fimware file contents from the local file path
+                # Read the firmware file contents from the local file path
                 async with aiofiles.open(file_path, "rb") as f:
                     upgrade_contents = await f.read()
 
+            # Encode the firmware contents in base64
+            encoded_contents = base64.b64encode(upgrade_contents).decode("utf-8")
+
+            # Upload the firmware file to the Auradine miner device
+            await self.ssh.send_command(
+                f"echo {encoded_contents} | base64 -d > /tmp/firmware.tar && sysupgrade /tmp/firmware.tar"
+            )
+
             logging.info("Firmware upgrade process completed successfully.")
             return True
         except Exception as e:
@@ -423,4 +430,4 @@ class Auradine(StockFirmware):
             try:
                 return rpc_summary["SUMMARY"][0]["Elapsed"]
             except LookupError:
-                pass
+                pass

pyasic/miners/device/firmware.py

@@ -21,47 +21,26 @@ from pyasic.miners.base import BaseMiner
 class StockFirmware(BaseMiner):
     firmware = MinerFirmware.STOCK
 
-    async def upgrade_firmware(self, *, url: str = None, version: str = "latest") -> bool:
-        return await super().upgrade_firmware(url=url, version=version)
-
 
 class BraiinsOSFirmware(BaseMiner):
     firmware = MinerFirmware.BRAIINS_OS
 
-    async def upgrade_firmware(self, *, url: str = None, version: str = "latest") -> bool:
-        return await super().upgrade_firmware(url=url, version=version)
-
 
 class VNishFirmware(BaseMiner):
     firmware = MinerFirmware.VNISH
 
-    async def upgrade_firmware(self, *, url: str = None, version: str = "latest") -> bool:
-        return await super().upgrade_firmware(url=url, version=version)
-
 
 class ePICFirmware(BaseMiner):
     firmware = MinerFirmware.EPIC
 
-    async def upgrade_firmware(self, *, url: str = None, version: str = "latest") -> bool:
-        return await super().upgrade_firmware(url=url, version=version)
-
 
 class HiveonFirmware(BaseMiner):
     firmware = MinerFirmware.HIVEON
 
-    async def upgrade_firmware(self, *, url: str = None, version: str = "latest") -> bool:
-        return await super().upgrade_firmware(url=url, version=version)
-
 
 class LuxOSFirmware(BaseMiner):
     firmware = MinerFirmware.LUXOS
 
-    async def upgrade_firmware(self, *, url: str = None, version: str = "latest") -> bool:
-        return await super().upgrade_firmware(url=url, version=version)
-
 
 class MaraFirmware(BaseMiner):
-    firmware = MinerFirmware.MARATHON
+    firmware = MinerFirmware.MARATHON
-
-    async def upgrade_firmware(self, *, url: str = None, version: str = "latest") -> bool:
-        return await super().upgrade_firmware(url=url, version=version)