mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 09:17:30 +03:00
8a3a99ba21
Summary: Be a better citizen under system python path. Fix all tests issues and change setup.py to use glob pattern to include all extensions. Test Plan: Run tests and `make local`. Also build and install the package and run `hg sl` in major repos. Reviewers: #mercurial, ttung, rmcelroy Reviewed By: rmcelroy Subscribers: rmcelroy, durham, mjpieters Differential Revision: https://phabricator.intern.facebook.com/D3534311 Signature: t1:3534311:1468275426:fe122646c8bd6c541e1889e73e9df28f86747ff2
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
# fbamend.py - improved amend functionality
|
|
#
|
|
# Copyright 2013 Facebook, Inc.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2 or any later version.
|
|
"""integrates the statprof profiler more tightly with Mercurial
|
|
|
|
This extension configures the statprof based on Mercurial config knobs.
|
|
|
|
statprof.mechanism - configures which profiling mechanism is used
|
|
(thread, signal)
|
|
statprof.format - configures the output format
|
|
(hotpath, json)
|
|
"""
|
|
|
|
from mercurial import dispatch, extensions
|
|
|
|
def extsetup(ui):
|
|
extensions.wrapfunction(dispatch, 'statprofile', statprofile)
|
|
|
|
def statprofile(orig, ui, func, fp):
|
|
try:
|
|
from hgext3rd import statprof
|
|
except ImportError:
|
|
raise error.Abort(_(
|
|
'statprof not available - install using "easy_install statprof"'))
|
|
|
|
freq = ui.configint('profiling', 'freq', default=1000)
|
|
if freq > 0:
|
|
statprof.reset(freq)
|
|
else:
|
|
ui.warn(_("invalid sampling frequency '%s' - ignoring\n") % freq)
|
|
|
|
mechanism = ui.config('statprof', 'mechanism', 'thread')
|
|
statprof.start(mechanism=mechanism)
|
|
try:
|
|
return func()
|
|
finally:
|
|
statprof.stop()
|
|
|
|
profformat = ui.config('statprof', 'format', 'hotpath')
|
|
|
|
if profformat == 'hotpath':
|
|
displayformat = statprof.DisplayFormats.Hotpath
|
|
elif profformat == 'json':
|
|
displayformat = statprof.DisplayFormats.Json
|
|
else:
|
|
ui.warn(_("unknown profiler output format: %s") % profformat)
|
|
displayformat = statprof.DisplayFormats.Hotpath
|
|
|
|
statprof.display(fp, format=displayformat)
|