upstream/mercurial-mirror Files · mercurial/pure/base85.py

tests: enforce the use of `from __future__ import annotations`...

tests: enforce the use of `from __future__ import annotations` A recent MR and a separate recently landed MR that extracted code to a new file overlooked this, so I think it's worth flagging to ensure consistency. We don't enforce the import for empty files (like `__init__.py`). I'd rather this go into `import-checker.py`, but the import of interest only happens at the top of the file, and its `verify_modern_convention()` calls itself recursively as it transits the AST where the annotations might be. After a few hours of hacking on trying to get it to enforce the import, but only if annotations are used in the module (we generally don't have or check annotations in test files, so don't need this import), I gave up and resorted to this. It won't handle multi-line imports, but this isn't something I'd expect to change often, so this is good enough for now.

Matt Harbison - - Load All Authors

File last commit:

r52830:936f85b2 default


                r53246:662b08ac

default

Download file

             base85.py
        
                    90 lines
            
             | 2.1 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / mercurial / pure / base85.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # base85.py: pure python base85 codec

      #

      # Copyright (C) 2009 Brendan Cully <brendan@kublai.com>

      #

      # This software may be used and distributed according to the terms of the

      # GNU General Public License version 2 or any later version.

      from __future__ import annotations

      import struct

      from .. import pycompat

      _b85chars = pycompat.bytestr(

          b"0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdef"

          b"ghijklmnopqrstuvwxyz!#$%&()*+-;<=>?@^_`{|}~"

      )

      _b85chars2 = [(a + b) for a in _b85chars for b in _b85chars]

      _b85dec = {}

      def _mkb85dec():

          for i, c in enumerate(_b85chars):

              _b85dec[c] = i

      def b85encode(text: bytes, pad: bool = False) -> bytes:

          """encode text in base85 format"""

          l = len(text)

          r = l % 4

          if r:

              text += b'\0' * (4 - r)

          longs = len(text) >> 2

          words = struct.unpack(b'>%dL' % longs, text)

          out = b''.join(

              _b85chars[(word // 52200625) % 85]

              + _b85chars2[(word // 7225) % 7225]

              + _b85chars2[word % 7225]

              for word in words

          )

          if pad:

              return out

          # Trim padding

          olen = l % 4

          if olen:

              olen += 1

          olen += l // 4 * 5

          return out[:olen]

      def b85decode(text: bytes) -> bytes:

          """decode base85-encoded text"""

          if not _b85dec:

              _mkb85dec()

          l = len(text)

          out = []

          acc = 0

          for i in range(0, len(text), 5):

              chunk = text[i : i + 5]

              chunk = pycompat.bytestr(chunk)

              acc = 0

              for j, c in enumerate(chunk):

                  try:

                      acc = acc * 85 + _b85dec[c]

                  except KeyError:

                      raise ValueError(

                          'bad base85 character at position %d' % (i + j)

                      )

              if acc > 4294967295:

                  raise ValueError('Base85 overflow in hunk starting at byte %d' % i)

              out.append(acc)

          # Pad final chunk if necessary

          cl = l % 5

          if cl:

              acc *= 85 ** (5 - cl)

              if cl > 1:

                  acc += 0xFFFFFF >> (cl - 2) * 8

              out[-1] = acc

          out = struct.pack(b'>%dL' % (len(out)), *out)

          if cl:

              out = out[: -(5 - cl)]

          return out

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				# base85.py: pure python base85 codec
				#
				# Copyright (C) 2009 Brendan Cully <brendan@kublai.com>
				#
				# This software may be used and distributed according to the terms of the
				# GNU General Public License version 2 or any later version.

				from __future__ import annotations

				import struct

				from .. import pycompat

				_b85chars = pycompat.bytestr(
				b"0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdef"
				b"ghijklmnopqrstuvwxyz!#$%&()*+-;<=>?@^_`{\|}~"
				)
				_b85chars2 = [(a + b) for a in _b85chars for b in _b85chars]
				_b85dec = {}


				def _mkb85dec():
				for i, c in enumerate(_b85chars):
				_b85dec[c] = i


				def b85encode(text: bytes, pad: bool = False) -> bytes:
				"""encode text in base85 format"""
				l = len(text)
				r = l % 4
				if r:
				text += b'\0' * (4 - r)
				longs = len(text) >> 2
				words = struct.unpack(b'>%dL' % longs, text)

				out = b''.join(
				_b85chars[(word // 52200625) % 85]
				+ _b85chars2[(word // 7225) % 7225]
				+ _b85chars2[word % 7225]
				for word in words
				)

				if pad:
				return out

				# Trim padding
				olen = l % 4
				if olen:
				olen += 1
				olen += l // 4 * 5
				return out[:olen]


				def b85decode(text: bytes) -> bytes:
				"""decode base85-encoded text"""
				if not _b85dec:
				_mkb85dec()

				l = len(text)
				out = []
				acc = 0

				for i in range(0, len(text), 5):
				chunk = text[i : i + 5]
				chunk = pycompat.bytestr(chunk)
				acc = 0
				for j, c in enumerate(chunk):
				try:
				acc = acc * 85 + _b85dec[c]
				except KeyError:
				raise ValueError(
				'bad base85 character at position %d' % (i + j)
				)
				if acc > 4294967295:
				raise ValueError('Base85 overflow in hunk starting at byte %d' % i)
				out.append(acc)

				# Pad final chunk if necessary
				cl = l % 5
				if cl:
				acc = 85 * (5 - cl)
				if cl > 1:
				acc += 0xFFFFFF >> (cl - 2) * 8
				out[-1] = acc

				out = struct.pack(b'>%dL' % (len(out)), *out)
				if cl:
				out = out[: -(5 - cl)]

				return out