Added option to set state attributes when templating entity states

gadgetchnnel · Nov 25, 2019 · 4f62db1 · 4f62db1
1 parent af4357e
commit 4f62db1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/custom_card.json b/custom_card.json
@@ -1,6 +1,6 @@
 {
     "lovelace-card-templater": {
-      "version": "0.0.5",
+      "version": "0.0.6b1",
       "remote_location": "https://raw.githubusercontent.com/gadgetchnnel/lovelace-card-templater/master/lovelace-card-templater.js",
       "visit_repo": "https://github.com/gadgetchnnel/lovelace-card-templater",
       "changelog": "https://github.com/gadgetchnnel/lovelace-card-templater"

diff --git a/lovelace-card-templater.js b/lovelace-card-templater.js
@@ -26,7 +26,7 @@ customElements.whenDefined('card-tools').then(() => {
             this.yaml = window.YAML;
         });
       }
-
+       
       createRenderRoot() {
         return this;
       }
@@ -81,12 +81,14 @@ customElements.whenDefined('card-tools').then(() => {
         return false;
       }
 
-	 getMockedState(stateObj, state){
+	 getMockedState(stateObj, state, attributes){
   		var newStateObj = {};
   		Object.assign(newStateObj, stateObj);
   		newStateObj.attributes = {};
   		Object.assign(newStateObj.attributes, stateObj.attributes);
-
+  		if(attributes) {
+  			Object.assign(newStateObj.attributes, attributes);
+  		}
   		newStateObj.state = state;
 
   		return newStateObj;
@@ -99,7 +101,8 @@ customElements.whenDefined('card-tools').then(() => {
               let stateObj = this._hass.states[entityConf.entity];
               if(stateObj) {
               	let state = await this.applyTemplate(entityConf.state_template);
-              	let mockState = this.getMockedState(stateObj, state);
+              	let attributes = entityConf.attributes ? await this.getCardConfig(entityConf.attributes) : null;
+              	let mockState = this.getMockedState(stateObj, state, attributes);
                 this._mockHass.states[entityConf.entity] = mockState;
               }
             }