upstream/mercurial-mirror Files · mercurial/testing/ps_util.py

interfaces: add a Protocol class for `scmutil.status`...

interfaces: add a Protocol class for `scmutil.status` I initially tried moving this to the `interfaces` package, both to have more cleanly defined interfaces (interfaces shouldn't have to reach into implementation files for their type info), and because importing `mercurial.ui` either directly or indirectly into `interfaces.repository` causes a situation where pytype stops inferring the type for `revlogutils.constants` that are imported by `revlog`. (Likely this is caused by a cycle. The `dirstate` interface already imports `scmutil`, which in turn imports `ui`, so the `repository` interface module importing the `dirstate` interface module as part of converting those classes to Protocol classes will trigger the issue.) I gave up on moving the class because `scmutil.status` depends on `stringutil`, which has a surprisingly long tail of dependencies. In any event, a standalone Protocol class might help with the Rust code.

Matt Harbison - - Load All Authors

File last commit:

r53247:d4e30c15 default


                r53347:a1c0f19e

default

Download file

             ps_util.py
        
                    53 lines
            
             | 1.3 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / mercurial / testing / ps_util.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # This python code can be imported into tests in order to terminate a process

      # with signal.SIGKILL on posix, or a roughly equivalent procedure on Windows.

      from __future__ import annotations

      import os

      import signal

      import subprocess

      import sys

      import tempfile

      from .. import (

          encoding,

          pycompat,

      )

      from ..utils import procutil

      def kill_nt(pid: int, exit_code: int):

          fd, pidfile = tempfile.mkstemp(

              prefix=b"sigkill-", dir=encoding.environ[b"HGTMP"], text=False

          )

          try:

              os.write(fd, b'%d\n' % pid)

          finally:

              os.close(fd)

          env = dict(encoding.environ)

          env[b"DAEMON_EXITCODE"] = b"%d" % exit_code

          # Simulate the message written to stderr for this process on non-Windows

          # platforms, for test consistency.

          print("Killed!", file=sys.stderr)

          subprocess.run(

              [

                  encoding.environ[b"PYTHON"],

                  b"%s/killdaemons.py"

                  % encoding.environ[b'RUNTESTDIR_FORWARD_SLASH'],

                  pidfile,

              ],

              env=procutil.tonativeenv(env),

          )

      def kill(pid: int):

          """Kill the process with the given PID with SIGKILL or equivalent."""

          if pycompat.iswindows:

              exit_code = 128 + 9

              kill_nt(pid, exit_code)

          else:

              os.kill(pid, signal.SIGKILL)

	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

				# This python code can be imported into tests in order to terminate a process
				# with signal.SIGKILL on posix, or a roughly equivalent procedure on Windows.

				from __future__ import annotations

				import os
				import signal
				import subprocess
				import sys
				import tempfile

				from .. import (
				encoding,
				pycompat,
				)

				from ..utils import procutil


				def kill_nt(pid: int, exit_code: int):
				fd, pidfile = tempfile.mkstemp(
				prefix=b"sigkill-", dir=encoding.environ[b"HGTMP"], text=False
				)
				try:
				os.write(fd, b'%d\n' % pid)
				finally:
				os.close(fd)

				env = dict(encoding.environ)
				env[b"DAEMON_EXITCODE"] = b"%d" % exit_code

				# Simulate the message written to stderr for this process on non-Windows
				# platforms, for test consistency.
				print("Killed!", file=sys.stderr)

				subprocess.run(
				[
				encoding.environ[b"PYTHON"],
				b"%s/killdaemons.py"
				% encoding.environ[b'RUNTESTDIR_FORWARD_SLASH'],
				pidfile,
				],
				env=procutil.tonativeenv(env),
				)


				def kill(pid: int):
				"""Kill the process with the given PID with SIGKILL or equivalent."""
				if pycompat.iswindows:
				exit_code = 128 + 9
				kill_nt(pid, exit_code)
				else:
				os.kill(pid, signal.SIGKILL)