scripts: black

This commit is contained in:
Albert Hopkins 2018-10-27 11:15:59 -07:00
parent 5a96cc75a1
commit b2c92c5ec7
2 changed files with 33 additions and 40 deletions

View File

@ -8,7 +8,7 @@ import sys
def main():
chroot = os.environ["CHROOT"]
target = sys.argv[-1]
chroot_target = os.path.join(chroot, target.lstrip('/'))
chroot_target = os.path.join(chroot, target.lstrip("/"))
command = ["cp"] + sys.argv[1:-1] + [chroot_target]
subprocess.check_call(command)

View File

@ -4,86 +4,82 @@ import os
import shutil
import subprocess
EMERGE = os.environ['EMERGE'].split()
USEPKG = os.environ['USEPKG'].split()
EMERGE = os.environ["EMERGE"].split()
USEPKG = os.environ["USEPKG"].split()
def get_latest_available_kernel():
kernel = os.environ['KERNEL']
package_name = 'sys-kernel/{}'.format(kernel)
kernel = os.environ["KERNEL"]
package_name = "sys-kernel/{}".format(kernel)
popen = subprocess.Popen(
['portageq', 'best_visible', '/', package_name],
stdout=subprocess.PIPE,
["portageq", "best_visible", "/", package_name], stdout=subprocess.PIPE
)
cpv = popen.stdout.read().decode().strip()
return cpv.rpartition('-')[2]
return cpv.rpartition("-")[2]
def get_current_kernel():
"""what kernel does /boot/vmlinuz point to"""
vmlinuz = '/boot/vmlinuz'
vmlinuz = "/boot/vmlinuz"
if not os.path.islink(vmlinuz):
return None
filename = os.path.basename(os.path.realpath(vmlinuz))
assert filename.startswith('vmlinuz')
assert filename.startswith("vmlinuz")
part_after_vmlinuz = filename[8:]
# may have "-gentoo" or whatever in the name
version = part_after_vmlinuz.rpartition('-')[0]
version = part_after_vmlinuz.rpartition("-")[0]
return version
def install_kernel_package():
kernel = os.environ['KERNEL']
package_name = 'sys-kernel/{}'.format(kernel)
cmd = EMERGE + USEPKG + ['--oneshot', '--noreplace', package_name]
kernel = os.environ["KERNEL"]
package_name = "sys-kernel/{}".format(kernel)
cmd = EMERGE + USEPKG + ["--oneshot", "--noreplace", package_name]
subprocess.check_call(cmd)
def copy_kernel_config():
filename = '/root/kernel.config'
filename = "/root/kernel.config"
shutil.copy(filename, '/usr/src/linux/.config')
shutil.copy(filename, "/usr/src/linux/.config")
def build_kernel():
makeopts = os.environ.get('MAKEOPTS', '')
makeopts = os.environ.get("MAKEOPTS", "")
subprocess.check_call([
'make', '-C', '/usr/src/linux', 'MAKEOPTS=' + makeopts, 'oldconfig'
])
subprocess.check_call([
'make', '-C', '/usr/src/linux', 'MAKEOPTS=' + makeopts
])
subprocess.check_call(
["make", "-C", "/usr/src/linux", "MAKEOPTS=" + makeopts, "oldconfig"]
)
subprocess.check_call(["make", "-C", "/usr/src/linux", "MAKEOPTS=" + makeopts])
def remove_old_kernels():
kernel_files = ['vmlinuz-', 'System.map-', 'config-']
kernel_files = ["vmlinuz-", "System.map-", "config-"]
for filename in os.listdir('/boot'):
for filename in os.listdir("/boot"):
for kernel_file in kernel_files:
if filename.startswith(kernel_file):
path = os.path.join('/boot', filename)
path = os.path.join("/boot", filename)
os.unlink(path)
if os.path.exists('/boot/vmlinuz'):
os.unlink('/boot/vmlinuz')
if os.path.exists("/boot/vmlinuz"):
os.unlink("/boot/vmlinuz")
shutil.rmtree('/lib/modules', ignore_errors=True)
shutil.rmtree("/lib/modules", ignore_errors=True)
def install_kernel():
makeopts = os.environ.get('MAKEOPTS', '')
makeopts = os.environ.get("MAKEOPTS", "")
subprocess.check_call([
'make', '-C', '/usr/src/linux', 'MAKEOPTS=' + makeopts, 'install',
'modules_install'
])
subprocess.check_call(
["make", "-C", "/usr/src/linux", "MAKEOPTS=" + makeopts, "install", "modules_install"]
)
# create the symlink. /sbin/installkernel claims this is not used on
# "modern" distributions and doesn't create it (unless it already
@ -97,15 +93,13 @@ def install_kernel():
def uninstall_kernel_package():
subprocess.check_call([
'make', '-C', '/usr/src/linux', 'distclean'
])
subprocess.check_call(["make", "-C", "/usr/src/linux", "distclean"])
subprocess.check_call(EMERGE + USEPKG + ['--depclean', '--with-bdeps=n'])
subprocess.check_call(EMERGE + USEPKG + ["--depclean", "--with-bdeps=n"])
def backup_kernel_config():
shutil.copy('/usr/src/linux/.config', '/root/kernel.config')
shutil.copy("/usr/src/linux/.config", "/root/kernel.config")
def main():
@ -126,4 +120,3 @@ def main():
main()