Call unload with ui arg

This commit is contained in:
dadav
2019-11-30 09:43:39 +01:00
parent f973997cdb
commit 608904daf8
3 changed files with 6 additions and 6 deletions

View File

@ -3,6 +3,7 @@ import glob
import _thread import _thread
import importlib, importlib.util import importlib, importlib.util
import logging import logging
from pwnagotchi.ui import view
default_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "default") default_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "default")
loaded = {} loaded = {}
@ -27,14 +28,14 @@ def toggle_plugin(name, enable=True):
global loaded, database global loaded, database
if not enable and name in loaded: if not enable and name in loaded:
if getattr(loaded[name], 'on_unload', None): if getattr(loaded[name], 'on_unload', None):
loaded[name].on_unload() loaded[name].on_unload(view.ROOT)
del loaded[name] del loaded[name]
return True return True
if enable and name in database and name not in loaded: if enable and name in database and name not in loaded:
load_from_file(database[name]) load_from_file(database[name])
one(name, 'loaded') one(name, 'loaded')
one(name, 'ui_setup') one(name, 'ui_setup', view.ROOT)
return True return True
return False return False

View File

@ -466,11 +466,10 @@ class BTTether(plugins.Plugin):
logging.info("BT-TETHER: Successfully loaded ...") logging.info("BT-TETHER: Successfully loaded ...")
self.ready = True self.ready = True
def on_unload(self): def on_unload(self, ui):
self.ui.remove_element('bluetooth') ui.remove_element('bluetooth')
def on_ui_setup(self, ui): def on_ui_setup(self, ui):
self.ui = ui
ui.add_element('bluetooth', LabeledValue(color=BLACK, label='BT', value='-', position=(ui.width() / 2 - 15, 0), ui.add_element('bluetooth', LabeledValue(color=BLACK, label='BT', value='-', position=(ui.width() / 2 - 15, 0),
label_font=fonts.Bold, text_font=fonts.Medium)) label_font=fonts.Bold, text_font=fonts.Medium))

View File

@ -26,7 +26,7 @@ class Example(plugins.Plugin):
logging.warning("WARNING: this plugin should be disabled! options = " % self.options) logging.warning("WARNING: this plugin should be disabled! options = " % self.options)
# called before the plugin is unloaded # called before the plugin is unloaded
def on_unload(self): def on_unload(self, ui):
pass pass
# called hen there's internet connectivity # called hen there's internet connectivity