Plan de particionado: GPT o MBR, UEFI o BIOS y alineación a 1 MiB

1import re
2import sys
3
4MIB = 1024 * 1024
5SECTOR = 512
6UNIDADES = {"MIB": MIB, "GIB": 1024 * MIB, "TIB": 1024 * 1024 * MIB, "MB": 10 ** 6, "GB": 10 ** 9, "TB": 10 ** 12}
7
8
9def tam(bytes_):
10    """Tamaño legible en binario con dos decimales y coma: 59,99 GiB."""
11    for unidad, valor in [("TiB", UNIDADES["TIB"]), ("GiB", UNIDADES["GIB"]), ("MiB", MIB)]:
12        if bytes_ >= valor:
13            return f"{bytes_ / valor:.2f}".replace(".", ",") + " " + unidad
14    return f"{bytes_} B"
15
16
17def main():
18    disco = None
19    tabla = arranque = None
20    particiones = []
21    errores = []
22    for n, linea in enumerate(sys.stdin.read().split("\n"), start=1):
23        p = linea.split()
24        if not p:
25            continue
26        clave = p[0].upper()
27        if clave == "DISCO" and len(p) == 3 and re.fullmatch(r"\d+", p[1]) and p[2].upper() in UNIDADES:
28            disco = int(p[1]) * UNIDADES[p[2].upper()]
29        elif clave == "TABLA" and len(p) == 2 and p[1].upper() in ("GPT", "MBR"):
30            tabla = p[1].upper()
31        elif clave == "ARRANQUE" and len(p) == 2 and p[1].upper() in ("UEFI", "BIOS"):
32            arranque = p[1].upper()
33        elif clave == "PARTICION" and len(p) == 3 and p[2].upper() == "RESTO":
34            particiones.append((p[1], None))
35        elif clave == "PARTICION" and len(p) == 4 and re.fullmatch(r"\d+", p[2]) and p[3].upper() in UNIDADES:
36            particiones.append((p[1], int(p[2]) * UNIDADES[p[3].upper()]))
37        else:
38            errores.append(f"Línea {n}: no se entiende «{linea.strip()}»")
39    if disco is None or tabla is None or arranque is None:
40        errores.append("Faltan DISCO, TABLA o ARRANQUE")
41    for e in errores:
42        print(e)
43    if errores:
44        return
45    print(f"Disco de {tam(disco)} ({disco // SECTOR} sectores de {SECTOR} bytes), tabla {tabla}, arranque {arranque}")
46    avisos = []
47    if tabla == "MBR" and disco > 2 * UNIDADES["TIB"]:
48        avisos.append("MBR solo direcciona 2 TiB: el resto del disco no se podría usar; usa GPT")
49    puntos = [nombre for nombre, _ in particiones]
50    if arranque == "UEFI" and "/boot/efi" not in puntos:
51        avisos.append("Con UEFI hace falta una partición de sistema EFI (/boot/efi, FAT32, de al menos 100 MiB)")
52    if arranque == "BIOS" and tabla == "GPT" and "bios_grub" not in puntos:
53        avisos.append("Con BIOS y GPT, GRUB necesita una pequeña partición bios_grub (1 MiB)")
54    if "/" not in puntos:
55        avisos.append("No hay partición raíz /")
56    efi = [t for nombre, t in particiones if nombre == "/boot/efi"]
57    if efi and efi[0] is not None and efi[0] < 100 * MIB:
58        avisos.append("La partición EFI debería tener al menos 100 MiB")
59    resto = [i for i, (_, t) in enumerate(particiones) if t is None]
60    if len(resto) > 1 or (resto and resto[0] != len(particiones) - 1):
61        print("Error: RESTO solo puede usarse una vez y en la última partición")
62        return
63    # alineación a 1 MiB: la primera empieza en el sector 2048 y GPT reserva su copia en el último MiB
64    inicio = MIB
65    fin_util = disco - (MIB if tabla == "GPT" else 0)
66    fin_util -= fin_util % MIB
67    if tabla == "MBR" and len(particiones) > 4:
68        avisos.append(f"MBR admite 4 particiones primarias: la 4.ª será extendida y las {len(particiones) - 3} últimas, lógicas")
69    filas = []
70    cabe = True
71    for i, (nombre, t) in enumerate(particiones):
72        if tabla == "MBR" and len(particiones) > 4 and i >= 3:
73            inicio += MIB                       # cada lógica empieza con su EBR, un MiB por alineación
74        tamano = (fin_util - inicio) if t is None else -(-t // MIB) * MIB
75        if tamano <= 0 or inicio + tamano > fin_util:
76            print(f"Error: {nombre} no cabe: quedan {tam(max(0, fin_util - inicio))}")
77            cabe = False
78            break
79        numero = i + 1 if not (tabla == "MBR" and len(particiones) > 4 and i >= 3) else i + 2
80        filas.append((numero, nombre, inicio // SECTOR, (inicio + tamano) // SECTOR - 1, tamano))
81        inicio += tamano
82    if cabe:
83        print("Nº  Partición        Inicio        Fin           Tamaño")
84        for numero, nombre, ini, fin, tamano in filas:
85            print(f"{numero:<3} {nombre:<16} {ini:<13} {fin:<13} {tam(tamano)}")
86        libre = fin_util - inicio
87        print(f"Sin asignar: {tam(libre) if libre else '0'}")
88    for a in avisos:
89        print(f"Aviso: {a}")
90    if cabe and not avisos:
91        print("Plan correcto")
92
93
94main()