From: Peter Palfrader <peter@palfrader.org>
Date: Mon, 12 Mar 2012 15:28:29 +0000 (+0100)
Subject: Profile if UD_PROFILE is in environment
X-Git-Tag: userdir-ldap-0.3.85~65^2~4
X-Git-Url: https://git.adam-barratt.org.uk/?p=mirror%2Fuserdir-ldap.git;a=commitdiff_plain;h=966a64834daa780fc48b2d24ac4c59bb8cf0beb4

Profile if UD_PROFILE is in environment
---

diff --git a/ud-generate b/ud-generate
index fd6a16b..b329719 100755
--- a/ud-generate
+++ b/ud-generate
@@ -1316,9 +1316,17 @@ def ud_generate():
       if lock is not None:
          lock.close()
 
-if __name__ == "__main__":
-   ud_generate()
 
+if __name__ == "__main__":
+   if 'UD_PROFILE' in os.environ:
+      import cProfile
+      import pstats
+      cProfile.run('ud_generate()', "udg_prof")
+      p = pstats.Stats('udg_prof')
+      ##p.sort_stats('time').print_stats()
+      p.sort_stats('cumulative').print_stats()
+   else:
+      ud_generate()
 
 # vim:set et:
 # vim:set ts=3: