1 files changed, 7 insertions, 4 deletions
diff --git a/pi_temp_pid.py b/pi_temp_pid.py
index 8f22694..a20f8b0 100644
--- a/pi_temp_pid.py
+++ b/pi_temp_pid.py
@@ -149,6 +149,10 @@ if not os.path.exists(CONFIG_FILE):
         sys.exit()
 
 
+class PidValue(BaseModel):
+    value: int = Field(..., ge=1, le=255)
+
+
 class ShroomboxSettings(BaseSettings):
     """
     Shroombox settings loaded from .env file.
@@ -270,12 +274,12 @@ class ShroomboxManager:
 
     def relay_switch(
             self,
-            on: bool,
+            on: bool = False,
     ) -> None:
         """
         Turn relay on or off.
         :param on: bool
-        :return:
+        :return: None
         """
         if on:
             logger.info('Turning on relay %time_now()')
@@ -311,14 +315,13 @@ class ShroomboxManager:
 
     def mosfet_set(
             self,
-            value,
+            value: PidValue = 0,
     ) -> bool:
         """
         Set mosfet value.
         :param value: int
         :return: bool
         """
-        result = True
         logger.debug(f'Changing pin {settings.mosfet_pin} to {value}')
         try:
             self.gpio.write(settings.mosfet_pin, value)