Implement thread profiling

1 files changed, 26 insertions, 0 deletions

diff --git a/py3status/__init__.py b/py3status/__init__.py
index b152366..81d0d99 100755
--- a/py3status/__init__.py
+++ b/py3status/__init__.py
@@ -2,6 +2,7 @@ from __future__ import print_function
 
 import argparse
 import ast
+import cProfile
 import imp
 import locale
 import os
@@ -29,6 +30,27 @@ try:
 except ImportError:
     pass
 
+# Used in development
+enable_profiling = True
+
+
+def profile(thread_run_fn):
+    if not enable_profiling:
+        return thread_run_fn
+
+    def wrapper_run(self):
+        """Wrap the Thread.run() method
+        """
+        profiler = cProfile.Profile()
+        try:
+            return profiler.runcall(thread_run_fn, self)
+        finally:
+            thread_id = getattr(self, 'ident', 'core')
+            profiler.dump_stats("py3status-%s.profile" % thread_id)
+
+    return wrapper_run
+
+
 @contextmanager
 def jsonify(string):
     """
@@ -472,6 +494,7 @@ class I3status(Thread):
                 self.write_in_tmpfile('}\n\n', tmpfile)
         tmpfile.flush()
 
+    @profile
     def run(self):
         """
         Spawn i3status using a self generated config file and poll its output.
@@ -758,6 +781,7 @@ class Events(Thread):
         finally:
             return (instance, name)
 
+    @profile
     def run(self):
         """
         Wait for an i3bar JSON event, then find the right module to dispatch
@@ -1001,6 +1025,7 @@ class Module(Thread):
             msg = 'on_click failed with ({}) for event ({})'.format(err, event)
             syslog(LOG_WARNING, msg)
 
+    @profile
     def run(self):
         """
         On a timely fashion, execute every method found for this module.
@@ -1502,6 +1527,7 @@ class Py3statusWrapper():
         """
         raise KeyboardInterrupt()
 
+    @profile
     def run(self):
         """
         Main py3status loop, continuously read from i3status and modules