#To define a particular parameter, replace the 'parameterName' inside itsm.getParameter('parameterName') with that parameter's name
import os
from subprocess import PIPE,Popen
import ctypes
import time
import sys

start=time.time()

def alert(arg):
    return sys.stderr.write("%d%d%d" % (arg, arg, arg))

class disable_file_system_redirection:
    _disable = ctypes.windll.kernel32.Wow64DisableWow64FsRedirection
    _revert = ctypes.windll.kernel32.Wow64RevertWow64FsRedirection
    def __enter__(self):
        self.old_value = ctypes.c_long()
        self.success = self._disable(ctypes.byref(self.old_value))
    def __exit__(self, type, value, traceback):
        if self.success:
            self._revert(self.old_value)
with disable_file_system_redirection():
    print "Process Test 1 SMART Status Check"
    process1=Popen("wmic diskdrive get status,caption",stdout=PIPE,stderr=PIPE)
    r1,e1=process1.communicate()
    if e1:
        print e1
    else:
        r1=r1.splitlines()
        print r1[0]
        for i in r1[1:]:
            print i
    print "Process Test 2 SMART Drive Predictive Failure"
    process2=Popen(r"wmic /namespace:\\root\wmi path MSStorageDriver_FailurePredictStatus",stdout=PIPE,stderr=PIPE)
    r2,e2=process2.communicate()
    if e2:
        print e2
    else:
        print r2
        r2=r2.splitlines()
        for i in r2[1:]:
            if not i:
                pass
            else:
                check=i.split()[1:]
                if check[1]=='FALSE':
                    print "Hard Drive is working no failure or error in %s"%check[0]
                    alert(0)
                else:
                    print "Failure In Disk %s"%check[0]
                    alert(1)
		
end=time.time()
print("Execution Time:%s"%(end-start))	