#To define a particular parameter, replace the 'parameterName' inside itsm.getParameter('parameterName') with that parameter's name
import ctypes
import subprocess
import sys

TotalDays = itsm.getParameter('TotalDays')


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 ExecuteCmd(cmd):
    with disable_file_system_redirection():
        obj = subprocess.Popen(["powershell", cmd], shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        out, err = obj.communicate()
        return out, err


def alert(arg):
    sys.stderr.write("%d%d%d" % (arg, arg, arg))


cmd = r'((get-date) - (Get-WinEvent -FilterHashtable @{LogName="System";ID=7001;} |select-object -First ' \
      r'1).TimeCreated ).Days '


res = ExecuteCmd(cmd)
if res[1] == '':
    print ("\nSystem Total Uptime:")
    NoDays = res[0].strip()
    print NoDays + " days"
    if NoDays >= TotalDays:
        ctypes.windll.user32.MessageBoxW(None, u'PC need to be restarted', u'Warning', 0)
        alert(1)
        print("PC need to be restarted")
    else:
        alert(0)
else:
    print (res[1])
