Merge pull request #467 from WillCodeForCats/new-entity-checks

Handle available TypeError in number platform
WillCodeForCats · Nov 12, 2023 · aa16ca9 · aa16ca9
2 parents 3e358db + 045d18b
commit aa16ca9
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/custom_components/solaredge_modbus_multi/number.py b/custom_components/solaredge_modbus_multi/number.py
@@ -137,7 +137,7 @@ def available(self) -> bool:
                 "ac_charge_policy"
             ] in [2, 3]
 
-        except KeyError:
+        except (TypeError, KeyError):
             return False
 
     @property
@@ -215,7 +215,7 @@ def available(self) -> bool:
 
             return super().available
 
-        except KeyError:
+        except (TypeError, KeyError):
             return False
 
     @property
@@ -270,7 +270,7 @@ def available(self) -> bool:
                 and self._platform.decoded_storage_control["control_mode"] == 4
             )
 
-        except KeyError:
+        except (TypeError, KeyError):
             return False
 
     @property
@@ -321,7 +321,7 @@ def available(self) -> bool:
                 and self._platform.decoded_storage_control["control_mode"] == 4
             )
 
-        except KeyError:
+        except (TypeError, KeyError):
             return False
 
     @property
@@ -378,7 +378,7 @@ def available(self) -> bool:
                 and self._platform.decoded_storage_control["control_mode"] == 4
             )
 
-        except KeyError:
+        except (TypeError, KeyError):
             return False
 
     @property
@@ -432,7 +432,7 @@ def available(self) -> bool:
                 or (int(self._platform.decoded_model["E_Lim_Ctl_Mode"]) >> 2) & 1
             )
 
-        except KeyError:
+        except (TypeError, KeyError):
             return False
 
     @property
@@ -481,7 +481,7 @@ def available(self) -> bool:
                 and (int(self._platform.decoded_model["E_Lim_Ctl_Mode"]) >> 10) & 1
             )
 
-        except KeyError:
+        except (TypeError, KeyError):
             return False
 
     @property
@@ -538,7 +538,7 @@ def available(self) -> bool:
 
             return super().available
 
-        except KeyError:
+        except (TypeError, KeyError):
             return False
 
     @property
@@ -592,7 +592,7 @@ def available(self) -> bool:
 
             return super().available
 
-        except KeyError:
+        except (TypeError, KeyError):
             return False
 
     @property