86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
|
from os import remove, getlogin, getcwd
|
||
|
from subprocess import Popen, PIPE
|
||
|
from os.path import join
|
||
|
|
||
|
|
||
|
def add_to_scheduler(foldername: str, appname: str, programpath: str, arguments: str = ""):
|
||
|
"""Add the task to the task scheduler\n
|
||
|
The task is triggered at the user who executed this function login\n
|
||
|
Parameters
|
||
|
----------
|
||
|
* folder_name\n
|
||
|
Define the folder of the task scheduler in which we're going to create our task
|
||
|
* app_name\n
|
||
|
Define the name of the task in the task scheduler
|
||
|
* program_path\n
|
||
|
Define the program to execute when task is triggered\n
|
||
|
* arguments\n
|
||
|
The arguments to pass to the program executed when task is triggered
|
||
|
Return
|
||
|
------
|
||
|
True if command created the task successfully, else, return False"""
|
||
|
|
||
|
|
||
|
user = getlogin()
|
||
|
path = join(getcwd(), "taskschedulerxml.xml")
|
||
|
|
||
|
XML = f"""<?xml version="1.0" encoding="UTF-16"?>
|
||
|
<Task version="1.2" xmlns="http://schemas.microsoft.com/windows/2004/02/mit/task">
|
||
|
<RegistrationInfo>
|
||
|
<Date>2023-11-27T00:00:00</Date>
|
||
|
<Author>{user}</Author>
|
||
|
</RegistrationInfo>
|
||
|
<Triggers>
|
||
|
<LogonTrigger>
|
||
|
<Enabled>true</Enabled>
|
||
|
<UserId>{user}</UserId>
|
||
|
</LogonTrigger>
|
||
|
</Triggers>
|
||
|
<Principals>
|
||
|
<Principal id="Author">
|
||
|
<UserId>{user}</UserId>
|
||
|
<LogonType>InteractiveToken</LogonType>
|
||
|
<RunLevel>LeastPrivilege</RunLevel>
|
||
|
</Principal>
|
||
|
</Principals>
|
||
|
<Settings>
|
||
|
<MultipleInstancesPolicy>IgnoreNew</MultipleInstancesPolicy>
|
||
|
<DisallowStartIfOnBatteries>false</DisallowStartIfOnBatteries>
|
||
|
<StopIfGoingOnBatteries>false</StopIfGoingOnBatteries>
|
||
|
<AllowHardTerminate>true</AllowHardTerminate>
|
||
|
<StartWhenAvailable>false</StartWhenAvailable>
|
||
|
<RunOnlyIfNetworkAvailable>false</RunOnlyIfNetworkAvailable>
|
||
|
<IdleSettings>
|
||
|
<StopOnIdleEnd>true</StopOnIdleEnd>
|
||
|
<RestartOnIdle>false</RestartOnIdle>
|
||
|
</IdleSettings>
|
||
|
<AllowStartOnDemand>true</AllowStartOnDemand>
|
||
|
<Enabled>true</Enabled>
|
||
|
<Hidden>false</Hidden>
|
||
|
<RunOnlyIfIdle>false</RunOnlyIfIdle>
|
||
|
<WakeToRun>false</WakeToRun>
|
||
|
<ExecutionTimeLimit>PT0S</ExecutionTimeLimit>
|
||
|
<Priority>7</Priority>
|
||
|
</Settings>
|
||
|
<Actions Context="Author">
|
||
|
<Exec>
|
||
|
<Command>{programpath}</Command>
|
||
|
<Arguments>{arguments}</Arguments>
|
||
|
</Exec>
|
||
|
</Actions>
|
||
|
</Task>"""
|
||
|
|
||
|
#Write the file so we can execute it later
|
||
|
with open(path, "w+") as f:
|
||
|
f.write(XML)
|
||
|
|
||
|
#Execute the command
|
||
|
process = Popen(["schtasks", "/create", "/XML", f'"{path}"', "/TN", f'{foldername}\\{appname}', "/F"], stderr=PIPE)
|
||
|
|
||
|
#Remove the file
|
||
|
remove(path)
|
||
|
|
||
|
#If no error output, return true, else, return false
|
||
|
if process.stderr.read() == b'':
|
||
|
return True
|
||
|
return False
|