clean up scripts

.config/polybar/scripts/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")