mirror of
https://github.com/facebook/sapling.git
synced 2024-10-16 03:39:48 +03:00
95eda34a4d
Summary: This is handy. Reviewed By: singhsrb Differential Revision: D9006067 fbshipit-source-id: 79603c8b64b77bb808bb59ac14bfa4f24a03a7d7
76 lines
1.8 KiB
Python
76 lines
1.8 KiB
Python
# debugshell extension
|
|
"""a python shell with repo, changelog & manifest objects"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import code
|
|
import sys
|
|
|
|
import mercurial
|
|
from mercurial import demandimport, registrar
|
|
from mercurial.i18n import _
|
|
|
|
|
|
cmdtable = {}
|
|
command = registrar.command(cmdtable)
|
|
|
|
|
|
def _assignobjects(objects, repo):
|
|
objects.update({"m": mercurial, "mercurial": mercurial})
|
|
if repo:
|
|
objects.update({"repo": repo, "cl": repo.changelog, "mf": repo.manifestlog})
|
|
|
|
# Import other handy modules
|
|
for name in ["os", "hgext", "subprocess", "re"]:
|
|
objects[name] = __import__(name)
|
|
|
|
|
|
def pdb(ui, repo, msg, **opts):
|
|
objects = {}
|
|
_assignobjects(objects, repo)
|
|
code.interact(msg, local=objects)
|
|
|
|
|
|
def ipdb(ui, repo, msg, **opts):
|
|
import IPython
|
|
|
|
_assignobjects(locals(), repo)
|
|
IPython.embed()
|
|
|
|
|
|
@command(
|
|
"debugshell|dbsh",
|
|
[("c", "command", "", _("program passed in as string"), _("CMD"))],
|
|
optionalrepo=True,
|
|
)
|
|
def debugshell(ui, repo, **opts):
|
|
command = opts.get("command")
|
|
if command:
|
|
_assignobjects(locals(), repo)
|
|
exec(command)
|
|
return 0
|
|
|
|
bannermsg = "loaded repo : %s\n" "using source: %s" % (
|
|
repo and repo.root or "(none)",
|
|
mercurial.__path__[0],
|
|
)
|
|
|
|
pdbmap = {"pdb": "code", "ipdb": "IPython"}
|
|
|
|
debugger = ui.config("ui", "debugger")
|
|
if not debugger:
|
|
debugger = "pdb"
|
|
|
|
# if IPython doesn't exist, fallback to code.interact
|
|
try:
|
|
with demandimport.deactivated():
|
|
__import__(pdbmap[debugger])
|
|
except ImportError:
|
|
ui.warn(
|
|
("%s debugger specified but %s module was not found\n")
|
|
% (debugger, pdbmap[debugger])
|
|
)
|
|
debugger = "pdb"
|
|
|
|
getattr(sys.modules[__name__], debugger)(ui, repo, bannermsg, **opts)
|