您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
66 行
3.1 KiB
66 行
3.1 KiB
import os
|
|
import shutil
|
|
import glob
|
|
from collections import OrderedDict
|
|
from unityparser import UnityDocument
|
|
|
|
# NOTE: This must match the flag defined in the Unity Integration Tests
|
|
_INTEGRATION_TEST_DEFINE = "INTEGRATION_TEST"
|
|
|
|
# This script is executed inside of a bokken image in order to automate the manual steps a user would
|
|
# perform when going through the tutorials. This allows us to perform integration tests on the expected final
|
|
# state of a tutorial project.
|
|
|
|
|
|
script_dir = os.path.dirname(os.path.realpath(__file__))
|
|
root_dir = os.path.join(script_dir, "..", "..", "tutorials", "pick_and_place")
|
|
external_scripts_dir = os.path.join(root_dir, "Scripts")
|
|
project_dir = os.path.join(root_dir, "PickAndPlaceProject")
|
|
project_scripts_dir = os.path.join(project_dir, "Assets", "Scripts")
|
|
# project_settings_file = os.path.join(project_dir, "ProjectSettings", "ProjectSettings.asset")
|
|
|
|
scripts_to_move = glob.glob(os.path.join(external_scripts_dir, "*.cs"))
|
|
for external_script in scripts_to_move:
|
|
script_name = os.path.basename(external_script)
|
|
script_destination = os.path.join(project_scripts_dir, script_name)
|
|
print(f">>> Copying {external_script} to {script_destination}")
|
|
shutil.copyfile(external_script, script_destination)
|
|
|
|
files_to_cat = []
|
|
message_dir = os.path.join(project_dir, "Assets", "RosMessages")
|
|
print(f">>> Files in {message_dir}:")
|
|
for root, _, files in os.walk(message_dir):
|
|
level = root.replace(message_dir, '').count(os.sep)
|
|
indent = ' ' * 4 * level
|
|
print('{}{}/'.format(indent, os.path.basename(root)))
|
|
subindent = ' ' * 4 * (level + 1)
|
|
for f in files:
|
|
print('{}{}'.format(subindent, f))
|
|
if f.endswith(".cs") or f.endswith(".asmdef"):
|
|
files_to_cat.append(os.path.join(root, f))
|
|
|
|
# On Yamato, Unity fails to recompile the message directory under mysterious circumstances, so moving it into
|
|
# the Scripts directory to attempt to force a recompile
|
|
print(f"Moving {message_dir} to {project_scripts_dir}")
|
|
shutil.move(message_dir, project_scripts_dir)
|
|
|
|
# for f in files_to_cat:
|
|
# print(f">>> {f}:")
|
|
# os.system(f"cat {f}")
|
|
# print("\n")
|
|
|
|
# We must keep a backup copy of the ProjectSettings.asset because UTR will serialize the one in the project as a binary
|
|
# file, making it impossible to load as a yaml file here
|
|
test_settings = os.path.join(".", ".yamato", "PickAndPlaceTests", "IntegrationTestSettings.asset")
|
|
project_settings_relative = os.path.join(".", "tutorials", "pick_and_place", "PickAndPlaceProject", "ProjectSettings", "ProjectSettings.asset")
|
|
shutil.copyfile(test_settings, project_settings_relative)
|
|
|
|
# TODO: We have to use the above, bad solution, for now, because the below solution won't work until
|
|
# UTR stops re-serializing the settings
|
|
# project_settings_asset = UnityDocument.load_yaml(project_settings_file)
|
|
# scripting_defines = project_settings_asset.entry.scriptingDefineSymbols # type: OrderedDict
|
|
# if scripting_defines[1]:
|
|
# scripting_defines[1] += f";{_INTEGRATION_TEST_DEFINE}"
|
|
# else:
|
|
# scripting_defines[1] = _INTEGRATION_TEST_DEFINE
|
|
# project_settings_asset.dump_yaml()
|