Skip to content
Snippets Groups Projects
Commit bbec5902 authored by efer's avatar efer
Browse files

check for nvme ses capability

parent 621a9701
No related branches found
No related tags found
No related merge requests found
......@@ -13,8 +13,9 @@
__version__ = '0.1'
import os
import json
from time import sleep
from subprocess import run, CalledProcessError
from subprocess import run, CalledProcessError, PIPE
try:
import pyudev
except ImportError:
......@@ -148,7 +149,14 @@ def unfreeze():
def erase(disk):
print(YELLOW+"Erasing "+disk+WHITE)
if DINFO[disk][1] == 'NVMe':
cmd = ['nvme', 'format', disk, '--ses=2']
checkcap = ['nvme', 'id-ctrl', disk, '-o', 'json']
getcap = run(checkcap, stdout=PIPE, check=True)
nvmecap = json.loads(getcap.stdout)
if nvmecap['fna'] == 5:
ses = '--ses=2'
else:
ses = '--ses=1'
cmd = ['nvme', 'format', disk, ses]
try:
run(cmd, check=True)
sleep(20)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment