| pulkit |
| hg-reviewers |
Otherwise Python 3.8 complains.
| Lint Skipped |
| Unit Tests Skipped |
| Path | Packages | |||
|---|---|---|---|---|
| M | contrib/automation/hgautomation/aws.py (2 lines) | |||
| M | contrib/automation/hgautomation/windows.py (8 lines) |
| Commit | Parents | Author | Summary | Date |
|---|---|---|---|---|
| Gregory Szorc | Apr 4 2019, 9:01 PM |
| } | } | ||||
| } | } | ||||
| # User Data for Windows EC2 instance. Mainly used to set the password | # User Data for Windows EC2 instance. Mainly used to set the password | ||||
| # and configure WinRM. | # and configure WinRM. | ||||
| # Inspired by the User Data script used by Packer | # Inspired by the User Data script used by Packer | ||||
| # (from https://www.packer.io/intro/getting-started/build-image.html). | # (from https://www.packer.io/intro/getting-started/build-image.html). | ||||
| WINDOWS_USER_DATA = ''' | WINDOWS_USER_DATA = r''' | ||||
| <powershell> | <powershell> | ||||
| # TODO enable this once we figure out what is failing. | # TODO enable this once we figure out what is failing. | ||||
| #$ErrorActionPreference = "stop" | #$ErrorActionPreference = "stop" | ||||
| # Set administrator password | # Set administrator password | ||||
| net user Administrator "%s" | net user Administrator "%s" | ||||
| wmic useraccount where "name='Administrator'" set PasswordExpires=FALSE | wmic useraccount where "name='Administrator'" set PasswordExpires=FALSE | ||||
| else: | else: | ||||
| raise ValueError('illegal arch: %s; must be x86 or x64' % arch) | raise ValueError('illegal arch: %s; must be x86 or x64' % arch) | ||||
| def fix_authorized_keys_permissions(winrm_client, path): | def fix_authorized_keys_permissions(winrm_client, path): | ||||
| commands = [ | commands = [ | ||||
| '$ErrorActionPreference = "Stop"', | '$ErrorActionPreference = "Stop"', | ||||
| 'Repair-AuthorizedKeyPermission -FilePath %s -Confirm:$false' % path, | 'Repair-AuthorizedKeyPermission -FilePath %s -Confirm:$false' % path, | ||||
| 'icacls %s /remove:g "NT Service\sshd"' % path, | r'icacls %s /remove:g "NT Service\sshd"' % path, | ||||
| ] | ] | ||||
| run_powershell(winrm_client, '\n'.join(commands)) | run_powershell(winrm_client, '\n'.join(commands)) | ||||
| def synchronize_hg(hg_repo: pathlib.Path, revision: str, ec2_instance): | def synchronize_hg(hg_repo: pathlib.Path, revision: str, ec2_instance): | ||||
| """Synchronize local Mercurial repo to remote EC2 instance.""" | """Synchronize local Mercurial repo to remote EC2 instance.""" | ||||
| """Purge the Mercurial source repository on an EC2 instance.""" | """Purge the Mercurial source repository on an EC2 instance.""" | ||||
| run_powershell(winrm_client, HG_PURGE) | run_powershell(winrm_client, HG_PURGE) | ||||
| def find_latest_dist(winrm_client, pattern): | def find_latest_dist(winrm_client, pattern): | ||||
| """Find path to newest file in dist/ directory matching a pattern.""" | """Find path to newest file in dist/ directory matching a pattern.""" | ||||
| res = winrm_client.execute_ps( | res = winrm_client.execute_ps( | ||||
| '$v = Get-ChildItem -Path C:\hgdev\src\dist -Filter "%s" ' | r'$v = Get-ChildItem -Path C:\hgdev\src\dist -Filter "%s" ' | ||||
| '| Sort-Object LastWriteTime -Descending ' | '| Sort-Object LastWriteTime -Descending ' | ||||
| '| Select-Object -First 1\n' | '| Select-Object -First 1\n' | ||||
| '$v.name' % pattern | '$v.name' % pattern | ||||
| ) | ) | ||||
| return res[0] | return res[0] | ||||
| def copy_latest_dist(winrm_client, pattern, dest_path): | def copy_latest_dist(winrm_client, pattern, dest_path): | ||||
| def run_tests(winrm_client, python_version, arch, test_flags=''): | def run_tests(winrm_client, python_version, arch, test_flags=''): | ||||
| """Run tests on a remote Windows machine. | """Run tests on a remote Windows machine. | ||||
| ``python_version`` is a ``X.Y`` string like ``2.7`` or ``3.7``. | ``python_version`` is a ``X.Y`` string like ``2.7`` or ``3.7``. | ||||
| ``arch`` is ``x86`` or ``x64``. | ``arch`` is ``x86`` or ``x64``. | ||||
| ``test_flags`` is a str representing extra arguments to pass to | ``test_flags`` is a str representing extra arguments to pass to | ||||
| ``run-tests.py``. | ``run-tests.py``. | ||||
| """ | """ | ||||
| if not re.match('\d\.\d', python_version): | if not re.match(r'\d\.\d', python_version): | ||||
| raise ValueError('python_version must be \d.\d; got %s' % | raise ValueError(r'python_version must be \d.\d; got %s' % | ||||
| python_version) | python_version) | ||||
| if arch not in ('x86', 'x64'): | if arch not in ('x86', 'x64'): | ||||
| raise ValueError('arch must be x86 or x64; got %s' % arch) | raise ValueError('arch must be x86 or x64; got %s' % arch) | ||||
| python_path = 'python%s-%s' % (python_version.replace('.', ''), arch) | python_path = 'python%s-%s' % (python_version.replace('.', ''), arch) | ||||
| ps = RUN_TESTS.format( | ps = RUN_TESTS.format( | ||||
| python_path=python_path, | python_path=python_path, | ||||
| test_flags=test_flags or '', | test_flags=test_flags or '', | ||||
| ) | ) | ||||
| run_powershell(winrm_client, ps) | run_powershell(winrm_client, ps) | ||||