cmk_base/profiling.py

   1 #!/usr/bin/env python
   2 # -*- encoding: utf-8; py-indent-offset: 4 -*-
   3 # +------------------------------------------------------------------+
   4 # |             ____ _               _        __  __ _  __           |
   5 # |            / ___| |__   ___  ___| | __   |  \/  | |/ /           |
   6 # |           | |   | '_ \ / _ \/ __| |/ /   | |\/| | ' /            |
   7 # |           | |___| | | |  __/ (__|   <    | |  | | . \            |
   8 # |            \____|_| |_|\___|\___|_|\_\___|_|  |_|_|\_\           |
   9 # |                                                                  |
  10 # | Copyright Mathias Kettner 2014             mk@mathias-kettner.de |
  11 # +------------------------------------------------------------------+
  12 #
  13 # This file is part of Check_MK.
  14 # The official homepage is at http://mathias-kettner.de/check_mk.
  15 #
  16 # check_mk is free software;  you can redistribute it and/or modify it
  17 # under the  terms of the  GNU General Public License  as published by
  18 # the Free Software Foundation in version 2.  check_mk is  distributed
  19 # in the hope that it will be useful, but WITHOUT ANY WARRANTY;  with-
  20 # out even the implied warranty of  MERCHANTABILITY  or  FITNESS FOR A
  21 # PARTICULAR PURPOSE. See the  GNU General Public License for more de-
  22 # tails. You should have  received  a copy of the  GNU  General Public
  23 # License along with GNU Make; see the file  COPYING.  If  not,  write
  24 # to the Free Software Foundation, Inc., 51 Franklin St,  Fifth Floor,
  25 # Boston, MA 02110-1301 USA.
  26
  27 import os
  28 import sys
  29
  30 import cmk_base.console as console
  31
  32 _profile = None
  33 _profile_path = "profile.out"
  34
  35
  36 def enable():
  37     global _profile
  38     import cProfile
  39     _profile = cProfile.Profile()
  40     _profile.enable()
  41     console.verbose("Enabled profiling.\n")
  42
  43
  44 def enabled():
  45     return _profile is not None
  46
  47
  48 def output_profile():
  49     if not _profile:
  50         return
  51
  52     _profile.dump_stats(_profile_path)
  53     show_profile = os.path.join(os.path.dirname(_profile_path), "show_profile.py")
  54     with file(show_profile, "w") as f:
  55         f.write("#!/usr/bin/python\n"
  56                 "import pstats\n"
  57                 "stats = pstats.Stats('%s')\n"
  58                 "stats.sort_stats('time').print_stats()\n" % _profile_path)
  59     os.chmod(show_profile, 0755)
  60
  61     console.output(
  62         "Profile '%s' written. Please run %s.\n" % (_profile_path, show_profile), stream=sys.stderr)