Python_script function

[foxsam (Sam Fox)]

I have a script to setup input_select menus… It works as is but if I want to reuse as a function it does not work… It should be exactly the same thing…

this works:

group_entities = hass.states.get('group.all_switches').attributes['entity_id']
# group_entities = ["switch.switch1","switch.sw2","switch.light1","switch.sonoff2,switch.yellow_room","switch.kmc_a","switch.kmc_b","switch.kmc_c"]
all_switches = []
for e in group_entities:
    if "kmc" in e and ("_a" in e or "_c" in e):
        all_switches.append(e)


service_data = {'entity_id': 'input_select.box_1',
                'options': all_switches}
hass.services.call('input_select', 'set_options', service_data)


service_data = {'entity_id': 'input_select.box_2',
                'options': all_switches}
hass.services.call('input_select', 'set_options', service_data)

but if I put it into a function then nothing happens…

def update_data(entity):
    service_data = {'entity_id': 'input_select.'+entity,
                    'options': all_switches}
    hass.services.call('input_select', 'set_options', service_data)


update_data('box_1')
update_data('box_2')

they both print() (if tried in python console without the hass.service.call()):

{'entity_id': 'input_select.box_1', 'options': ['switch.kmc_a', 'switch.kmc_c']}
{'entity_id': 'input_select.box_2', 'options': ['switch.kmc_a', 'switch.kmc_c']}

[david1 (David)]

i know very little python, but don’t you need to return something in the function?

https://www.programiz.com/python-programming/function

[foxsam (Sam Fox)]

I also know very little but in the link you posted it says:

7. An optional return statement to return a value from the function.

I don’t think a return is required.

[andrewdolphin]

Anything in the logs?

Python scripts are run within RestrictedPython so running the script in the console isn’t proof the script is working within that environment.

I’m no expert but I’m wondering if it might be a scoping issue.

Perhaps pass the all_switches variable into the function?

[123 (Taras)]

You have to pass the hass object as a parameter to the function.

def update_data(hass, entity):

Automation template - how to detect slider last and new state and compare it in action template

It would appear you have to pass the hass object as a parameter to the call_service function in order to be able to use it within the function. There’s an example of it in this recent post by Marius: So you will have to change the function’s signature to this: def call_service(hass, host, rfkey, count): and modify every call to the function. For example, this: call_service('192.168.2.1', 'C', 10) changes to this: call_service(hass, '192.168.2.1', 'C', 10)

[petro]

@foxsam

group_entities = hass.states.get(‘group.all_switches’).attributes[‘entity_id’] # group_entities = [“switch.switch1”,“switch.sw2”,“switch.light1”,“switch.sonoff2,switch.yellow_room”,“switch.kmc_a”,“switch.kmc_b”,“switch.kmc_c”] all_switches = for e in group_entities: if “kmc” in e and (“_a” in e or “_c” in e): all_switches.append(e) service_data = {‘entity_id’: ‘input_select.box_1’, ‘options’: all_switches} hass.services.call(‘input_select’, ‘set_options’, service_data) service_data = {‘entity_id’: ‘input_select.box_2’, ‘options’: all_switches} hass.services.call(‘input_select’, ‘set_options’, service_data)

you can just do another for loop instead of a function. Typically functions in python scripts are very limited because the environment is limited. It’s not like normal python.

group_entities = hass.states.get('group.all_switches').attributes['entity_id']
# group_entities = ["switch.switch1","switch.sw2","switch.light1","switch.sonoff2,switch.yellow_room","switch.kmc_a","switch.kmc_b","switch.kmc_c"]
all_switches = []
for e in group_entities:
    if "kmc" in e and ("_a" in e or "_c" in e):
        all_switches.append(e)

service_data =  {'options': all_switches}
for n in ['box_1', 'box_2']:
    service_data['entity_id'] = 'input_select.{}'.format(n)
    hass.services.call('input_select', 'set_options', service_data)