mirror of
https://github.com/facebook/sapling.git
synced 2024-10-12 01:39:21 +03:00
e4ce4c6416
Summary: Add more shortcuts to common modules. This make it easier to test changed code paths. Reviewed By: markbt Differential Revision: D15952263 fbshipit-source-id: c0eca6a61902d36a26a99f85e29dc70f431eca59
83 lines
2.1 KiB
Python
83 lines
2.1 KiB
Python
# Copyright 2010 Mercurial Contributors
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2 or any later version.
|
|
|
|
# debugshell extension
|
|
"""a python shell with repo, changelog & manifest objects"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import code
|
|
import edenscm
|
|
import edenscmnative
|
|
import os
|
|
import sys
|
|
|
|
from edenscm import hgext, mercurial
|
|
from edenscm.mercurial import demandimport, registrar, thirdparty
|
|
from edenscm.mercurial.i18n import _
|
|
|
|
from edenscmnative import bindings
|
|
|
|
cmdtable = {}
|
|
command = registrar.command(cmdtable)
|
|
|
|
|
|
def _assignobjects(objects, repo):
|
|
objects.update(
|
|
{
|
|
"m": mercurial,
|
|
"e": edenscm,
|
|
"n": edenscmnative,
|
|
"b": bindings,
|
|
"x": hgext,
|
|
"mercurial": mercurial,
|
|
}
|
|
)
|
|
if repo:
|
|
objects.update({"repo": repo, "cl": repo.changelog, "mf": repo.manifestlog})
|
|
|
|
# Import other handy modules
|
|
for name in ["os", "subprocess", "re"]:
|
|
objects[name] = __import__(name)
|
|
|
|
|
|
@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],
|
|
) + (
|
|
"\n\nAvailable variables:\n"
|
|
" e: edenscm\n"
|
|
" n: edenscmnative\n"
|
|
" m: edenscm.mercurial\n"
|
|
" x: edenscm.hgext\n"
|
|
" b: edenscmnative.bindings\n"
|
|
" ui: the ui object"
|
|
)
|
|
if repo:
|
|
bannermsg += (
|
|
"\n repo: the repo object\n cl: repo.changelog\n mf: repo.manifestlog"
|
|
)
|
|
|
|
_assignobjects(locals(), repo)
|
|
|
|
# demandimport is incompatible with many IPython dependencies, both at
|
|
# import time and at runtime.
|
|
with demandimport.deactivated():
|
|
import IPython
|
|
|
|
IPython.embed(header=bannermsg)
|