mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 07:17:55 +03:00
584656dff3
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
198 lines
4.2 KiB
Python
198 lines
4.2 KiB
Python
# coding=UTF-8
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import base64
|
|
import zlib
|
|
|
|
from mercurial import changegroup, exchange, extensions, filelog, revlog, util
|
|
|
|
|
|
# Test only: These flags are defined here only in the context of testing the
|
|
# behavior of the flag processor. The canonical way to add flags is to get in
|
|
# touch with the community and make them known in revlog.
|
|
REVIDX_NOOP = 1 << 3
|
|
REVIDX_BASE64 = 1 << 2
|
|
REVIDX_GZIP = 1 << 1
|
|
REVIDX_FAIL = 1
|
|
|
|
|
|
def validatehash(self, text):
|
|
return True
|
|
|
|
|
|
def bypass(self, text):
|
|
return False
|
|
|
|
|
|
def noopdonothing(self, text):
|
|
return (text, True)
|
|
|
|
|
|
def b64encode(self, text):
|
|
return (base64.b64encode(text), False)
|
|
|
|
|
|
def b64decode(self, text):
|
|
return (base64.b64decode(text), True)
|
|
|
|
|
|
def gzipcompress(self, text):
|
|
return (zlib.compress(text), False)
|
|
|
|
|
|
def gzipdecompress(self, text):
|
|
return (zlib.decompress(text), True)
|
|
|
|
|
|
def supportedoutgoingversions(orig, repo):
|
|
versions = orig(repo)
|
|
versions.discard("01")
|
|
versions.discard("02")
|
|
versions.add("03")
|
|
return versions
|
|
|
|
|
|
def allsupportedversions(orig, ui):
|
|
versions = orig(ui)
|
|
versions.add("03")
|
|
return versions
|
|
|
|
|
|
def noopaddrevision(
|
|
orig,
|
|
self,
|
|
text,
|
|
transaction,
|
|
link,
|
|
p1,
|
|
p2,
|
|
cachedelta=None,
|
|
node=None,
|
|
flags=revlog.REVIDX_DEFAULT_FLAGS,
|
|
):
|
|
if "[NOOP]" in text:
|
|
flags |= REVIDX_NOOP
|
|
return orig(
|
|
self,
|
|
text,
|
|
transaction,
|
|
link,
|
|
p1,
|
|
p2,
|
|
cachedelta=cachedelta,
|
|
node=node,
|
|
flags=flags,
|
|
)
|
|
|
|
|
|
def b64addrevision(
|
|
orig,
|
|
self,
|
|
text,
|
|
transaction,
|
|
link,
|
|
p1,
|
|
p2,
|
|
cachedelta=None,
|
|
node=None,
|
|
flags=revlog.REVIDX_DEFAULT_FLAGS,
|
|
):
|
|
if "[BASE64]" in text:
|
|
flags |= REVIDX_BASE64
|
|
return orig(
|
|
self,
|
|
text,
|
|
transaction,
|
|
link,
|
|
p1,
|
|
p2,
|
|
cachedelta=cachedelta,
|
|
node=node,
|
|
flags=flags,
|
|
)
|
|
|
|
|
|
def gzipaddrevision(
|
|
orig,
|
|
self,
|
|
text,
|
|
transaction,
|
|
link,
|
|
p1,
|
|
p2,
|
|
cachedelta=None,
|
|
node=None,
|
|
flags=revlog.REVIDX_DEFAULT_FLAGS,
|
|
):
|
|
if "[GZIP]" in text:
|
|
flags |= REVIDX_GZIP
|
|
return orig(
|
|
self,
|
|
text,
|
|
transaction,
|
|
link,
|
|
p1,
|
|
p2,
|
|
cachedelta=cachedelta,
|
|
node=node,
|
|
flags=flags,
|
|
)
|
|
|
|
|
|
def failaddrevision(
|
|
orig,
|
|
self,
|
|
text,
|
|
transaction,
|
|
link,
|
|
p1,
|
|
p2,
|
|
cachedelta=None,
|
|
node=None,
|
|
flags=revlog.REVIDX_DEFAULT_FLAGS,
|
|
):
|
|
# This addrevision wrapper is meant to add a flag we will not have
|
|
# transforms registered for, ensuring we handle this error case.
|
|
if "[FAIL]" in text:
|
|
flags |= REVIDX_FAIL
|
|
return orig(
|
|
self,
|
|
text,
|
|
transaction,
|
|
link,
|
|
p1,
|
|
p2,
|
|
cachedelta=cachedelta,
|
|
node=node,
|
|
flags=flags,
|
|
)
|
|
|
|
|
|
def extsetup(ui):
|
|
# Enable changegroup3 for flags to be sent over the wire
|
|
wrapfunction = extensions.wrapfunction
|
|
wrapfunction(changegroup, "supportedoutgoingversions", supportedoutgoingversions)
|
|
wrapfunction(changegroup, "allsupportedversions", allsupportedversions)
|
|
|
|
# Teach revlog about our test flags
|
|
flags = [REVIDX_NOOP, REVIDX_BASE64, REVIDX_GZIP, REVIDX_FAIL]
|
|
revlog.REVIDX_KNOWN_FLAGS |= util.bitsfrom(flags)
|
|
revlog.REVIDX_FLAGS_ORDER.extend(flags)
|
|
|
|
# Teach exchange to use changegroup 3
|
|
for k in exchange._bundlespeccgversions.keys():
|
|
exchange._bundlespeccgversions[k] = "03"
|
|
|
|
# Add wrappers for addrevision, responsible to set flags depending on the
|
|
# revision data contents.
|
|
wrapfunction(filelog.filelog, "addrevision", noopaddrevision)
|
|
wrapfunction(filelog.filelog, "addrevision", b64addrevision)
|
|
wrapfunction(filelog.filelog, "addrevision", gzipaddrevision)
|
|
wrapfunction(filelog.filelog, "addrevision", failaddrevision)
|
|
|
|
# Register flag processors for each extension
|
|
revlog.addflagprocessor(REVIDX_NOOP, (noopdonothing, noopdonothing, validatehash))
|
|
revlog.addflagprocessor(REVIDX_BASE64, (b64decode, b64encode, bypass))
|
|
revlog.addflagprocessor(REVIDX_GZIP, (gzipdecompress, gzipcompress, bypass))
|