1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#############################################################################
#
# Copyright (C) 2022 The Qt Company Ltd.
# Contact: https://www.qt.io/licensing/
#
# This file is part of the release tools of the Qt Toolkit.
#
# $QT_BEGIN_LICENSE:GPL-EXCEPT$
# Commercial License Usage
# Licensees holding valid commercial Qt licenses may use this file in
# accordance with the commercial license agreement provided with the
# Software or, alternatively, in accordance with the terms contained in
# a written agreement between you and The Qt Company. For licensing terms
# and conditions see https://www.qt.io/terms-conditions. For further
# information use the contact form at https://www.qt.io/contact-us.
#
# GNU General Public License Usage
# Alternatively, this file may be used under the terms of the GNU
# General Public License version 3 as published by the Free Software
# Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT
# included in the packaging of this file. Please review the following
# information to ensure the GNU General Public License requirements will
# be met: https://www.gnu.org/licenses/gpl-3.0.html.
#
# $QT_END_LICENSE$
#
#############################################################################
import itertools
import subprocess
import os
# http://stackoverflow.com/questions/1214496/how-to-get-environment-from-a-subprocess-in-python
#def validate_pair(ob):
# try:
# if not (len(ob) == 2):
# print("Unexpected result:", ob, file = sys.stderr)
# raise ValueError
# except:
# return False
# return True
def consume(iterator):
try:
while True:
next(iterator)
except StopIteration:
pass
class PackagingError(Exception):
pass
def sanity_check_env(env_cmd, env_dict):
# throw error if 'Path' exists in environment dictionary
if "Path" in env_dict:
raise PackagingError("Configuration error, possible duplicate path 'Path' found in {0}"
.format(env_cmd))
def get(env_cmd, initial = None, arguments = None):
"""
Take a command (either a single command or list of arguments)
and return the environment created after running that command.
Note that if the command must be a batch file or .cmd file, or the
changes to the environment will not be captured.
If initial is supplied, it is used as the initial environment passed
to the child process.
"""
if not os.path.lexists(env_cmd):
raise Exception("Can not find {0} to get an environment from it.".format(env_cmd))
#if not isinstance(env_cmd, (list, tuple)):
# env_cmd = [env_cmd]
# construct the command that will alter the environment
#env_cmd = subprocess.list2cmdline(env_cmd)
# create a tag so we can tell in the output when the proc is done
tag = 'Done running command'
# construct a cmd.exe command to do accomplish this
cmd = 'cmd.exe /s /c "\"{env_cmd}\" {arguments}&& echo "{tag}" && set"'.format(**vars())
# launch the process
proc = subprocess.Popen(cmd, stdout = subprocess.PIPE, env = initial, universal_newlines = True)
# parse the output sent to stdout
lines = proc.stdout
# consume whatever output occurs until the tag is reached
consume(itertools.takewhile(lambda line: tag not in line, lines))
# parse key/values into pairs
pairs = (line.rstrip().split('=', 1) for line in lines)
# make sure the pairs are valid
valid_pairs = (pair for pair in pairs if len(pair) == 2)
# construct a dictionary of the pairs
result = dict(valid_pairs)
# let the process finish
proc.communicate()
# check resulting env for duplicate path
sanity_check_env(env_cmd, result)
return result
|