Notification after shut down on windows using python

Question

I want to create a function that displays a message for the user when windows start again after a shutdown.

It is used together with the shutdown command shutdown \s \hybrid

Heres my attempt with the help of AI

import os


def leave_message_on_startup(message):
    script_content = f"""
import pymsgbox

message = "{message}"
pymsgbox.alert(message, "Startup Message")
"""

    dirname = os.path.dirname
    script_path = f"{dirname(dirname(dirname(__file__)))}\\temp\\DisplayMessageOnStartup.pyw"
    os.makedirs("temp", exist_ok=True)
    with open(script_path, "w+") as script_file:
        script_file.write(script_content)

    create_task_command = f'schtasks /create /tn "{task_name}" /tr "py {script_path}" /sc onstart /f'
    os.system(create_task_command)

def trigger_task(task_name):
    trigger_task_command = f'schtasks /run /tn "{task_name}"'
    os.system(trigger_task_command)


task_name = "DisplayMessageOnStartup"
message = "Sample message"
leave_message_on_startup(message)
trigger_task(task_name)

After running the code, I double-clicked on the created script in File Explorer and it worked.

Even though the script creation was successful, the task creation seems to have failed.

I can't find the problem.

Please enlighten me on ways to achieve notifications after windows shutdown.

Are you sure the pyw file is a Windows (PE) executable directly executable on Windows? If it just a script you need to execute the interpreter executable and provide the script as parameters. Windows Explorer may do this automatically, but everywhere else you need to manually supply the real executable. — Robert
– Robert, Commented Jan 27 at 19:50

Fritz Robinson · Accepted Answer · 2025-01-27 20:12:50Z

1

Schtasks is assuming that Python is in %systemroot%\System32 (C:\Windows\System32).

Replace py with the full path to Python.

edited Jan 27 at 20:12

answered Jan 27 at 20:04

Fritz Robinson

225 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Electron X · Accepted Answer · 2025-02-01 14:43:58Z

Final working version:

import re
import os
import ctypes

def get_startup_folder():
    CSIDL_STARTUP = 0x0007
    SHGFP_TYPE_CURRENT = 0  # Get current, not default value

    buf = ctypes.create_unicode_buffer(260)  # Maximum path length
    ctypes.windll.shell32.SHGetFolderPathW(None, CSIDL_STARTUP, None, SHGFP_TYPE_CURRENT, buf)

    return buf.value

def clean_message(text):
    """Escape special characters for VBScript message content"""
    text = text.replace('\\', '\\\\')
    text = text.replace('"', '""')
    text = text.replace("\n", "\" & vbCrLf & \"")
    return text


def leave_notif_on_login(message, title="Python"):
    """
    Creates a VBScript in the Startup folder to display
    a notification at login. Files will self-delete after execution.
    """
    message = clean_message(message)
    title = clean_message(title)

    # Get Windows Startup folder path
    startup_folder = get_startup_folder()
    task_name = re.sub(r'[^a-zA-Z0-9-]', '_', title)

    # Create path for the VBScript file
    vbs_path = os.path.join(startup_folder,  f"{task_name}.vbs")

    # VBScript content for message display
    vbs_content = f'''
Set WshShell = CreateObject("WScript.Shell")
Set FSO = CreateObject("Scripting.FileSystemObject")

' Create the PowerShell script file
psScriptPath = FSO.GetSpecialFolder(2) & "\\notification.ps1"
psScriptContent = "[Windows.UI.Notifications.ToastNotificationManager, Windows.UI.Notifications, ContentType = WindowsRuntime]" & vbCrLf & _
    "$Template = [Windows.UI.Notifications.ToastNotificationManager]::GetTemplateContent([Windows.UI.Notifications.ToastTemplateType]::ToastImageAndText02)" & vbCrLf & _
    "$ToastXml = $Template" & vbCrLf & _
    "$ToastXml.GetElementsByTagName('text')[0].AppendChild($ToastXml.CreateTextNode(""{message}""))" & vbCrLf & _
    "$Toast = [Windows.UI.Notifications.ToastNotification]::new($ToastXml)" & vbCrLf & _
    "$Notifier = [Windows.UI.Notifications.ToastNotificationManager]::CreateToastNotifier(""{title}"")" & vbCrLf & _
    "$Notifier.Show($Toast)"

Set psScriptFile = FSO.CreateTextFile(psScriptPath, True)
psScriptFile.WriteLine psScriptContent
psScriptFile.Close

' Run the PowerShell script
WshShell.Run "powershell.exe -ExecutionPolicy Bypass -WindowStyle Hidden -File """ & psScriptPath & """", 0, True

' Delete the PowerShell script file
FSO.DeleteFile psScriptPath, True
FSO.DeleteFile WScript.ScriptFullName, True
'''

    with open(vbs_path, "w", encoding="utf-8") as vbs_file:
        vbs_file.write(vbs_content)

    return vbs_path

Collectives™ on Stack Overflow

Notification after shut down on windows using python

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related