power menu, redshift, polybar

.config/polybar/scripts/info-softwarecounter/README.md

@@ -0,0 +1,25 @@
+# Script: info-softwarecounter
+
+A script that counts the number of specified running software including GUIs and processes.
+
+Options to only monitor GUI applications or only processes exist. Users can add their own applications or processes they wish to watch simply by updating the existing dictionaries at the start of the script.
+
+Arbitrary program counts can be combined, for example, the vim/nvim or chrome/chromium counts can be added together and the total displayed.
+
+![info-softwarecounter](screenshots/1.png)
+
+
+## Dependencies
+
+* `wmctrl`
+* `pgrep`
+
+
+## Module
+
+```
+[module/info-softwarecounter]
+type = custom/script
+exec = ~/polybar-scripts/info-softwarecounter.py
+interval = 10
+```

.config/polybar/scripts/info-softwarecounter/info-softwarecounter.py

@@ -0,0 +1,223 @@
+#!/usr/bin/env python3
+
+import subprocess
+from collections import Counter, OrderedDict
+import logging
+
+
+# update GUIs
+GUI = True
+
+# update processes
+PROCESS = True
+
+# Debugging feature
+DEBUG = False
+
+
+logger = logging.getLogger(__name__)
+if DEBUG:
+    FORMAT = '%(asctime)s - %(levelname)s - %(funcName)s:%(lineno)d - %(message)s'
+    logger.setLevel(logging.DEBUG)
+    fh = logging.FileHandler(
+            filename='./info-softwarecounter.log',
+            encoding='utf-8'
+    )
+    fh.setLevel(logging.DEBUG)
+    fh.setFormatter(logging.Formatter(FORMAT))
+    logger.addHandler(fh)
+
+
+#   _____ _    _ _____
+#  / ____| |  | |_   _|
+# | |  __| |  | | | |
+# | | |_ | |  | | | |
+# | |__| | |__| |_| |_
+#  \_____|\____/|_____|
+
+
+# program/process names and corresponding gylphs
+guis = OrderedDict({
+    'terminals': '#',
+    'chromes': '#',
+    'firefox': '#',
+    'skypeforlinux': '#',
+    'filemanager': '#',
+    'remote-desktop': '#',
+    'pdfviewer': '#',
+    'image': '#',
+})
+
+# combine counts of program/process names in the tuple
+# the resulting glpyh used will be that of the corresponding key
+combine_guis = {
+    'terminals': ('gnome-terminal-server', 'xfce4-terminal', 'alacritty', 'termite', 'terminator', 'urxvt'),
+    'chromes': ('chromium', 'chrome'),
+    'filemanager': ('nemo', 'thunar', 'dolphin', 'nautilus', 'pcmanfm'),
+    'remote-desktop': ('TeamViewer', ),
+    'pdfviewer': ('evince', 'okular', 'zathura'),
+    'image': ('gthumb', 'shotwell', 'deepin-image-vi', 'eog', 'gimp-2.10'),
+}
+
+
+def get(cmd):
+    return subprocess.check_output(cmd).decode("utf-8").strip()
+
+
+# def check_wtype(w_id):
+#     # check the type of window, only list "NORMAL" windows
+#     return "_NET_WM_WINDOW_TYPE_NORMAL" in get(["xprop", "-id", w_id])
+
+
+def get_process(w_id):
+    # get the name of the process, owning the window
+    proc = get(["ps", "-p", w_id, "-o", "cmd="])
+    return proc.split()[0].split("/")[-1]
+
+
+def get_running_guis():
+
+    wlist = [
+        line.split() for line in subprocess.check_output(
+            ["wmctrl", "-lp"]
+        ).decode("utf-8").splitlines()
+    ]
+
+    logger.debug("wlist after LC:")
+    logger.debug("---------")
+    for i in wlist:
+        logger.debug(i)
+    logger.debug("---------")
+
+    validprocs = [
+        get_process(w[2]) for w in wlist if w[2] != '0'  # and check_wtype(w[0])
+    ]
+
+    logger.debug(f"validprocs -> {validprocs}")
+    return validprocs
+
+
+def GUI(gui_output=''):
+
+    # get list of running GUI programs
+    gui_counts = Counter(get_running_guis())
+    logger.debug(f"gui_counts -> {gui_counts}")
+
+    logger.debug("combine_guis items:")
+    logger.debug("---------")
+    # combine programs in program combine list
+    for k, lst in combine_guis.items():
+        logger.debug(f"{k} -> {lst}")
+        count = 0
+        for i in lst:
+            try:
+                count += gui_counts.pop(i)
+            except KeyError:
+                pass
+        if count:
+            gui_counts[k] += count
+    logger.debug("---------")
+    logger.debug(f"gui_counts after for loop -> {gui_counts}")
+
+    logger.debug("guis items:")
+    logger.debug("---------")
+    # generate program output
+    for k, v in guis.items():
+        logger.debug(f"{k} -> {v}")
+        try:
+            logger.debug(f"k, gui_counts[k] -> {k}, {gui_counts[k]}")
+            c = gui_counts[k]
+            if c:
+                gui_output += '%s %i ' % (v, c)
+        except Exception:
+            pass
+    logger.debug("---------")
+
+    logger.debug(f"gui_output -> {gui_output}")
+    return gui_output
+
+
+#  _____  _____   ____   _____ ______  _____ _____
+# |  __ \|  __ \ / __ \ / ____|  ____|/ ____/ ____|
+# | |__) | |__) | |  | | |    | |__  | (___| (___
+# |  ___/|  _  /| |  | | |    |  __|  \___ \\___ \
+# | |    | | \ \| |__| | |____| |____ ____) |___) |
+# |_|    |_|  \_\\____/ \_____|______|_____/_____/
+
+
+processes = OrderedDict({
+    'vims': '#',
+    'ssh': '#',
+    'updater': '#',
+})
+
+combine_proccesses = {
+    'vims': ('nvim', 'vim', 'atom'),
+    'updater': ('pacman', 'yay', 'trizen', 'yaourt', 'makepkg', 'auracle'),
+}
+
+
+def get_running_proc(process_name_list):
+    counts = [None] * len(process_name_list)
+
+    for i, p in enumerate(process_name_list):
+        try:
+            count = int(
+                subprocess.check_output(
+                    ['pgrep', '-c', '-x', p]
+                ).decode('utf-8')
+            )
+        except subprocess.CalledProcessError:
+            count = 0
+
+        counts[i] = (p, count)
+    return dict(counts)
+
+
+def PROCESS(process_output=''):
+
+    # count running proccesses
+    process_counts = get_running_proc(processes.keys())
+    combine_counts = get_running_proc(
+        list(sum(combine_proccesses.values(), ())))
+    process_counts.update(combine_counts)
+
+    # combine processes in process combine list
+    for k, lst in combine_proccesses.items():
+        count = 0
+        for i in lst:
+            try:
+                count += process_counts.pop(i)
+            except KeyError:
+                pass
+        if count:
+            process_counts[k] += count
+
+    # generate process output
+    for k, v in processes.items():
+        try:
+            c = process_counts[k]
+            if c:
+                process_output += '%s %i ' % (v, c)
+        except Exception:
+            pass
+
+    logger.debug(f"process_output -> {process_output}")
+    return process_output
+
+
+def main():
+
+    if GUI:
+        gui_output = GUI()
+
+    if PROCESS:
+        process_output = PROCESS()
+
+    print(gui_output + process_output)
+
+
+if __name__ == "__main__":
+    logger.debug("------------ Script runs  ------------")
+    main()
+    logger.debug("------------ Script stops ------------\n")