From 0afd338cc7bb495f8f0f2c92ba084419b87ff775 Mon Sep 17 00:00:00 2001
From: angelblue05 <tamara.angel05@gmail.com>
Date: Thu, 17 Mar 2016 02:55:00 -0500
Subject: [PATCH] Make profiling a decorator

You can call it before any functions/methods using @utils.profiling()
<-- you can specify the sort order of the results, by default it sorts
by cumulative. It will print the results straight into the Kodi log.
---
 resources/lib/utils.py | 53 ++++++++++++++----------------------------
 1 file changed, 18 insertions(+), 35 deletions(-)

diff --git a/resources/lib/utils.py b/resources/lib/utils.py
index 9e8c1707..c9b3474f 100644
--- a/resources/lib/utils.py
+++ b/resources/lib/utils.py
@@ -7,6 +7,7 @@ import inspect
 import json
 import pstats
 import sqlite3
+import StringIO
 import os
 from datetime import datetime, time
 import time
@@ -266,44 +267,26 @@ def reset():
         line1="Database reset has completed, Kodi will now restart to apply the changes.")
     xbmc.executebuiltin('RestartApp')
 
-def startProfiling():
-    
-    pr = cProfile.Profile()
-    pr.enable()
-    
-    return pr
+def profiling(sortby="cumulative"):
+    # Will print results to Kodi log
+    def decorator(func):
+        def wrapper(*args, **kwargs):
+            
+            pr = cProfile.Profile()
 
-def stopProfiling(pr, profileName):
-    
-    pr.disable()
-    ps = pstats.Stats(pr)
-    
-    profiles = xbmc.translatePath("%sprofiles/"
-                % xbmcaddon.Addon().getAddonInfo('profile')).decode('utf-8')
+            pr.enable()
+            result = func(*args, **kwargs)
+            pr.disable()
 
-    if not xbmcvfs.exists(profiles):
-        # Create the profiles folder
-        xbmcvfs.mkdir(profiles)
+            s = StringIO.StringIO()
+            ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
+            ps.print_stats()
+            logMsg("EMBY Profiling", s.getvalue(), 1)
 
-    timestamp = time.strftime("%Y-%m-%d %H-%M-%S")
-    profile = "%s%s_profile_(%s).tab" % (profiles, profileName, timestamp)
-    
-    f = xbmcvfs.File(profile, 'w')
-    f.write("NumbCalls\tTotalTime\tCumulativeTime\tFunctionName\tFileName\r\n")
-    for (key, value) in ps.stats.items():
-        (filename, count, func_name) = key
-        (ccalls, ncalls, total_time, cumulative_time, callers) = value
-        try:
-            f.write(
-                "%s\t%s\t%s\t%s\t%s\r\n"
-                % (ncalls, "{:10.4f}".format(total_time),
-                    "{:10.4f}".format(cumulative_time), func_name, filename))
-        except ValueError:
-            f.write(
-                "%s\t%s\t%s\t%s\t%s\r\n"
-                % (ncalls, "{0}".format(total_time),
-                    "{0}".format(cumulative_time), func_name, filename))
-    f.close()
+            return result
+
+        return wrapper
+    return decorator
 
 def convertdate(date):
     try: