mirror of
https://github.com/facebook/sapling.git
synced 2024-10-12 01:39:21 +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
99 lines
3.1 KiB
Python
99 lines
3.1 KiB
Python
# Copyright 2018-present Facebook. All Rights Reserved.
|
|
#
|
|
# clienttelemetry: provide information about the client in server telemetry
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2 or any later version.
|
|
"""provide information about the client in server telemetry
|
|
|
|
[clienttelemetry]
|
|
# whether or not to announce the remote hostname when connecting
|
|
announceremotehostname = False
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import random
|
|
import socket
|
|
import string
|
|
|
|
from mercurial import dispatch, extensions, hg, wireproto
|
|
from mercurial.i18n import _
|
|
|
|
|
|
# Client telemetry functions generate client telemetry data at connection time.
|
|
_clienttelemetryfuncs = {}
|
|
|
|
|
|
def clienttelemetryfunc(f):
|
|
"""Decorator for registering client telemetry functions."""
|
|
_clienttelemetryfuncs[f.__name__] = f
|
|
return f
|
|
|
|
|
|
@clienttelemetryfunc
|
|
def hostname(ui):
|
|
return socket.gethostname()
|
|
|
|
|
|
_correlator = None
|
|
|
|
|
|
@clienttelemetryfunc
|
|
def correlator(ui):
|
|
"""
|
|
The correlator is a random string that is logged on both the client and
|
|
server. This can be used to correlate the client logging to the server
|
|
logging.
|
|
"""
|
|
global _correlator
|
|
if _correlator is None:
|
|
alphabet = string.ascii_letters + string.digits
|
|
_correlator = "".join(random.choice(alphabet) for _x in range(16))
|
|
ui.log("clienttelemetry", "", client_correlator=_correlator)
|
|
return _correlator
|
|
|
|
|
|
# Client telemetry data is generated before connection and stored here.
|
|
_clienttelemetrydata = {}
|
|
|
|
|
|
def _clienttelemetry(repo, proto, args):
|
|
"""Handle received client telemetry"""
|
|
logargs = {"client_%s" % key: value for key, value in args.items()}
|
|
repo.ui.log("clienttelemetry", "", **logargs)
|
|
return socket.gethostname()
|
|
|
|
|
|
def _capabilities(orig, repo, proto):
|
|
result = orig(repo, proto)
|
|
result.append("clienttelemetry")
|
|
return result
|
|
|
|
|
|
def _runcommand(orig, lui, repo, cmd, fullargs, ui, options, d, cmdpats, cmdoptions):
|
|
# Record the command that is running in the client telemetry data.
|
|
_clienttelemetrydata["command"] = cmd
|
|
_clienttelemetrydata["fullcommand"] = dispatch._formatargs(fullargs)
|
|
return orig(lui, repo, cmd, fullargs, ui, options, d, cmdpats, cmdoptions)
|
|
|
|
|
|
def _peersetup(ui, peer):
|
|
if peer.capable("clienttelemetry"):
|
|
logargs = {name: f(ui) for name, f in _clienttelemetryfuncs.items()}
|
|
logargs.update(_clienttelemetrydata)
|
|
peername = peer._call("clienttelemetry", **logargs)
|
|
ui.log("clienttelemetry", "", server_realhostname=peername)
|
|
ann = ui.configbool("clienttelemetry", "announceremotehostname", None)
|
|
if ann is None:
|
|
ann = not ui.plain() and ui._isatty(ui.ferr)
|
|
if ann and not ui.quiet:
|
|
ui.warn(_("connected to %s\n") % peername)
|
|
|
|
|
|
def uisetup(ui):
|
|
wireproto.wireprotocommand("clienttelemetry", "*")(_clienttelemetry)
|
|
extensions.wrapfunction(wireproto, "_capabilities", _capabilities)
|
|
hg.wirepeersetupfuncs.append(_peersetup)
|
|
extensions.wrapfunction(dispatch, "runcommand", _runcommand)
|