ereuse
/
workbench-script
8
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

use of sanitize in workbench-script

This commit is contained in:
Cayo Puigdefabregas 2024-10-18 12:16:00 +02:00
parent be28ec2a1f
commit 731567b8ea
1 changed files with 7 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
workbench-script.py
View File

@ -76,45 +76,10 @@ SNAPSHOT_BASE = {
## Command Functions ## ## Command Functions ##
@logs @logs
def get_disks(): def gen_erase(type_erase, user_disk=None):
disks = json.loads( if user_disk:
exec_cmd('lsblk -Jdo NAME,TYPE,MOUNTPOINTS,ROTA,TRAN') return exec_cmd(f"sanitize -d {user_disk} -m {type_erase} --confirm")
) return exec_cmd(f"sanitize -a -m {type_erase} --confirm")
return disks.get('blockdevices', [])
@logs
def gen_erase(all_disks, type_erase, user_disk=None):
erase = []
for disk in all_disks:
if user_disk and disk['name'] not in user_disk:
continue
if disk['type'] != 'disk':
continue
if 'boot' in disk['mountpoints']:
continue
if not disk['rota']:
# if soport nvme erase
erase.append(nvme_secure_erase(disk['name']))
elif disk['tran'] in ['ata', 'sata']:
# if soport ata erase
if type_erase == 'basic':
erase.append(ata_secure_erase_null(disk['name']))
elif type_erase == 'baseline':
erase.append(ata_secure_erase_null(disk['name']))
elif type_erase == 'enhanced':
erase.append(ata_secure_erase_enhanced(disk['name']))
else:
# For old disks
if type_erase == 'basic':
erase.append(erase_basic(disk['name']))
elif type_erase == 'baseline':
erase.append(erase_baseline(disk['name']))
elif type_erase == 'enhanced':
erase.append(erase_enhanced(disk['name']))
return erase
@logs @logs
@ -314,16 +279,12 @@ def main():
if os.geteuid() != 0: if os.geteuid() != 0:
logger.warning(_("This script must be run as root. Collected data will be incomplete or unusable")) logger.warning(_("This script must be run as root. Collected data will be incomplete or unusable"))
all_disks = get_disks()
snapshot = gen_snapshot(all_disks) snapshot = gen_snapshot(all_disks)
if config['erase'] and config['device'] and not config.get("legacy"): if config.get("legacy")
snapshot['erase'] = gen_erase(all_disks, config['erase'], user_disk=config['device'])
elif config['erase'] and not config.get("legacy"):
snapshot['erase'] = gen_erase(all_disks, config['erase'])
if config.get("legacy"):
convert_to_legacy_snapshot(snapshot) convert_to_legacy_snapshot(snapshot)
else:
snapshot['erase'] = gen_erase(config['erase'].upper(), user_disk=config['device'])
save_snapshot_in_disk(snapshot, config['path']) save_snapshot_in_disk(snapshot, config['path'])