From d71e261829c650e0c6921a7fb4912dde4a01f987 Mon Sep 17 00:00:00 2001
From: Natalia Serrano <natalia.serrano@qindel.com>
Date: Fri, 14 Feb 2025 14:02:24 +0100
Subject: [PATCH] refs #1512 add ogUnhidePartition

---
 client/lib/engine/bin/DiskLib.py          | 48 ++++++-----------------
 client/shared/functions/ogUnhidePartition | 23 +++++++++++
 2 files changed, 35 insertions(+), 36 deletions(-)
 create mode 100755 client/shared/functions/ogUnhidePartition

diff --git a/client/lib/engine/bin/DiskLib.py b/client/lib/engine/bin/DiskLib.py
index 9493ac4..8fe5b9e 100755
--- a/client/lib/engine/bin/DiskLib.py
+++ b/client/lib/engine/bin/DiskLib.py
@@ -1381,45 +1381,21 @@ def ogTypeToId (type, pttype='MSDOS'):
 #@exception OG_ERR_NOTFOUND  disco o particion no detectado (no es un dispositivo).
 #@exception OG_ERR_PARTITION tipo de partición no reconocido.
 #*/ ##
-def ogUnhidePartition(*args):
-    # Variables locales
-    PART = None
-    TYPE = None
-    NEWTYPE = None
+def ogUnhidePartition (disk, par):
+    PART = ogDiskToDev (disk, par)
+    if not PART: return None
 
-    # Si se solicita, mostrar ayuda.
-    if len(args) == 1 and args[0] == "help":
-        SystemLib.ogHelp('ogUnhidePartition', 'ogUnhidePartition int_ndisk int_npartition', 'ogUnhidePartition 1 1')
-        return
-
-    # Error si no se reciben 2 parámetros.
-    if len(args) != 2:
-        SystemLib.ogRaiseError(OG_ERR_FORMAT)
-        return
-
-    PART = ogDiskToDev(args[0], args[1])
-    if PART is None:
-        return
-
-    # Obtener tipo de partición.
-    TYPE = ogGetPartitionType(args[0], args[1])
-    if TYPE == "HNTFS":
-        NEWTYPE = "NTFS"
-    elif TYPE == "HFAT32":
-        NEWTYPE = "FAT32"
-    elif TYPE == "HFAT16":
-        NEWTYPE = "FAT16"
-    elif TYPE == "HFAT12":
-        NEWTYPE = "FAT12"
-    elif TYPE == "WIN-RESERV":
-        NEWTYPE = "WINDOWS"
+    TYPE = ogGetPartitionType (disk, par)
+    if   'HNTFS'      == TYPE: NEWTYPE = 'NTFS'
+    elif 'HFAT32'     == TYPE: NEWTYPE = 'FAT32'
+    elif 'HFAT16'     == TYPE: NEWTYPE = 'FAT16'
+    elif 'HFAT12'     == TYPE: NEWTYPE = 'FAT12'
+    elif 'WIN-RESERV' == TYPE: NEWTYPE = 'WINDOWS'
     else:
-        SystemLib.ogRaiseError(OG_ERR_PARTITION, TYPE)
-        return
+        SystemLib.ogRaiseError ([], ogGlobals.OG_ERR_PARTITION, TYPE)
+        return None
 
-    # Cambiar tipo de partición.
-    ogSetPartitionType(args[0], args[1], NEWTYPE)
-    return
+    ogSetPartitionType (disk, par, NEWTYPE)
 
 
 #/**
diff --git a/client/shared/functions/ogUnhidePartition b/client/shared/functions/ogUnhidePartition
new file mode 100755
index 0000000..539303e
--- /dev/null
+++ b/client/shared/functions/ogUnhidePartition
@@ -0,0 +1,23 @@
+#!/usr/bin/python3
+
+import sys
+import argparse
+from SystemLib import ogHelp
+from DiskLib import ogUnhidePartition
+
+parser = argparse.ArgumentParser (add_help=False)
+parser.add_argument ('disk')
+parser.add_argument ('par')
+
+if 2 == len (sys.argv) and 'help' == sys.argv[1]:
+    #parser.print_help() sale en inglés aunque la locale indique otra cosa
+    ogHelp ('ogUnhidePartition', 'ogUnhidePartition int_ndisk int_npartition', ['ogUnhidePartition 1 1'])
+    sys.exit (0)
+
+args = parser.parse_args()
+ret = ogUnhidePartition (args.disk, args.par)
+
+if ret is not None:
+    if   ret == True: sys.exit (0)
+    elif ret == False: sys.exit (1)
+    else: print (ret)