William Pietri
113 lines
3.2 KiB
Python
113 lines
3.2 KiB
Python
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")
|