vdstatus/vdstatus

#!/usr/bin/python3
import argparse
import importlib
import json
import os
import sys
import time
import yaml


DEFAULT_CONFIG = os.path.join(os.environ['HOME'], '.config/vdstatus/conf.yaml')
DEFAULTS = {
    'output_format': 'term',
    'output_freq': 1,
    'plugins': [{'name': 'date'}]
}


def parse_arguments():
    desc = ('A simple i3status replacement, '
            'and more. Warning: WIP, may be broken.')
    parser = argparse.ArgumentParser(description=desc)
    parser.add_argument('-c', '--conf', default=DEFAULT_CONFIG,
                        help='configuration file')
    return parser.parse_args()


class PluginRunner:
    def __init__(self, config_file=DEFAULT_CONFIG):
        self.conf = dict()
        self.conf.update(DEFAULTS)
        with open(config_file) as config_data:
            self.conf.update(yaml.safe_load(config_data))
        self.plugins_loaded = list()
        self.format_output = self.format_term
        for plugin in self.conf['plugins']:
            mod = importlib.import_module('.' + plugin['name'], 'plugins')
            thread_object = mod.PluginThread(plugin)
            self.plugins_loaded.append(thread_object)

    def start(self):
        if self.conf['output_format'] == 'i3':
            print('{"version":1}\n[', flush=True)
            self.format_output = self.format_i3wm
        for plugin in self.plugins_loaded:
            plugin.start()

    def query(self):
        outputs = list()
        for plugin in self.plugins_loaded:
            if \
                    'full_text' in plugin.status and (
                            not plugin.conf['hide_ok'] or not plugin.hide
                    ):
                outputs.append(plugin.status)
        print(self.format_output(outputs), flush=True)

    def run(self):
        while True:
            try:
                self.query()
                time.sleep(self.conf['output_freq'])
            except (KeyboardInterrupt, SystemExit):
                sys.exit()

    @staticmethod
    def format_i3wm(inputs):
        return json.dumps(inputs, ensure_ascii=False) + ','

    @staticmethod
    def format_term(inputs):
        return_info = list()
        for item in inputs:
            return_info.append(item['full_text'])
        return ' \033[1m|\033[0m '.join(return_info)


def main():
    args = parse_arguments()
    plugin_runner = PluginRunner(args.conf)
    plugin_runner.start()
    time.sleep(0.1)
    plugin_runner.run()


if __name__ == '__main__':
    main()