chromium/chrome/updater/test/test_installer/test_app1_setup.py

#!/usr/bin/env python3
# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Example of setup script that writes the app 'pv' value into the registry."""

import argparse
import sys
import winreg


def main():
    parser = argparse.ArgumentParser(
        description=__doc__,
        formatter_class=argparse.RawDescriptionHelpFormatter)
    parser.add_argument('--company',
                        default='Google',
                        required=False,
                        help='App ID.')
    parser.add_argument('--appid',
                        default='{AE098195-B8DB-4A49-8E23-84FCACB61FF1}',
                        required=False,
                        help='App ID.')
    parser.add_argument('--system',
                        default=False,
                        required=False,
                        action='store_true',
                        help='Specify that this is a system install.')
    parser.add_argument('--product_version',
                        default='1.0.0.0',
                        required=False,
                        help='App version to install.')

    args = parser.parse_args()

    try:
        reg_hive = winreg.ConnectRegistry(
            None, winreg.HKEY_LOCAL_MACHINE
            if args.system else winreg.HKEY_CURRENT_USER)
        update_clients_key = winreg.CreateKeyEx(
            reg_hive,
            'Software\\%s\\Update\\Clients\\%s' % (args.company, args.appid),
            access=winreg.KEY_WOW64_32KEY | winreg.KEY_WRITE)
        winreg.SetValueEx(update_clients_key, 'pv', 0, winreg.REG_SZ,
                          args.product_version)
    except WindowsError:
        sys.exit(1)

    sys.exit(0)


if __name__ == '__main__':
    main()