From e41ea4207412276eb99c2685b65b7076cfd74bc5 Mon Sep 17 00:00:00 2001
From: Wauter <Wauter@users.noreply.github.com>
Date: Sun, 12 Jul 2020 00:55:30 +0200
Subject: [PATCH] feature request 398 add 'hide timestamp' option for version
 text sensor (#1085)

Co-authored-by: Guillermo Ruffino <glm.net@gmail.com>
---
 esphome/components/version/text_sensor.py          |  6 ++++--
 esphome/components/version/version_text_sensor.cpp | 10 ++++++++--
 esphome/components/version/version_text_sensor.h   |  4 ++++
 esphome/const.py                                   |  1 +
 tests/test1.yaml                                   |  3 +++
 5 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/esphome/components/version/text_sensor.py b/esphome/components/version/text_sensor.py
index 21044bb89f..01cf8ba30b 100644
--- a/esphome/components/version/text_sensor.py
+++ b/esphome/components/version/text_sensor.py
@@ -1,14 +1,15 @@
 import esphome.codegen as cg
 import esphome.config_validation as cv
 from esphome.components import text_sensor
-from esphome.const import CONF_ID, CONF_ICON, ICON_NEW_BOX
+from esphome.const import CONF_ID, CONF_ICON, ICON_NEW_BOX, CONF_HIDE_TIMESTAMP
 
 version_ns = cg.esphome_ns.namespace('version')
 VersionTextSensor = version_ns.class_('VersionTextSensor', text_sensor.TextSensor, cg.Component)
 
 CONFIG_SCHEMA = text_sensor.TEXT_SENSOR_SCHEMA.extend({
     cv.GenerateID(): cv.declare_id(VersionTextSensor),
-    cv.Optional(CONF_ICON, default=ICON_NEW_BOX): text_sensor.icon
+    cv.Optional(CONF_ICON, default=ICON_NEW_BOX): text_sensor.icon,
+    cv.Optional(CONF_HIDE_TIMESTAMP, default=False): cv.boolean
 }).extend(cv.COMPONENT_SCHEMA)
 
 
@@ -16,3 +17,4 @@ def to_code(config):
     var = cg.new_Pvariable(config[CONF_ID])
     yield text_sensor.register_text_sensor(var, config)
     yield cg.register_component(var, config)
+    cg.add(var.set_hide_timestamp(config[CONF_HIDE_TIMESTAMP]))
diff --git a/esphome/components/version/version_text_sensor.cpp b/esphome/components/version/version_text_sensor.cpp
index 6aedfdedcd..fa8e6a9d01 100644
--- a/esphome/components/version/version_text_sensor.cpp
+++ b/esphome/components/version/version_text_sensor.cpp
@@ -8,9 +8,15 @@ namespace version {
 
 static const char *TAG = "version.text_sensor";
 
-void VersionTextSensor::setup() { this->publish_state(ESPHOME_VERSION " " + App.get_compilation_time()); }
+void VersionTextSensor::setup() {
+  if (this->hide_timestamp_) {
+    this->publish_state(ESPHOME_VERSION);
+  } else {
+    this->publish_state(ESPHOME_VERSION " " + App.get_compilation_time());
+  }
+}
 float VersionTextSensor::get_setup_priority() const { return setup_priority::DATA; }
-
+void VersionTextSensor::set_hide_timestamp(bool hide_timestamp) { this->hide_timestamp_ = hide_timestamp; }
 std::string VersionTextSensor::unique_id() { return get_mac_address() + "-version"; }
 void VersionTextSensor::dump_config() { LOG_TEXT_SENSOR("", "Version Text Sensor", this); }
 
diff --git a/esphome/components/version/version_text_sensor.h b/esphome/components/version/version_text_sensor.h
index cc798939ef..9355e78442 100644
--- a/esphome/components/version/version_text_sensor.h
+++ b/esphome/components/version/version_text_sensor.h
@@ -8,10 +8,14 @@ namespace version {
 
 class VersionTextSensor : public text_sensor::TextSensor, public Component {
  public:
+  void set_hide_timestamp(bool hide_timestamp);
   void setup() override;
   void dump_config() override;
   float get_setup_priority() const override;
   std::string unique_id() override;
+
+ protected:
+  bool hide_timestamp_{false};
 };
 
 }  // namespace version
diff --git a/esphome/const.py b/esphome/const.py
index 61ac2563ed..1760db0dab 100644
--- a/esphome/const.py
+++ b/esphome/const.py
@@ -201,6 +201,7 @@ CONF_HEAT_ACTION = 'heat_action'
 CONF_HEAT_MODE = 'heat_mode'
 CONF_HEATER = 'heater'
 CONF_HIDDEN = 'hidden'
+CONF_HIDE_TIMESTAMP = 'hide_timestamp'
 CONF_HIGH = 'high'
 CONF_HIGH_VOLTAGE_REFERENCE = 'high_voltage_reference'
 CONF_HOUR = 'hour'
diff --git a/tests/test1.yaml b/tests/test1.yaml
index 5d8108c4be..63320dfe29 100644
--- a/tests/test1.yaml
+++ b/tests/test1.yaml
@@ -1749,6 +1749,9 @@ text_sensor:
     name: "BSSID"
   mac_address:
     name: "Mac Address"
+- platform: version
+  name: "ESPHome Version No Timestamp"
+  hide_timestamp: True
 
 sn74hc595:
   - id: 'sn74hc595_hub'