5fe6d049d2
settings attribute for every class inheriting from IntervalModule are automatically extended by interval setting if it does not exist. A function to flatten settings in SettingsBase was extracted and added as a static method.
73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
|
|
from i3pystatus.core.settings import SettingsBase
|
|
from i3pystatus.core.threading import Manager
|
|
from i3pystatus.core.util import convert_position
|
|
|
|
|
|
class Module(SettingsBase):
|
|
output = None
|
|
position = 0
|
|
|
|
def registered(self, status_handler):
|
|
"""Called when this module is registered with a status handler"""
|
|
|
|
def inject(self, json):
|
|
if self.output:
|
|
if "name" not in self.output:
|
|
self.output["name"] = self.__name__
|
|
self.output["instance"] = str(id(self))
|
|
json.insert(convert_position(self.position, json), self.output)
|
|
|
|
def run(self):
|
|
pass
|
|
|
|
def on_click(self, button):
|
|
if button == 1: # Left mouse button
|
|
self.on_leftclick()
|
|
elif button == 3: # Right mouse button
|
|
self.on_rightclick()
|
|
|
|
def move(self, position):
|
|
self.position = position
|
|
return self
|
|
|
|
def on_leftclick(self):
|
|
pass
|
|
|
|
def on_rightclick(self):
|
|
pass
|
|
|
|
|
|
class IntervalModuleMeta(type):
|
|
"""Add interval setting to `settings` attribute if it does not exist."""
|
|
|
|
def __init__(cls, name, bases, namespace):
|
|
super(IntervalModuleMeta, cls).__init__(name, bases, namespace)
|
|
if not hasattr(cls, 'settings'):
|
|
cls.settings = tuple()
|
|
if not 'interval' in SettingsBase.flatten_settings(cls.settings):
|
|
cls.settings += ('interval', )
|
|
|
|
|
|
class IntervalModule(Module, metaclass=IntervalModuleMeta):
|
|
interval = 5 # seconds
|
|
managers = {}
|
|
|
|
def registered(self, status_handler):
|
|
if self.interval in IntervalModule.managers:
|
|
IntervalModule.managers[self.interval].append(self)
|
|
else:
|
|
am = Manager(self.interval)
|
|
am.append(self)
|
|
IntervalModule.managers[self.interval] = am
|
|
am.start()
|
|
|
|
def __call__(self):
|
|
self.run()
|
|
|
|
def run(self):
|
|
"""Called approximately every self.interval seconds
|
|
|
|
Do not rely on this being called from the same thread at all times.
|
|
If you need to always have the same thread context, subclass AsyncModule."""
|