OctoPrint-MyStromSwitch/octoprint_mystromswitch/__init__.py
2019-11-20 21:50:02 +01:00

119 lines
3.7 KiB
Python

# coding=utf-8
from __future__ import absolute_import
import ssl
import octoprint.plugin
import requests
from octoprint.util import RepeatedTimer
class MyStromSwitchPlugin(octoprint.plugin.SettingsPlugin,
octoprint.plugin.AssetPlugin,
octoprint.plugin.TemplatePlugin,
octoprint.plugin.StartupPlugin,
octoprint.plugin.ShutdownPlugin):
def __init__(self):
self.ip = None
self.intervall = 1
self._timer = None
self.ctx = ssl.create_default_context()
self.ctx.check_hostname = False
self.ctx.verify_mode = ssl.CERT_NONE
def initialize(self):
self.ip = self._settings.get(["ip"])
self._logger.debug("ip: %s" % self.ip)
self.intervall = self._settings.get_int(["intervall"])
self._logger.debug("intervall: %s" % self.intervall)
self._timer_start()
def get_assets(self):
return dict(js=["js/mystromswitch.js"], css=["css/mystromswitch.css"])
def get_template_configs(self):
return [dict(type="sidebar",
name="MyStrom Switch",
custom_bindings=False,
icon="power-off"),
dict(type="settings", custom_bindings=False)]
def _timer_start(self):
if self._timer is not None:
self._timer.cancel()
self._logger.info("Canceling Timer")
if self.intervall >= 1 and self.ip is not None:
self._logger.info("Starting timer")
self._timer = RepeatedTimer(self.intervall, self._timer_task)
self._timer.start()
def _timer_task(self):
if self.ip is not None:
try:
request = requests.get(
'http://{}/report'.format(self.ip), timeout=1)
self._logger.info(request.json())
self._plugin_manager.send_plugin_message(self._identifier,
request.json())
except (requests.exceptions.ConnectionError, ValueError):
self._logger.info('Connection Error Host: {}'.format(self.ip))
else:
self._logger.info("Ip is None")
def on_after_startup(self):
self._logger.info("Hello World!")
def on_shutdown(self):
self._logger.info("Hello World!")
def on_settings_migrate(self, target, current):
pass
def get_settings_version(self):
return 1
def get_settings_defaults(self):
return dict(
ip=None,
intervall=1
)
def on_settings_save(self, data):
self._logger.info("on_settings_save")
octoprint.plugin.SettingsPlugin.on_settings_save(self, data)
self.initialize()
def get_update_information(self):
return dict(
mystromswitch=dict(
displayName="OctoPrint-MyStromSwitch",
displayVersion=self._plugin_version,
# version check: github repository
type="github_release",
user="da4id",
repo="OctoPrint-MyStromSwitch",
current=self._plugin_version,
# update method: pip w/ dependency links
pip="https://github.com/da4id/OctoPrint-MyStromSwitch/archive/{target_version}.zip"
)
)
__plugin_name__ = "MyStrom Switch"
def __plugin_load__():
global __plugin_implementation__
__plugin_implementation__ = MyStromSwitchPlugin()
global __plugin_hooks__
__plugin_hooks__ = {
"octoprint.plugin.softwareupdate.check_config": __plugin_implementation__.get_update_information
}