RomanNum3ral
/
archinstall
mirror of https://github.com/archlinux/archinstall.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Extract custom-commands function into general.py

This commit is contained in:
Dylan Taylor 2021-05-20 16:51:05 -04:00
parent 11c9632322
commit 5652ba20ae
2 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
archinstall/lib/general.py
View File

@ -366,3 +366,13 @@ def pid_exists(pid: int):
return any(subprocess.check_output(['/usr/bin/ps', '--no-headers', '-o', 'pid', '-p', str(pid)]).strip())
except subprocess.CalledProcessError:
return False
def run_custom_user_commands(commands):
for index, command in enumerate(commands):
log(f'Executing custom command "{command}" ...', fg='yellow')
with open(f"/mnt/var/tmp/user-command.{index}.sh", "w") as temp_script:
temp_script.write(command)
execution_output = SysCommand(f"arch-chroot /mnt bash /var/tmp/user-command.{index}.sh")
log(execution_output)
os.unlink(f"/mnt/var/tmp/user-command.{index}.sh")

13
examples/guided.py
View File

@ -4,10 +4,9 @@ import os
import time
import archinstall
from archinstall.lib.general import SysCommand
from archinstall.lib.general import run_custom_user_commands
from archinstall.lib.hardware import has_uefi
from archinstall.lib.networking import check_mirror_reachable
from archinstall.lib.profiles import Profile
if archinstall.arguments.get('help'):
print("See `man archinstall` for help.")
@ -378,14 +377,8 @@ def perform_installation(mountpoint):
exit(1)
# If the user provided custom commands to be run post-installation, execute them now.
if len(archinstall.arguments['custom-commands']):
for index, command in enumerate(archinstall.arguments['custom-commands']):
archinstall.log(f'Executing custom command "{command}" ...', fg='yellow')
with open(f"/mnt/var/tmp/user-command.{index}.sh", "w") as temp_script:
temp_script.write(command)
execution_output = SysCommand(f"arch-chroot /mnt bash /var/tmp/user-command.{index}.sh")
archinstall.log(execution_output)
os.unlink(f"/mnt/var/tmp/user-command.{index}.sh")
if archinstall.arguments.get('custom-commands', None):
run_custom_user_commands(archinstall.arguments['custom-commands'])
installation.log("For post-installation tips, see https://wiki.archlinux.org/index.php/Installation_guide#Post-installation", fg="yellow")
if not archinstall.arguments.get('silent'):