von/vdstatus
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add pytz to the mix, so that we can display time for specific time zones (totally optional dependency)

Browse source
This commit is contained in:
Von Random 2017-11-30 01:20:35 +03:00
parent e84936b3f8
commit f5c49008ec
2 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
plugins/date.py
View file

@ -1,4 +1,4 @@
import time import datetime
import plugins import plugins
@ -6,6 +6,13 @@ class PluginThread(plugins.PluginThreadCommon):
def __init__(self, section, config): def __init__(self, section, config):
super(PluginThread, self).__init__(section, config) super(PluginThread, self).__init__(section, config)
self.date_format = config.get(section, 'format', fallback='%c') self.date_format = config.get(section, 'format', fallback='%c')
tz = config.get(section, 'TZ', fallback=None)
if tz:
import pytz
self.tz = pytz.timezone(tz)
else:
self.tz = None
def main(self): def main(self):
self.status['full_text'] = time.strftime(self.date_format) now = datetime.datetime.now(tz=self.tz)
self.status['full_text'] = now.strftime(self.date_format)

6
vdstatus
View file

@ -7,9 +7,9 @@ import configparser
import importlib import importlib
import json import json
import os import os
import plugins
import sys import sys
import time import time
import plugins
DEFAULT_CONFIG = os.path.join(os.environ['HOME'], '.config/vdstatus/conf.ini') DEFAULT_CONFIG = os.path.join(os.environ['HOME'], '.config/vdstatus/conf.ini')
@ -36,8 +36,8 @@ class PluginRunner:
self.format_output = self.format_term self.format_output = self.format_term
for section in self.config.sections(): for section in self.config.sections():
plugin_name = self.config.get(section, 'plugin') plugin_name = self.config.get(section, 'plugin')
module = importlib.import_module('.' + plugin_name, 'plugins') mod = importlib.import_module('.' + plugin_name, 'plugins')
thread_object = module.PluginThread(section, self.config) thread_object = mod.PluginThread(section, self.config)
self.plugins_loaded.append(thread_object) self.plugins_loaded.append(thread_object)
def start(self): def start(self):