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

2 files changed, 12 insertions, 5 deletions

diff --git a/plugins/date.py b/plugins/date.py
index df967f8..d147051 100644
--- a/plugins/date.py
+++ b/plugins/date.py
@@ -1,4 +1,4 @@
-import time
+import datetime
 import plugins
 
 
@@ -6,6 +6,13 @@ class PluginThread(plugins.PluginThreadCommon):
     def __init__(self, section, config):
         super(PluginThread, self).__init__(section, config)
         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):
-        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)
diff --git a/vdstatus b/vdstatus
index 83d8b27..f9b4111 100755
--- a/vdstatus
+++ b/vdstatus
@@ -7,9 +7,9 @@ import configparser
 import importlib
 import json
 import os
-import plugins
 import sys
 import time
+import plugins
 
 
 DEFAULT_CONFIG = os.path.join(os.environ['HOME'], '.config/vdstatus/conf.ini')
@@ -36,8 +36,8 @@ class PluginRunner:
         self.format_output = self.format_term
         for section in self.config.sections():
             plugin_name = self.config.get(section, 'plugin')
-            module = importlib.import_module('.' + plugin_name, 'plugins')
-            thread_object = module.PluginThread(section, self.config)
+            mod = importlib.import_module('.' + plugin_name, 'plugins')
+            thread_object = mod.PluginThread(section, self.config)
             self.plugins_loaded.append(thread_object)
 
     def start(self):