sapling/hgext/debugshell.py
Jun Wu 584656dff3 codemod: join the auto-formatter party
Summary:
Turned on the auto formatter. Ran `arc lint --apply-patches --take BLACK **/*.py`.
Then run `arc lint` again so some other autofixers like spellchecker etc. looked
at the code base. Manually accept the changes whenever they make sense, or use
a workaround (ex. changing "dict()" to "dict constructor") where autofix is false
positive. Disabled linters on files that are hard (i18n/polib.py) to fix, or less
interesting to fix (hgsubversion tests), or cannot be fixed without breaking
OSS build (FBPYTHON4).

Conflicted linters (test-check-module-imports.t, part of test-check-code.t,
test-check-pyflakes.t) are removed or disabled.

Duplicated linters (test-check-pyflakes.t, test-check-pylint.t) are removed.

An issue of the auto-formatter is lines are no longer guarnateed to be <= 80
chars. But that seems less important comparing with the benefit auto-formatter
provides.

As we're here, also remove test-check-py3-compat.t, as it is currently broken
if `PYTHON3=/bin/python3` is set.

Reviewed By: wez, phillco, simpkins, pkaush, singhsrb

Differential Revision: D8173629

fbshipit-source-id: 90e248ae0c5e6eaadbe25520a6ee42d32005621b
2018-05-25 22:17:29 -07:00

72 lines
1.7 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})
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)