Support command for any rorg (#55)

kipe · Jul 17, 2017 · 28d266b · 28d266b
1 parent 7a0b619
commit 28d266b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/enocean/protocol/eep.py b/enocean/protocol/eep.py
@@ -167,12 +167,12 @@ def find_profile(self, bitarray, eep_rorg, rorg_func, rorg_type, direction=None,
 
         profile = self.telegrams[eep_rorg][rorg_func][rorg_type]
 
-        if eep_rorg == RORG.VLD:
-            # For VLD; multiple commands can be defined, with the command id always in same location (per RORG-FUNC-TYPE).
+        if command:
+            # multiple commands can be defined, with the command id always in same location (per RORG-FUNC-TYPE).
             eep_command = profile.find('command', recursive=False)
             # If commands are not set in EEP, or command is None,
             # get the first data as a "best guess".
-            if not eep_command or command is None:
+            if not eep_command:
                 return profile.find('data', recursive=False)
 
             # If eep_command is defined, so should be data.command