upstream/mercurial-mirror Files · mercurial/diffhelper.py

typing: add type hints to the `charencode` module...

typing: add type hints to the `charencode` module Since this module is dynamically imported from either `mercurial.pure` or `mercurial.cext`, these hints aren't detected in `mercurial.encoding`, and need to be imported directly there during the type-checking phase. This keeps the runtime selection via the policy config in place, but allows pytype to see these as functions with proper signatures instead of just `Any`. We don't attempt to import the `mercurial.cext` version yet because there's no types stubs for that module, but this will get the ball rolling. I thought this would spill over into other modules from there, but the only two *.pyi files that changed were for `encoding` and `charencode`. Applying this to other dynamically selected modules will clean some things up in other files, so this is a start. I had originally redefined the functions in the type-checking block (like some of the `os.path` aliasing in `mercurial.util`), but this is better because we won't have another duplication of the definitions that may get out of date.

Manuel Jacob - - Load All Authors

File last commit:

r50179:d44e3c45 default


                r52615:43adbe03

default

Download file

             diffhelper.py
        
                    81 lines
            
             | 2.2 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / mercurial / diffhelper.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # diffhelper.py - helper routines for patch

      #

      # Copyright 2009 Olivia Mackall <olivia@selenic.com> and others

      #

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

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

      from .i18n import _

      from . import (

          error,

      )

      MISSING_NEWLINE_MARKER = b'\\ No newline at end of file\n'

      def addlines(fp, hunk, lena, lenb, a, b):

          """Read lines from fp into the hunk

          The hunk is parsed into two arrays, a and b. a gets the old state of

          the text, b gets the new state. The control char from the hunk is saved

          when inserting into a, but not b (for performance while deleting files.)

          """

          while True:

              todoa = lena - len(a)

              todob = lenb - len(b)

              num = max(todoa, todob)

              if num == 0:

                  break

              for i in range(num):

                  s = fp.readline()

                  if not s:

                      raise error.ParseError(_(b'incomplete hunk'))

                  if s == MISSING_NEWLINE_MARKER:

                      fixnewline(hunk, a, b)

                      continue

                  if s == b'\n' or s == b'\r\n':

                      # Some patches may be missing the control char

                      # on empty lines. Supply a leading space.

                      s = b' ' + s

                  hunk.append(s)

                  if s.startswith(b'+'):

                      b.append(s[1:])

                  elif s.startswith(b'-'):

                      a.append(s)

                  else:

                      b.append(s[1:])

                      a.append(s)

      def fixnewline(hunk, a, b):

          """Fix up the last lines of a and b when the patch has no newline at EOF"""

          l = hunk[-1]

          # tolerate CRLF in last line

          if l.endswith(b'\r\n'):

              hline = l[:-2]

          else:

              hline = l[:-1]

          if hline.startswith((b' ', b'+')):

              b[-1] = hline[1:]

          if hline.startswith((b' ', b'-')):

              a[-1] = hline

          hunk[-1] = hline

      def testhunk(a, b, bstart):

          """Compare the lines in a with the lines in b

          a is assumed to have a control char at the start of each line, this char

          is ignored in the compare.

          """

          alen = len(a)

          blen = len(b)

          if alen > blen - bstart or bstart < 0:

              return False

          for i in range(alen):

              if a[i][1:] != b[i + bstart]:

                  return False

          return True

	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

				# diffhelper.py - helper routines for patch
				#
				# Copyright 2009 Olivia Mackall <olivia@selenic.com> and others
				#
				# This software may be used and distributed according to the terms of the
				# GNU General Public License version 2 or any later version.


				from .i18n import _

				from . import (
				error,
				)

				MISSING_NEWLINE_MARKER = b'\\ No newline at end of file\n'


				def addlines(fp, hunk, lena, lenb, a, b):
				"""Read lines from fp into the hunk

				The hunk is parsed into two arrays, a and b. a gets the old state of
				the text, b gets the new state. The control char from the hunk is saved
				when inserting into a, but not b (for performance while deleting files.)
				"""
				while True:
				todoa = lena - len(a)
				todob = lenb - len(b)
				num = max(todoa, todob)
				if num == 0:
				break
				for i in range(num):
				s = fp.readline()
				if not s:
				raise error.ParseError(_(b'incomplete hunk'))
				if s == MISSING_NEWLINE_MARKER:
				fixnewline(hunk, a, b)
				continue
				if s == b'\n' or s == b'\r\n':
				# Some patches may be missing the control char
				# on empty lines. Supply a leading space.
				s = b' ' + s
				hunk.append(s)
				if s.startswith(b'+'):
				b.append(s[1:])
				elif s.startswith(b'-'):
				a.append(s)
				else:
				b.append(s[1:])
				a.append(s)


				def fixnewline(hunk, a, b):
				"""Fix up the last lines of a and b when the patch has no newline at EOF"""
				l = hunk[-1]
				# tolerate CRLF in last line
				if l.endswith(b'\r\n'):
				hline = l[:-2]
				else:
				hline = l[:-1]

				if hline.startswith((b' ', b'+')):
				b[-1] = hline[1:]
				if hline.startswith((b' ', b'-')):
				a[-1] = hline
				hunk[-1] = hline


				def testhunk(a, b, bstart):
				"""Compare the lines in a with the lines in b

				a is assumed to have a control char at the start of each line, this char
				is ignored in the compare.
				"""
				alen = len(a)
				blen = len(b)
				if alen > blen - bstart or bstart < 0:
				return False
				for i in range(alen):
				if a[i][1:] != b[i + bstart]:
				return False
				return True