initial import

2017-10-31 10:55:18 -07:00
commit 20579d92ab
4 changed files with 170 additions and 0 deletions
@@ -0,0 +1,112 @@
+import configparser
+import itertools
+import logging
+import os
+import time
+from threading import Event
+
+import click
+from sleekxmpp import ClientXMPP
+from sleekxmpp.xmlstream import ET
+
+
+class VacBot(ClientXMPP):
+    def __init__(self, user, domain, resource, secret, vacuum):
+        ClientXMPP.__init__(self, user + '@' + domain, '0/' + resource + '/' + secret)
+
+        self.user = user
+        self.domain = domain
+        self.resource = resource
+        self.vacuum = vacuum
+        self.credentials['authzid'] = user
+        self.add_event_handler("session_start", self.session_start)
+
+        self.ready_flag = Event()
+
+    def wait_until_ready(self):
+        self.ready_flag.wait()
+
+    def session_start(self, event):
+        print("----------------- starting session ----------------")
+        self.ready_flag.set()
+
+    def clean(self, speed='standard', type='auto'):
+        self.make_command('clean', {'speed': speed, 'type': type}).send()
+
+    def charge(self):
+        self.make_command('charge', {'type': 'go'}).send()
+
+    def make_command(self, name, args):
+        clean = ET.Element(name, args)
+        ctl = ET.Element('ctl', {'td': name.capitalize()})
+        ctl.append(clean)
+        command = self.wrap_command(ctl)
+        return command
+
+    def wrap_command(self, ctl):
+        q = self.make_iq_query(xmlns=u'com:ctl', ito=self.vacuum + '/atom',
+                               ifrom=self.user + '@' + self.domain + '/' + self.resource)
+        q['type'] = 'set'
+        for child in q.xml:
+            if child.tag.endswith('query'):
+                child.append(ctl)
+                return q
+
+    def run(self):
+        self.connect(('47.88.66.164', '5223'))  # TODO: change to domain name
+        click.echo("starting")
+        self.process()
+        click.echo("done with process")
+        self.wait_until_ready()
+
+
+def read_config(filename):
+    parser = configparser.ConfigParser()
+    with open(filename) as fp:
+        parser.read_file(itertools.chain(['[global]'], fp), source=filename)
+    return parser['global']
+
+
+@click.group(chain=True)
+@click.option('--charge/--no-charge', default=True, help='Return to charge after running. Defaults to yes.')
+@click.option('--debug/--no-debug', default=False)
+def cli(charge, debug):
+    logging.basicConfig(level=logging.DEBUG, format='%(levelname)-8s %(message)s')
+
+
+@cli.command(help='cleans for the specified number of minutes')
+@click.argument('minutes', type=click.FLOAT)
+def clean(minutes):
+    def action(vacbot):
+        click.echo('cleaning')
+        vacbot.clean()
+        time.sleep(minutes * 60)
+
+    return action
+
+
+@cli.command(help='returns to charger')
+def charge(vacbot): # TODO convert
+    click.echo('charging')
+    vacbot.charge()
+    time.sleep(20)
+
+
+@cli.resultcallback()
+def run(actions, charge, debug):
+    config = read_config(os.path.expanduser('~/.config/sucks.conf'))
+    vacbot = VacBot(config['user'], config['domain'], config['resource'], config['secret'],
+                    config['vacuum'])
+    vacbot.run()
+    for action in actions:
+        click.echo("running " + str(action))
+        action(vacbot)
+    if charge:
+        vacbot.charge()
+    vacbot.disconnect(wait=True)
+
+
+if __name__ == '__main__':
+    click.echo("starting commands")
+    cli()
+    click.echo("done with commands")