1 files changed, 138 insertions, 0 deletions
diff --git a/glances/plugins/processcount/__init__.py b/glances/plugins/processcount/__init__.py
index e69de29b..fba9fc41 100644
--- a/glances/plugins/processcount/__init__.py
+++ b/glances/plugins/processcount/__init__.py
@@ -0,0 +1,138 @@
+# -*- coding: utf-8 -*-
+#
+# This file is part of Glances.
+#
+# SPDX-FileCopyrightText: 2022 Nicolas Hennion <nicolas@nicolargo.com>
+#
+# SPDX-License-Identifier: LGPL-3.0-only
+#
+
+"""Process count plugin."""
+
+from glances.processes import glances_processes, sort_for_human
+from glances.plugins.plugin.model import GlancesPluginModel
+
+# Define the history items list
+items_history_list = [
+    {'name': 'total', 'description': 'Total number of processes', 'y_unit': ''},
+    {'name': 'running', 'description': 'Total number of running processes', 'y_unit': ''},
+    {'name': 'sleeping', 'description': 'Total number of sleeping processes', 'y_unit': ''},
+    {'name': 'thread', 'description': 'Total number of threads', 'y_unit': ''},
+]
+
+
+class PluginModel(GlancesPluginModel):
+    """Glances process count plugin.
+
+    stats is a list
+    """
+
+    def __init__(self, args=None, config=None):
+        """Init the plugin."""
+        super(PluginModel, self).__init__(args=args, config=config, items_history_list=items_history_list)
+
+        # We want to display the stat in the curse interface
+        self.display_curse = True
+
+        # Note: 'glances_processes' is already init in the glances_processes.py script
+
+    def enable_extended(self):
+        """Enable extended stats."""
+        glances_processes.enable_extended()
+
+    def disable_extended(self):
+        """Disable extended stats."""
+        glances_processes.disable_extended()
+
+    @GlancesPluginModel._check_decorator
+    @GlancesPluginModel._log_result_decorator
+    def update(self):
+        """Update processes stats using the input method."""
+        # Init new stats
+        stats = self.get_init_value()
+
+        if self.input_method == 'local':
+            # Update stats using the standard system lib
+            # Here, update is call for processcount AND processlist
+            glances_processes.update()
+
+            # Return the processes count
+            stats = glances_processes.get_count()
+        elif self.input_method == 'snmp':
+            # Update stats using SNMP
+            # Not available
+            pass
+
+        # Update the stats
+        self.stats = stats
+
+        return self.stats
+
+    def msg_curse(self, args=None, max_width=None):
+        """Return the dict to display in the curse interface."""
+        # Init the return message
+        ret = []
+
+        # Only process if stats exist and display plugin enable...
+        if args.disable_process:
+            msg = "PROCESSES DISABLED (press 'z' to display)"
+            ret.append(self.curse_add_line(msg))
+            return ret
+
+        if not self.stats:
+            return ret
+
+        # Display the filter (if it exists)
+        if glances_processes.process_filter is not None:
+            msg = 'Processes filter:'
+            ret.append(self.curse_add_line(msg, "TITLE"))
+            msg = ' {} '.format(glances_processes.process_filter)
+            if glances_processes.process_filter_key is not None:
+                msg += 'on column {} '.format(glances_processes.process_filter_key)
+            ret.append(self.curse_add_line(msg, "FILTER"))
+            msg = '(\'ENTER\' to edit, \'E\' to reset)'
+            ret.append(self.curse_add_line(msg))
+            ret.append(self.curse_new_line())
+
+        # Build the string message
+        # Header
+        msg = 'TASKS'
+        ret.append(self.curse_add_line(msg, "TITLE"))
+        # Compute processes
+        other = self.stats['total']
+        msg = '{:>4}'.format(self.stats['total'])
+        ret.append(self.curse_add_line(msg))
+
+        if 'thread' in self.stats:
+            msg = ' ({} thr),'.format(self.stats['thread'])
+            ret.append(self.curse_add_line(msg))
+
+        if 'running' in self.stats:
+            other -= self.stats['running']
+            msg = ' {} run,'.format(self.stats['running'])
+            ret.append(self.curse_add_line(msg))
+
+        if 'sleeping' in self.stats:
+            other -= self.stats['sleeping']
+            msg = ' {} slp,'.format(self.stats['sleeping'])
+            ret.append(self.curse_add_line(msg))
+
+        msg = ' {} oth '.format(other)
+        ret.append(self.curse_add_line(msg))
+
+        # Display sort information
+        msg = 'Programs' if self.args.programs else 'Threads'
+        try:
+            sort_human = sort_for_human[glances_processes.sort_key]
+        except KeyError:
+            sort_human = glances_processes.sort_key
+        if glances_processes.auto_sort:
+            msg += ' sorted automatically'
+            ret.append(self.curse_add_line(msg))
+            msg = ' by {}'.format(sort_human)
+        else:
+            msg += ' sorted by {}'.format(sort_human)
+        ret.append(self.curse_add_line(msg))
+
+        # Return the message with decoration
+        return ret