forked from Mirror/pmbootstrap
34dd9d42ba
Cease merging pmbootstrap.cfg into args, implement a Context type to let us pull globals out of thin air (as an intermediate workaround) and rip args out of a lot of the codebase. This is just a first pass, after this we can split all the state that leaked over into Context into types with narrower scopes (like a BuildContext(), etc). Signed-off-by: Caleb Connolly <caleb@postmarketos.org>
53 lines
1.6 KiB
Python
53 lines
1.6 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 import get_context
|
|
|
|
|
|
def find_path(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((get_context().config.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(aports: Path, 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 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(aports: Path):
|
|
"""Get all device vendors, for which aports are available.
|
|
|
|
:returns: {"vendor1", "vendor2", ...}
|
|
"""
|
|
ret = set()
|
|
for path in (aports / "device").glob("*/device-*"):
|
|
vendor = path.name.split("-", 2)[1]
|
|
ret.add(vendor)
|
|
return ret
|