Add support for multiple entities for attribute selector

This commit is contained in:
Paul Bottein 2025-07-17 14:23:29 +02:00
parent 77afb9ac1a
commit 81870d0e7d
No known key found for this signature in database
6 changed files with 74 additions and 48 deletions

View File

@ -6,6 +6,8 @@ import { computeAttributeNameDisplay } from "../../common/entity/compute_attribu
import type { HomeAssistant, ValueChangedEvent } from "../../types"; import type { HomeAssistant, ValueChangedEvent } from "../../types";
import "../ha-combo-box"; import "../ha-combo-box";
import type { HaComboBox } from "../ha-combo-box"; import type { HaComboBox } from "../ha-combo-box";
import { ensureArray } from "../../common/array/ensure-array";
import { fireEvent } from "../../common/dom/fire_event";
export type HaEntityPickerEntityFilterFunc = (entityId: HassEntity) => boolean; export type HaEntityPickerEntityFilterFunc = (entityId: HassEntity) => boolean;
@ -13,7 +15,7 @@ export type HaEntityPickerEntityFilterFunc = (entityId: HassEntity) => boolean;
class HaEntityAttributePicker extends LitElement { class HaEntityAttributePicker extends LitElement {
@property({ attribute: false }) public hass!: HomeAssistant; @property({ attribute: false }) public hass!: HomeAssistant;
@property({ attribute: false }) public entityId?: string; @property({ attribute: false }) public entityId?: string | string[];
/** /**
* List of attributes to be hidden. * List of attributes to be hidden.
@ -48,23 +50,37 @@ class HaEntityAttributePicker extends LitElement {
} }
protected updated(changedProps: PropertyValues) { protected updated(changedProps: PropertyValues) {
if (changedProps.has("_opened") && this._opened) { if (
const entityState = this.entityId (changedProps.has("_opened") && this._opened) ||
? this.hass.states[this.entityId] changedProps.has("entityId") ||
: undefined; changedProps.has("attribute")
(this._comboBox as any).items = entityState ) {
? Object.keys(entityState.attributes) const entityIds = this.entityId ? ensureArray(this.entityId) : [];
.filter((key) => !this.hideAttributes?.includes(key)) const options: { value: string; label: string }[] = [];
.map((key) => ({ const attributesSet = new Set<string>();
value: key,
for (const entityId of entityIds) {
const stateObj = this.hass.states[entityId];
const attributes = Object.keys(stateObj.attributes).filter(
(a) => !this.hideAttributes?.includes(a)
);
for (const a of attributes) {
if (!attributesSet.has(a)) {
attributesSet.add(a);
options.push({
value: a,
label: computeAttributeNameDisplay( label: computeAttributeNameDisplay(
this.hass.localize, this.hass.localize,
entityState, stateObj,
this.hass.entities, this.hass.entities,
key a
), ),
})) });
: []; }
}
}
(this._comboBox as any).filteredItems = options;
} }
} }
@ -73,21 +89,10 @@ class HaEntityAttributePicker extends LitElement {
return nothing; return nothing;
} }
const stateObj = this.hass.states[this.entityId!] as HassEntity | undefined;
return html` return html`
<ha-combo-box <ha-combo-box
.hass=${this.hass} .hass=${this.hass}
.value=${this.value .value=${this.value}
? stateObj
? computeAttributeNameDisplay(
this.hass.localize,
stateObj,
this.hass.entities,
this.value
)
: this.value
: ""}
.autofocus=${this.autofocus} .autofocus=${this.autofocus}
.label=${this.label ?? .label=${this.label ??
this.hass.localize( this.hass.localize(
@ -97,6 +102,7 @@ class HaEntityAttributePicker extends LitElement {
.required=${this.required} .required=${this.required}
.helper=${this.helper} .helper=${this.helper}
.allowCustomValue=${this.allowCustomValue} .allowCustomValue=${this.allowCustomValue}
item-id-path="value"
item-value-path="value" item-value-path="value"
item-label-path="label" item-label-path="label"
@opened-changed=${this._openedChanged} @opened-changed=${this._openedChanged}
@ -106,12 +112,28 @@ class HaEntityAttributePicker extends LitElement {
`; `;
} }
private get _value() {
return this.value || "";
}
private _openedChanged(ev: ValueChangedEvent<boolean>) { private _openedChanged(ev: ValueChangedEvent<boolean>) {
this._opened = ev.detail.value; this._opened = ev.detail.value;
} }
private _valueChanged(ev: ValueChangedEvent<string>) { private _valueChanged(ev: ValueChangedEvent<string>) {
this.value = ev.detail.value; ev.stopPropagation();
const newValue = ev.detail.value;
if (newValue !== this._value) {
this._setValue(newValue);
}
}
private _setValue(value: string) {
this.value = value;
setTimeout(() => {
fireEvent(this, "value-changed", { value });
fireEvent(this, "change");
}, 0);
} }
} }

View File

@ -32,7 +32,7 @@ class HaEntityStatePicker extends LitElement {
public allowCustomValue; public allowCustomValue;
@property({ attribute: false }) @property({ attribute: false })
public excludeStates?: string[]; public hideStates?: string[];
@property() public label?: string; @property() public label?: string;
@ -56,19 +56,19 @@ class HaEntityStatePicker extends LitElement {
changedProps.has("extraOptions") changedProps.has("extraOptions")
) { ) {
const entityIds = this.entityId ? ensureArray(this.entityId) : []; const entityIds = this.entityId ? ensureArray(this.entityId) : [];
const stateOptions: { value: string; label: string }[] = []; const options: { value: string; label: string }[] = [];
const statesSet = new Set<string>(); const statesSet = new Set<string>();
for (const entityId of entityIds) { for (const entityId of entityIds) {
const stateObj = this.hass.states[entityId]; const stateObj = this.hass.states[entityId];
const states = getStates(this.hass, stateObj, this.attribute).filter( const states = getStates(this.hass, stateObj, this.attribute).filter(
(s) => !this.excludeStates || !this.excludeStates.includes(s) (s) => !this.hideStates || !this.hideStates.includes(s)
); );
for (const s of states) { for (const s of states) {
if (!statesSet.has(s)) { if (!statesSet.has(s)) {
statesSet.add(s); statesSet.add(s);
const options = { options.push({
value: s, value: s,
label: !this.attribute label: !this.attribute
? this.hass.formatEntityState(stateObj, s) ? this.hass.formatEntityState(stateObj, s)
@ -77,15 +77,14 @@ class HaEntityStatePicker extends LitElement {
this.attribute, this.attribute,
s s
), ),
}; });
stateOptions.push(options);
} }
} }
} }
(this._comboBox as any).filteredItems = [ (this._comboBox as any).filteredItems = [
...(this.extraOptions ?? []), ...(this.extraOptions ?? []),
...stateOptions, ...options,
]; ];
} }
} }
@ -106,6 +105,7 @@ class HaEntityStatePicker extends LitElement {
.required=${this.required} .required=${this.required}
.helper=${this.helper} .helper=${this.helper}
.allowCustomValue=${this.allowCustomValue} .allowCustomValue=${this.allowCustomValue}
item-id-path="value"
item-value-path="value" item-value-path="value"
item-label-path="label" item-label-path="label"
@opened-changed=${this._openedChanged} @opened-changed=${this._openedChanged}

View File

@ -5,6 +5,7 @@ import { fireEvent } from "../../common/dom/fire_event";
import type { AttributeSelector } from "../../data/selector"; import type { AttributeSelector } from "../../data/selector";
import type { HomeAssistant } from "../../types"; import type { HomeAssistant } from "../../types";
import "../entity/ha-entity-attribute-picker"; import "../entity/ha-entity-attribute-picker";
import { ensureArray } from "../../common/array/ensure-array";
@customElement("ha-selector-attribute") @customElement("ha-selector-attribute")
export class HaSelectorAttribute extends LitElement { export class HaSelectorAttribute extends LitElement {
@ -23,7 +24,7 @@ export class HaSelectorAttribute extends LitElement {
@property({ type: Boolean }) public required = true; @property({ type: Boolean }) public required = true;
@property({ attribute: false }) public context?: { @property({ attribute: false }) public context?: {
filter_entity?: string; filter_entity?: string | string[];
}; };
protected render() { protected render() {
@ -69,11 +70,16 @@ export class HaSelectorAttribute extends LitElement {
// Validate that that the attribute is still valid for this entity, else unselect. // Validate that that the attribute is still valid for this entity, else unselect.
let invalid = false; let invalid = false;
if (this.context.filter_entity) { if (this.context.filter_entity) {
const stateObj = this.hass.states[this.context.filter_entity]; const entityIds = ensureArray(this.context.filter_entity);
if (!(stateObj && this.value in stateObj.attributes)) { invalid = !entityIds.some((entityId) => {
invalid = true; const stateObj = this.hass.states[entityId];
} return (
stateObj &&
this.value in stateObj.attributes &&
stateObj.attributes[this.value] !== undefined
);
});
} else { } else {
invalid = this.value !== undefined; invalid = this.value !== undefined;
} }

View File

@ -41,7 +41,7 @@ export class HaSelectorState extends SubscribeMixin(LitElement) {
.disabled=${this.disabled} .disabled=${this.disabled}
.required=${this.required} .required=${this.required}
allow-custom-value allow-custom-value
.excludeStates=${this.selector.state?.exclude_states} .hideStates=${this.selector.state?.hide_states}
></ha-entity-state-picker> ></ha-entity-state-picker>
`; `;
} }

View File

@ -395,7 +395,7 @@ export interface StateSelector {
extra_options?: { label: string; value: any }[]; extra_options?: { label: string; value: any }[];
entity_id?: string | string[]; entity_id?: string | string[];
attribute?: string; attribute?: string;
exclude_states?: string[]; hide_states?: string[];
} | null; } | null;
} }

View File

@ -57,7 +57,7 @@ export class HaStateTrigger extends LitElement implements TriggerElement {
} }
private _schema = memoizeOne( private _schema = memoizeOne(
(localize: LocalizeFunc, entityId, attribute) => (localize: LocalizeFunc, attribute) =>
[ [
{ {
name: "entity_id", name: "entity_id",
@ -66,9 +66,11 @@ export class HaStateTrigger extends LitElement implements TriggerElement {
}, },
{ {
name: "attribute", name: "attribute",
context: {
filter_entity: "entity_id",
},
selector: { selector: {
attribute: { attribute: {
entity_id: entityId ? entityId[0] : undefined,
hide_attributes: [ hide_attributes: [
"access_token", "access_token",
"available_modes", "available_modes",
@ -211,11 +213,7 @@ export class HaStateTrigger extends LitElement implements TriggerElement {
if (!data.attribute && data.from === null) { if (!data.attribute && data.from === null) {
data.from = ANY_STATE_VALUE; data.from = ANY_STATE_VALUE;
} }
const schema = this._schema( const schema = this._schema(this.hass.localize, this.trigger.attribute);
this.hass.localize,
this.trigger.entity_id,
this.trigger.attribute
);
return html` return html`
<ha-form <ha-form