narenas
/
test-ogclone
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refs #1513 add ogSetPartitionId

code-review
Natalia Serrano 2025-02-14 13:54:20 +01:00
parent 788ae31b0f
commit 0d8d543526
1 changed files with 24 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
client/lib/engine/bin/DiskLib.py
View File

@ -10,6 +10,7 @@ import filecmp
import subprocess import subprocess
import shutil import shutil
import os import os
import re
import stat import stat
from pathlib import Path from pathlib import Path
@ -1167,52 +1168,35 @@ def ogSetPartitionActive (disk, par):
#@exception OG_ERR_PARTITION Error al cambiar el id. de partición. #@exception OG_ERR_PARTITION Error al cambiar el id. de partición.
#@attention Requisitos: fdisk, sgdisk #@attention Requisitos: fdisk, sgdisk
#*/ ## #*/ ##
def ogSetPartitionId(*args): def ogSetPartitionId (disk, par, id):
# Variables locales DISK = ogDiskToDev (disk)
DISK = None if not DISK: return None
PART = None
PTTYPE = None
ID = None
# Si se solicita, mostrar ayuda. PART = ogDiskToDev (disk, par)
if len(args) == 1 and args[0] == "help": if not PART: return None
SystemLib.ogHelp('ogSetPartitionId', 'ogSetPartitionId int_ndisk int_npartition hex_partid', 'ogSetPartitionId 1 1 7')
return
# Error si no se reciben 3 parámetros.
if len(args) != 3:
SystemLib.ogRaiseError(OG_ERR_FORMAT)
return
# Sustituye nº de disco y nº partición por su dispositivo.
DISK = ogDiskToDev(args[0])
if DISK is None:
return
PART = ogDiskToDev(args[0], args[1])
if PART is None:
return
# Error si el id. de partición no es hexadecimal. # Error si el id. de partición no es hexadecimal.
ID = args[2].upper() if not re.match ('^[0-9A-Fa-f]+$', id):
if not re.match("^[0-9A-F]+$", ID): SystemLib.ogRaiseError ([], ogGlobals.OG_ERR_OUTOFLIMIT, id)
SystemLib.ogRaiseError(OG_ERR_OUTOFLIMIT, args[2]) return None
return
# Elección del tipo de partición. # Elección del tipo de partición.
PTTYPE = ogGetPartitionTableType(args[0]) PTTYPE = ogGetPartitionTableType (disk)
if PTTYPE == "GPT": if 'GPT' == PTTYPE:
subprocess.run(["sgdisk", f"-t{args[1]}:{ID}", DISK], stderr=subprocess.DEVNULL) p = subprocess.run (['sgdisk', f'-t{par}:{id.upper()}', DISK])
elif PTTYPE == "MSDOS": elif 'MSDOS' == PTTYPE:
subprocess.run(["sfdisk", f"--id", DISK, args[1], ID], stderr=subprocess.DEVNULL) p = subprocess.run (['sfdisk', '--id', DISK, par, id.upper()])
else: else:
SystemLib.ogRaiseError(OG_ERR_OUTOFLIMIT, f"{args[0]},{PTTYPE}") SystemLib.ogRaiseError ([], ogGlobals.OG_ERR_OUTOFLIMIT, f'{disk},{PTTYPE}')
return return None
# MSDOS) Correcto si fdisk sin error o con error pero realiza Syncing # MSDOS) Correcto si fdisk sin error o con error pero realiza Syncing
if subprocess.run(["partprobe", DISK], stderr=subprocess.DEVNULL).returncode == 0: if 0 == p.returncode:
return subprocess.run (['partprobe', DISK])
return True
else: else:
SystemLib.ogRaiseError(OG_ERR_PARTITION, f"{args[0]},{args[1]},{args[2]}") SystemLib.ogRaiseError ([], ogGlobals.OG_ERR_PARTITION, f'{disk},{part},{id}')
return return None
#/** #/**