121 lines
3.5 KiB
Python
121 lines
3.5 KiB
Python
# oscillode
|
|
# Copyright (C) 2024 oscillode Project Contributors
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
# blender_maxwell
|
|
# Copyright (C) 2024 blender_maxwell Project Contributors
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
# noqa: INP001
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import info
|
|
|
|
|
|
####################
|
|
# - Blender Runner
|
|
####################
|
|
def run_blender(
|
|
py_script: Path | None,
|
|
load_devfile: bool = False,
|
|
headless: bool = True,
|
|
monitor: bool = False,
|
|
):
|
|
process = subprocess.Popen(
|
|
[
|
|
'blender',
|
|
*(['--background'] if headless else []),
|
|
*(
|
|
[
|
|
'--python',
|
|
str(py_script),
|
|
]
|
|
if py_script is not None
|
|
else []
|
|
),
|
|
*([info.PATH_ADDON_DEV_BLEND] if load_devfile else []),
|
|
],
|
|
env=os.environ | {'PYTHONUNBUFFERED': '1'},
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT,
|
|
text=True,
|
|
)
|
|
output = []
|
|
printing_live = monitor
|
|
|
|
# Process Real-Time Output
|
|
for line in iter(process.stdout.readline, b''):
|
|
if not line:
|
|
break
|
|
|
|
if printing_live:
|
|
print(line, end='') # noqa: T201
|
|
elif (
|
|
info.SIGNAL_START_CLEAN_BLENDER in line
|
|
# or 'Traceback (most recent call last)' in line
|
|
):
|
|
printing_live = True
|
|
print(''.join(output)) # noqa: T201
|
|
else:
|
|
output.append(line)
|
|
|
|
# Wait for the process to finish and get the exit code
|
|
process.wait()
|
|
return process.returncode, output
|
|
|
|
|
|
####################
|
|
# - Main
|
|
####################
|
|
if __name__ == '__main__':
|
|
# Uninstall Addon
|
|
print(f'Blender: Uninstalling "{info.ADDON_NAME}"...')
|
|
return_code, output = run_blender(info.PATH_BL_DELETE_ADDON, monitor=False)
|
|
if return_code == info.STATUS_UNINSTALLED_ADDON:
|
|
print(f'\tBlender: Uninstalled "{info.ADDON_NAME}"')
|
|
elif return_code == info.STATUS_NOCHANGE_ADDON:
|
|
print(f'\tBlender: "{info.ADDON_NAME}" Not Installed')
|
|
|
|
# Install Addon
|
|
print(f'Blender: Installing & Enabling "{info.ADDON_NAME}"...')
|
|
return_code, output = run_blender(info.PATH_BL_INSTALL_ADDON, monitor=False)
|
|
if return_code == info.STATUS_INSTALLED_ADDON:
|
|
print(f'\tBlender: Install & Enable "{info.ADDON_NAME}"')
|
|
else:
|
|
print(f'\tBlender: "{info.ADDON_NAME}" Not Installed')
|
|
print(*output, sep='')
|
|
sys.exit(1)
|
|
|
|
# Run Addon
|
|
print(f'Blender: Running "{info.ADDON_NAME}"...')
|
|
return_code, output = run_blender(
|
|
None, headless=False, load_devfile=True, monitor=True
|
|
)
|