fixed temperature units

2 files changed, 7 insertions, 4 deletions

diff --git a/src/constants.rs b/src/constants.rs
index a67b6fe..ca4e53c 100644
--- a/src/constants.rs
+++ b/src/constants.rs
@@ -95,3 +95,7 @@ pub const HA_DEVICE_CLASS_BUTTON_UPDATE: &str = "update";
 
 pub const HA_DEVICE_CLASS_SWITCH_OUTLET: &str = "outlet";
 pub const HA_DEVICE_CLASS_SWITCH_SWITCH: &str = "switch";
+
+pub const HA_UNIT_TEMPERATURE_CELSIUS: &str = "°C";
+pub const HA_UNIT_TEMPERATURE_KELVIN: &str = "K";
+pub const HA_UNIT_TEMPERATURE_FAHRENHEIT: &str = "°F";
diff --git a/src/unit.rs b/src/unit.rs
index 4f3ca19..f0557da 100644
--- a/src/unit.rs
+++ b/src/unit.rs
@@ -8,11 +8,10 @@ pub enum TemperatureUnit {
 
 impl TemperatureUnit {
     pub fn as_str(&self) -> &'static str {
-        // TODO: improve
         match self {
-            TemperatureUnit::Celcius => "C",
-            TemperatureUnit::Kelvin => "k",
-            TemperatureUnit::Fahrenheit => "F",
+            TemperatureUnit::Celcius => crate::constants::HA_UNIT_TEMPERATURE_CELSIUS,
+            TemperatureUnit::Kelvin => crate::constants::HA_UNIT_TEMPERATURE_KELVIN,
+            TemperatureUnit::Fahrenheit => crate::constants::HA_UNIT_TEMPERATURE_FAHRENHEIT,
             TemperatureUnit::Other(other) => other,
         }
     }