Mercurial > hg
view mercurial/revlogutils/concurrency_checker.py @ 49803:55d45d0de4e7
typing: add type hints to pycompat.bytestr
The problem with leaving pytype to its own devices here was that for functions
that returned a bytestr, pytype inferred `Union[bytes, int]`. It now accepts
that it can be treated as plain bytes.
I wasn't able to figure out the arg type for `__getitem__`- `SupportsIndex`
(which PyCharm indicated is how the superclass function is typed) got flagged:
File "/mnt/c/Users/Matt/hg/mercurial/pycompat.py", line 236, in __getitem__:
unsupported operand type(s) for item retrieval: bytestr and SupportsIndex [unsupported-operands]
Function __getitem__ on bytestr expects int
But some caller got flagged when I marked it as `int`.
There's some minor spillover problems elsewhere- pytype doesn't seem to
recognize that `bytes.startswith()` can optionally take a 3rd and 4th arg, so
those few places have the warning disabled. It also flags where the tar API is
being abused, but that would be a tricky refactor (and would require typing
extensions until py3.7 is dropped), so disable those too.
author | Matt Harbison <matt_harbison@yahoo.com> |
---|---|
date | Wed, 14 Dec 2022 01:51:33 -0500 |
parents | e9901d01d135 |
children |
line wrap: on
line source
from ..i18n import _ from .. import error def get_checker(ui, revlog_name=b'changelog'): """Get a function that checks file handle position is as expected. This is used to ensure that files haven't been modified outside of our knowledge (such as on a networked filesystem, if `hg debuglocks` was used, or writes to .hg that ignored locks happened). Due to revlogs supporting a concept of buffered, delayed, or diverted writes, we're allowing the files to be shorter than expected (the data may not have been written yet), but they can't be longer. Please note that this check is not perfect; it can't detect all cases (there may be false-negatives/false-OKs), but it should never claim there's an issue when there isn't (false-positives/false-failures). """ vpos = ui.config(b'debug', b'revlog.verifyposition.' + revlog_name) # Avoid any `fh.tell` cost if this isn't enabled. if not vpos or vpos not in [b'log', b'warn', b'fail']: return None def _checker(fh, fn, expected): if fh.tell() <= expected: return msg = _(b'%s: file cursor at position %d, expected %d') # Always log if we're going to warn or fail. ui.log(b'debug', msg + b'\n', fn, fh.tell(), expected) if vpos == b'warn': ui.warn((msg + b'\n') % (fn, fh.tell(), expected)) elif vpos == b'fail': raise error.RevlogError(msg % (fn, fh.tell(), expected)) return _checker