# -*- coding: utf-8 -*-
from setuphelpers import *
import psutil
import time
import datetime
import ctypes
try:
from waptenterprise.waptservice.enterprise import get_active_sessions, start_interactive_process
except:
from waptservice.enterprise import get_active_sessions, start_interactive_process
import win32api
import sys
import platform
# Usable WAPT package functions: install(), uninstall(), session_setup(), audit(), update_package()
# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls
allow_force_reboot = False
boot_limit = 7 # In days
boot_limit_max = 30 # In days
force_reboot_timeout = 7200 # In seconds
message_box_title = "Information message about your PC uptime - The IT Support"
message_box_title_fr = "Message d'information à propos de la durée d'allumage du PC - Le Support Informatique"
# boot_limit_message = "This PC must be reboot because he's up since too long, Would you like to reboot now?"
boot_limit_message_string = "This PC must be rebooted because he's up since: %s, Would you like to reboot now?"
boot_limit_message_fr_string = "Ce PC doit être redémarré, car il est allumé depuis : %s, Voulez-vous redémarrer maintenant ?"
# boot_limit_max_message = "This PC will be FORCE reboot in 2 hours because he's up since too long"
boot_limit_max_message_string = "This PC will be FORCE reboot in %s hours because he's up since: %s"
boot_limit_max_message_fr_string = "Ce PC va être redémarré de manière FORCÉ dans %s heures, car il est allumé depuis : %s"
def install():
pass
r""" # Initializing variables
wapt_enterprise_dir = makepath(WAPT.wapt_base_dir,'waptenterprise')
# Checking that the WAPT Agent is well in Enterprise version
if isdir(wapt_enterprise_dir):
print('Continuing: WAPT Agent is well installed in Enterprise version')
else:
error('WAPT Agent is NOT installed in Enterprise version, please upgrade')
"""
def session_setup():
print("This PC is up since:")
print(convert_time_to_date(uptime_seconds()))
# Translating message
if get_language() == "fr":
message_box_title = message_box_title_fr
boot_limit_message_string = boot_limit_message_fr_string
boot_limit_max_message_string = boot_limit_max_message_fr_string
# Avoiding user pop-up spamming
check_timeout = 1800
date_format = "%Y-%m-%d %H:%M:%S" # datetime.datetime.strptime(date_pwsh, "%Y/%m/%d %H:%M:%S")
date = run_powershell('Get-Date -Format "yyyy-MM-dd HH:mm:ss"') # datetime.datetime.now()
date_reg = registry_readstring(HKEY_CURRENT_USER, r"SOFTWARE\WAPT\Shutdown Information", "Last Check")
if date_reg != "":
date_diff = datetime.datetime.strptime(date, date_format) - datetime.datetime.strptime(date_reg, date_format)
if date_diff.total_seconds() < check_timeout:
return "Already checked"
registry_set(HKEY_CURRENT_USER, r"SOFTWARE\WAPT\Shutdown Information", "Last Check", date)
# Notify the user
if allow_force_reboot:
if uptime_days() > boot_limit_max:
message_box(
message_box_title,
boot_limit_max_message_string % (convert_time_to_hours(force_reboot_timeout), convert_time_to_date(uptime_seconds(), translate=True)),
)
reboot_machine(force=1, timeout=force_reboot_timeout)
return "ERROR"
if uptime_days() > boot_limit:
message_box_ask_reboot(message_box_title, boot_limit_message_string % convert_time_to_date(uptime_seconds(), translate=True))
return "WARNING"
if uptime_days() < boot_limit:
return "OK"
def audit():
print("This PC is up since:")
print(convert_time_to_date(uptime_seconds()))
if allow_force_reboot:
if uptime_days() > boot_limit_max:
run_session_setup(control.package)
print("Force rebooting the PC after the defined timeout.")
reboot_machine(force=1, timeout=force_reboot_timeout)
return "ERROR"
if uptime_days() > boot_limit:
run_session_setup(control.package)
return "WARNING"
if uptime_days() < boot_limit:
return "OK"
def run_session_setup(package_name):
for session_id in get_active_sessions():
start_interactive_process("wapt-get", "--hide session-setup %s -f" % package_name, session_id=session_id) # , minimize=True
def message_box(title, text, style=0):
if platform.python_version_tuple()[0] == "3":
import importlib
importlib.reload(sys)
else:
reload(sys)
sys.setdefaultencoding("utf-8")
MB_OK = 0
MB_OKCANCEL = 1
MB_YESNOCANCEL = 3
MB_YESNO = 4
ctypes.windll.user32.MessageBoxW(0, "%s" % text, "%s" % title, style)
return
def message_box_ask_reboot(title, text):
if platform.python_version_tuple()[0] == "3":
import importlib
importlib.reload(sys)
else:
reload(sys)
sys.setdefaultencoding("utf-8")
ID_OK = 0
ID_CANCEL = 2
ID_ABORT = 3
ID_YES = 6
ID_NO = 7
response = ctypes.windll.user32.MessageBoxW(0, "%s" % text, "%s" % title, 4)
if response == ID_YES:
reboot_machine()
else:
pass
def uptime_seconds():
return time.time() - psutil.boot_time()
def uptime_minutes():
return (time.time() - psutil.boot_time()) / 60
def uptime_hours():
return (time.time() - psutil.boot_time()) / 3600
def uptime_days():
return ((time.time() - psutil.boot_time()) / 3600) / 24
def convert_time_to_hours(seconds):
return seconds / 3600
def convert_time_to_date(seconds, translate=False):
from datetime import datetime, timedelta
sec = timedelta(seconds=seconds)
d = datetime(1, 1, 1) + sec
# print("DAYS:HOURS:MIN:SEC")
# up 23 weeks, 3 days, 23 hours, 16 minutes
# up 23 weeks 3 days 23 hours 16 minutes
if translate:
if get_language() == "fr":
return "%d jours %d heures %d minutes %d secondes" % (d.day - 1, d.hour, d.minute, d.second)
return "%d days %d hours %d minutes %d seconds" % (d.day - 1, d.hour, d.minute, d.second)