forked from Mirror/pmbootstrap
31cc898dd5
With the new chroot type, we can now write fancy paths in the pythonic way. Convert most of the codebase over, as well as adding various other type hints. Signed-off-by: Caleb Connolly <caleb@postmarketos.org>
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
# Copyright 2023 Oliver Smith
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
import os
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
from pmb.core.types import PmbArgs
|
|
import pmb.parse
|
|
|
|
|
|
def find_path(args: PmbArgs, codename: str, file='') -> Optional[Path]:
|
|
"""Find path to device APKBUILD under `device/*/device-`.
|
|
|
|
:param codename: device codename
|
|
:param file: file to look for (e.g. APKBUILD or deviceinfo), may be empty
|
|
:returns: path to APKBUILD
|
|
"""
|
|
g = list((args.aports / "device").glob(f"*/device-{codename}/{file}"))
|
|
if not g:
|
|
return None
|
|
|
|
if len(g) != 1:
|
|
raise RuntimeError(codename + " found multiple times in the device"
|
|
" subdirectory of pmaports")
|
|
|
|
return g[0]
|
|
|
|
|
|
def list_codenames(args: PmbArgs, vendor=None, archived=True):
|
|
"""Get all devices, for which aports are available.
|
|
|
|
:param vendor: vendor name to choose devices from, or None for all vendors
|
|
:param archived: include archived devices
|
|
:returns: ["first-device", "second-device", ...]
|
|
"""
|
|
ret = []
|
|
for path in args.aports.glob("device/*/device-*"):
|
|
if not archived and 'archived' in path.parts:
|
|
continue
|
|
device = os.path.basename(path).split("-", 1)[1]
|
|
if (vendor is None) or device.startswith(vendor + '-'):
|
|
ret.append(device)
|
|
return ret
|
|
|
|
|
|
def list_vendors(args: PmbArgs):
|
|
"""Get all device vendors, for which aports are available.
|
|
|
|
:returns: {"vendor1", "vendor2", ...}
|
|
"""
|
|
ret = set()
|
|
for path in (args.aports / "device").glob("*/device-*"):
|
|
vendor = path.name.split("-", 2)[1]
|
|
ret.add(vendor)
|
|
return ret
|
|
|
|
|
|
def list_apkbuilds(args: PmbArgs):
|
|
""":returns: { "first-device": {"pkgname": ..., "pkgver": ...}, ... }"""
|
|
ret = {}
|
|
for device in list_codenames(args):
|
|
apkbuild_path = next(args.aports.glob(f"device/*/device-{device}/APKBUILD"))
|
|
ret[device] = pmb.parse.apkbuild(apkbuild_path)
|
|
return ret
|
|
|
|
|
|
def list_deviceinfos(args: PmbArgs):
|
|
""":returns: { "first-device": {"name": ..., "screen_width": ...}, ... }"""
|
|
ret = {}
|
|
for device in list_codenames(args):
|
|
ret[device] = pmb.parse.deviceinfo(args, device)
|
|
return ret
|