mia/bumper
1
0
Fork 0
Code Issues 47 Pull requests 11 Projects Releases Packages Wiki Activity Actions

Add bumper plugin to mqtt

Browse source
This commit is contained in:
Brian Martin 2019-01-30 10:20:54 -05:00
parent 134d698eeb
commit b184e1c2c6
2 changed files with 36 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

41
bumper/mqttserver.py
View file

@ -5,19 +5,43 @@ import asyncio
import os
from hbmqtt.broker import Broker
import hbmqtt
import pkg_resources
class BumperMQTTPlugin:
def __init__(self, context):
self.context = context
logging.debug('Bumper Plugin Initialized')
@asyncio.coroutine
def on_broker_client_connected(self, client_id):
logging.debug('Bumper Connection: %s connected' % client_id)
#yield from self.context.broadcast_message('location/%s' % client_id, b'home')
@asyncio.coroutine
def on_broker_client_disconnected(self, client_id):
logging.debug('Bumper Connection: %s disconnected' % client_id)
#yield from self.context.broadcast_message('location/%s' % client_id, b'not_home')
class MQTTServer():
clients = []
exit_flag = False
default_config = {}
default_config = {}
@asyncio.coroutine
def broker_coro(self):
broker = hbmqtt.broker.Broker(config=self.default_config, plugin_namespace=".")
yield from broker.start()
def broker_coro(self):
broker = hbmqtt.broker.Broker(config=self.default_config)
yield from broker.start()
def __init__(self, address, run_async=False):
#The below adds a plugin to the hbmqtt.broker.plugins without having to futz with setup.py
distribution = pkg_resources.Distribution("hbmqtt.broker.plugins")
bumper_plugin = pkg_resources.EntryPoint.parse('bumper = bumper.mqttserver:BumperMQTTPlugin', dist=distribution)
distribution._ep_map = {"hbmqtt.broker.plugins": {"bumper": bumper_plugin}}
pkg_resources.working_set.add(distribution)
#for entry_point in pkg_resources.iter_entry_points("hbmqtt.broker.plugins"):
# print(entry_point)
try:
# Initialize bot server
self.default_config = {
@ -37,7 +61,12 @@ class MQTTServer():
'allow-anonymous': True,
'password-file': os.path.join(os.path.dirname(os.path.realpath(__file__)), "passwd"),
'plugins': [
'VacBotAuth'
'' #No plugins == no auth
]
},
'broker': {
'plugins': [
'bumper'
]
},
'topic-check': {