upstream/mercurial-mirror Files · contrib/automation/automation.py

pycompat: use os.fsencode() to re-encode sys.argv...

pycompat: use os.fsencode() to re-encode sys.argv Historically, the previous code made sense, as Py_EncodeLocale() and fs.fsencode() could possibly use different encodings. However, this is not the case anymore for Python 3.2, which uses the locale encoding as the filesystem encoding (this is not true for later Python versions, but see below). See https://vstinner.github.io/painful-history-python-filesystem-encoding.html for a source and more background information. Using os.fsencode() is safer, as the documentation for sys.argv says that it can be used to get the original bytes. When doing further changes, the Python developers will take care that this continues to work. One concrete case where os.fsencode() is more correct is when enabling Python's UTF-8 mode. Py_DecodeLocale() will use UTF-8 in this case. Our previous code would have encoded it using the locale encoding (which might be different), whereas os.fsencode() will encode it with UTF-8. Since we don’t claim to support the UTF-8 mode, this is not really a bug and the patch can go to the default branch. It might be a good idea to not commit this to the stable branch, as it could in theory introduce regressions.

Augie Fackler - - Load All Authors

File last commit:

r43346:2372284d default


                r45533:f2de8f31

default

Download file

             automation.py
        
                    74 lines
            
             | 1.7 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / contrib / automation / automation.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      #!/usr/bin/env python3

      #

      # automation.py - Perform tasks on remote machines

      #

      # Copyright 2019 Gregory Szorc <gregory.szorc@gmail.com>

      #

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

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

      import os

      import pathlib

      import subprocess

      import sys

      import venv

      HERE = pathlib.Path(os.path.abspath(__file__)).parent

      REQUIREMENTS_TXT = HERE / 'requirements.txt'

      SOURCE_DIR = HERE.parent.parent

      VENV = SOURCE_DIR / 'build' / 'venv-automation'

      def bootstrap():

          venv_created = not VENV.exists()

          VENV.parent.mkdir(exist_ok=True)

          venv.create(VENV, with_pip=True)

          if os.name == 'nt':

              venv_bin = VENV / 'Scripts'

              pip = venv_bin / 'pip.exe'

              python = venv_bin / 'python.exe'

          else:

              venv_bin = VENV / 'bin'

              pip = venv_bin / 'pip'

              python = venv_bin / 'python'

          args = [

              str(pip),

              'install',

              '-r',

              str(REQUIREMENTS_TXT),

              '--disable-pip-version-check',

          ]

          if not venv_created:

              args.append('-q')

          subprocess.run(args, check=True)

          os.environ['HGAUTOMATION_BOOTSTRAPPED'] = '1'

          os.environ['PATH'] = '%s%s%s' % (venv_bin, os.pathsep, os.environ['PATH'])

          subprocess.run([str(python), __file__] + sys.argv[1:], check=True)

      def run():

          import hgautomation.cli as cli

          # Need to strip off main Python executable.

          cli.main()

      if __name__ == '__main__':

          try:

              if 'HGAUTOMATION_BOOTSTRAPPED' not in os.environ:

                  bootstrap()

              else:

                  run()

          except subprocess.CalledProcessError as e:

              sys.exit(e.returncode)

          except KeyboardInterrupt:

              sys.exit(1)

	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

				#!/usr/bin/env python3
				#
				# automation.py - Perform tasks on remote machines
				#
				# Copyright 2019 Gregory Szorc <gregory.szorc@gmail.com>
				#
				# This software may be used and distributed according to the terms of the
				# GNU General Public License version 2 or any later version.

				import os
				import pathlib
				import subprocess
				import sys
				import venv


				HERE = pathlib.Path(os.path.abspath(__file__)).parent
				REQUIREMENTS_TXT = HERE / 'requirements.txt'
				SOURCE_DIR = HERE.parent.parent
				VENV = SOURCE_DIR / 'build' / 'venv-automation'


				def bootstrap():
				venv_created = not VENV.exists()

				VENV.parent.mkdir(exist_ok=True)

				venv.create(VENV, with_pip=True)

				if os.name == 'nt':
				venv_bin = VENV / 'Scripts'
				pip = venv_bin / 'pip.exe'
				python = venv_bin / 'python.exe'
				else:
				venv_bin = VENV / 'bin'
				pip = venv_bin / 'pip'
				python = venv_bin / 'python'

				args = [
				str(pip),
				'install',
				'-r',
				str(REQUIREMENTS_TXT),
				'--disable-pip-version-check',
				]

				if not venv_created:
				args.append('-q')

				subprocess.run(args, check=True)

				os.environ['HGAUTOMATION_BOOTSTRAPPED'] = '1'
				os.environ['PATH'] = '%s%s%s' % (venv_bin, os.pathsep, os.environ['PATH'])

				subprocess.run([str(python), __file__] + sys.argv[1:], check=True)


				def run():
				import hgautomation.cli as cli

				# Need to strip off main Python executable.
				cli.main()


				if __name__ == '__main__':
				try:
				if 'HGAUTOMATION_BOOTSTRAPPED' not in os.environ:
				bootstrap()
				else:
				run()
				except subprocess.CalledProcessError as e:
				sys.exit(e.returncode)
				except KeyboardInterrupt:
				sys.exit(1)