From 9e6a2c56be156fe4719e4fdc1c919614218ad6de Mon Sep 17 00:00:00 2001
From: UpstreamData <brett@upstreamdata.ca>
Date: Tue, 12 Oct 2021 16:13:21 -0600
Subject: [PATCH] added some basic config changing functions

---
 main.py            |  2 +-
 miners/bosminer.py | 41 +++++++++++++++++++++++++++++++++++------
 2 files changed, 36 insertions(+), 7 deletions(-)

diff --git a/main.py b/main.py
index 175380d0..921209b5 100644
--- a/main.py
+++ b/main.py
@@ -8,7 +8,7 @@ async def main():
     data = await miner_network.scan_network_for_miners()
     await data[0].get_config()
     config = await data[0].set_config_format()
-    print(config['format'])
+    print(config)
 
 if __name__ == '__main__':
     asyncio.get_event_loop().run_until_complete(main())
diff --git a/miners/bosminer.py b/miners/bosminer.py
index c9e94d5b..914b40ed 100644
--- a/miners/bosminer.py
+++ b/miners/bosminer.py
@@ -9,7 +9,7 @@ class BOSminer(BaseMiner):
     def __init__(self, ip: str):
         api = BOSMinerAPI(ip)
         super().__init__(ip, api)
-        self.config = {}
+        self.config = None
 
     def __repr__(self):
         return f"BOSminer: {str(self.ip)}"
@@ -62,9 +62,38 @@ class BOSminer(BaseMiner):
                 async with sftp.open('/etc/bosminer.toml') as file:
                     toml_data = toml.loads(await file.read())
         self.config = toml_data
-        return toml_data
 
-    async def set_config_format(self):
-        self.config['format']['generator'] = 'upstream_data_configurator'
-        self.config['format']['timestamp'] = int(time.time())
-        return self.config
+    def update_config(self, config: dict):
+        self.config = config
+
+    async def change_config_format(self, update_config: bool = False):
+        if not self.config:
+            await self.get_config()
+        config = self.config
+        config['format']['generator'] = 'upstream_data_configurator'
+        config['format']['timestamp'] = int(time.time())
+        if update_config:
+            self.update_config(config)
+        return config
+
+    async def change_config_tuning(self, wattage: int, enabled: bool = True, update_config: bool = False):
+        if not self.config:
+            await self.get_config()
+        config = self.config
+        config['autotuning']['psu_power_limit'] = wattage
+        config['autotuning']['enabled'] = enabled
+        if update_config:
+            self.update_config(config)
+        return config
+
+    async def change_config_temp_ctrl(self, target: float = 80.0, hot: float = 100.0,
+                                      dangerous: float = 110.0, update_config: bool = False):
+        if not self.config:
+            await self.get_config()
+        config = self.config
+        config['target_temp'] = target
+        config['hot_temp'] = hot
+        config['dangerous_temp'] = dangerous
+        if update_config:
+            self.update_config(config)
+        return config