Merge pull request jterrace#3 from siam28/feature/change-channel

Added change_channel method and HA change channel helper

pyharmony/__main__.py

@@ -276,6 +276,15 @@ def ha_sync(token, ip, port):
     client.disconnect(send_close=True)
     return 0
 
+def ha_change_channel(token, ip, port, channel):
+    client = ha_get_client(token, ip, port)
+    status = client.change_channel(channel)
+    client.disconnect(send_close=True)
+    if status:
+        return True
+    else:
+        logger.error('Unable to change the channel')
+        return False
 
 # Functions for use on command line
 def show_config(args):

pyharmony/client.py

@@ -158,6 +158,35 @@ def send_command(self, device, command):
         result = iq_cmd.send(block=False)
         return result
 
+    def change_channel(self, channel):
+        """Changes a channel.
+        Args:
+            channel: Channel number
+        Returns:
+          An HTTP 200 response (hopefully)
+        """
+        iq_cmd = self.Iq()
+        iq_cmd['type'] = 'get'
+        action_cmd = ET.Element('oa')
+        action_cmd.attrib['xmlns'] = 'connect.logitech.com'
+        action_cmd.attrib['mime'] = ('harmony.engine?changeChannel')
+        cmd = 'channel=' + str(channel) + ':timestamp=0'
+        action_cmd.text = cmd
+        iq_cmd.set_payload(action_cmd)
+        try:
+            result = iq_cmd.send(block=True)
+        except Exception:
+            logger.info('XMPP timeout, reattempting')
+            result = iq_cmd.send(block=True)
+        payload = result.get_payload()
+        assert len(payload) == 1
+        action_cmd = payload[0]
+        if action_cmd.text == None:
+            return True
+        else:
+            return False
+
+
     def power_off(self):
         """Turns the system off if it's on, otherwise it does nothing.