#To define a particular parameter, replace the 'parameterName' inside itsm.getParameter('parameterName') with that parameter's name

import ctypes
from subprocess import PIPE, Popen
import os
import ctypes

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)

def ecmd(command):
    with disable_file_system_redirection():
        obj = Popen(command, shell = True, stdout = PIPE, stderr = PIPE)
    out, err = obj.communicate()
    ret=obj.returncode
    if ret==0:
        if out:
            return out
        else:
            return ret
    else:
        if err:
            return err.strip()
        else:
            return ret
	
path = os.environ['PROGRAMDATA'] +'\diskscript.txt'

ps_content=r'''

Get-Disk

'''

file_name='get_disk.ps1'
file_path=os.path.join(os.environ['TEMP'], file_name)
with open(file_path, 'wb') as wr:
    wr.write(ps_content)

ecmd('powershell "Set-ExecutionPolicy RemoteSigned"')
list_disk = ecmd('powershell "%s"'%file_path)

for i in list_disk.split('\n')[3:]:
    try:
        if i[0].isdigit():
            print "Selected Disk: ",i[0]
            contents = """
            select disk %s
            clean
            """%(i[0])
            with disable_file_system_redirection(): 
                try:
                    with open(path, 'w') as fiOb:
                        fiOb.write(contents)
                except:
                    print 'Fail: Check your path!!'

            if os.path.isfile(path):
                command=r'diskpart /s '+ path
                print ecmd(command)
    except:
        pass