You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
2.1 KiB
Python
54 lines
2.1 KiB
Python
#
|
|
# All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
|
|
# its licensors.
|
|
#
|
|
# For complete copyright and license terms please see the LICENSE at the root of this
|
|
# distribution (the "License"). All use of this software is governed by the License,
|
|
# or, if provided, by the license below or the license accompanying this file. Do not
|
|
# remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
#
|
|
import marshal
|
|
from subprocess import Popen, PIPE, check_output
|
|
from typing import List
|
|
|
|
|
|
def run_p4_command(command: str, client: str = None, raw_output: bool = False) -> List[dict]:
|
|
"""Executes a Perforce command by calling p4 in a subprocess
|
|
|
|
:param command: The command to run (not including p4, e.g., not ['p4', 'opened'] but ['opened'])
|
|
:param client: The Perforce client to use when running the command
|
|
:param raw_output: if True, return the raw output from p4 instead of using the python dict.
|
|
:return: The list of results from the command where each result is a dict (refer to p4's global -G option)
|
|
if 'raw_output' is true, return will be verbatim string from p4.
|
|
"""
|
|
results = []
|
|
|
|
base_command = ['p4']
|
|
|
|
if not raw_output:
|
|
base_command.append('-G')
|
|
|
|
if client is not None:
|
|
base_command += ['-c', client]
|
|
|
|
if raw_output:
|
|
return check_output(base_command + command.split()).decode('utf8')
|
|
|
|
pipe = Popen(base_command + command.split(), stdout=PIPE).stdout
|
|
try:
|
|
while True:
|
|
result = marshal.load(pipe)
|
|
results.append(result)
|
|
except EOFError:
|
|
pass
|
|
finally:
|
|
pipe.close()
|
|
|
|
decoded_results = [{k.decode(): v.decode() if isinstance(v, bytes) else str(v) for k, v in r.items()} for r in results]
|
|
if decoded_results and decoded_results[0]['code'] == 'error':
|
|
command_was = ' '.join(base_command + command.split())
|
|
error_was = decoded_results[0]['data']
|
|
raise RuntimeError(f'P4 Command failed: "{command_was}"\nERROR FROM P4: {error_was}\n')
|
|
return decoded_results
|