Allow user to specify entity IDs directly

Author: sunnythaper

README.md

@@ -19,6 +19,8 @@ Here's how it works:
 3. If you set `save_file` to false, go to the Home Assistant logs to copy the generated YAML for the scene.
 4. If you set `save_file` to true, go to your configuration folder and open `generated_scene.yaml`.
 
+This script allows you to specify either domains or entities directly. If both domains and entities are defined, the script will prioritize entities over domains.
+
 ### EXAMPLE SERVICE DATA
 
 ![example service call](https://github.com/sunnythaper/python_scripts/raw/master/readme-assets/service_call.png)
@@ -28,6 +30,9 @@ domains:
   - light
   - switch
   - fan
+entities:
+  - light.living_room
+  - switch.kitchen
 attributes:
   - brightness
   - color_temp
@@ -36,7 +41,7 @@ attributes:
 save_file: true
 ```
 
-In this example, we're generating a scene for all light, switch, and fan entities, and including brightness, color temperature, and color attributes.
+In this example, we're generating a scene for all light, switch, and fan entities defined in the domains, as well as specifically for `light.living_room` and `switch.kitchen` entities. It includes brightness, color temperature, and color attributes. If both domains and entities are defined, the script will process only the entities and ignore the domains.
 
 ### FILE SAVING
 

readme-assets/service_call.png

@@ -6,26 +6,40 @@ def format_entity_status(entity, status, attributes):
                        for attribute in attributes if status.attributes.get(attribute)]
     return f"{entity}:\n  state: {status.state}\n" + "".join(attribute_texts) + "\n"
 
-def process_domain(domain, attributes):
-    text_lines = [format_header(domain)]
-    entities = hass.states.entity_ids(domain)
+def process_entities(entities, attributes):
+    text_lines = []
     for entity_id in entities:
         entity_status = hass.states.get(entity_id)
         text_lines.append(format_entity_status(entity_id, entity_status, attributes))
     return "".join(text_lines)
 
+def process_domain(domain, attributes):
+    text_lines = [format_header(domain)]
+    entities = hass.states.entity_ids(domain)
+    text_lines.append(process_entities(entities, attributes))
+    return "".join(text_lines)
+
 def process_domains(domains, attributes):
+    if not domains:
+        return ""
     return "".join(process_domain(domain, attributes) for domain in domains)
 
+def process_input(domains, entities, attributes):
+    if entities:
+        return process_entities(entities, attributes)
+    else:
+        return process_domains(domains, attributes)
+
 def export_results(text, save_file):
     if save_file:
         hass.services.call("notify", "scene_generator", {"message": "{}".format(text)})
     else:
         logger.info(text)
 
 domains = data.get('domains', ['light','switch','fan'])
+entities = data.get('entities')
 attributes = data.get('attributes', ['brightness','color_temp','xy_color','rgb_color'])
 save_file = data.get('save_file')
 
-text = process_domains(domains, attributes)
+text = process_input(domains, entities, attributes)
 export_results(text, save_file)

scene_generator.py

@@ -1,13 +1,18 @@
 scene_generator:
   name: Scene Generator
-  description: Generate a scene based on the state of specific entities in specified domains.
+  description: Generate a scene based on the state of specific entities in specified domains or directly defined entities.
   fields:
     domains:
-      description: The domains to be processed for generating the scene.
+      description: The domains to be processed for generating the scene. Ignored if entities are defined.
       example:
         - light
         - switch
         - fan
+    entities:
+      description: Specific entities to be processed for generating the scene. Takes precedence over domains if both are defined.
+      example:
+        - light.living_room
+        - switch.kitchen
     attributes:
       description: The attributes of the entities to be processed.
       example: