github/plexpy
1
0
Fork 0
mirror of https://github.com/Tautulli/Tautulli.git synced 2025-08-19 04:49:36 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Use updater to modify Task Scheduler XML

Browse source
This commit is contained in:
JonnyWong16 2020-12-24 15:26:35 -08:00
commit 00629c0983
No known key found for this signature in database
GPG key ID: B1F1F9807184697A
1 changed files with 33 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

41
package/updater-windows.py
View file

@ -16,14 +16,17 @@
# along with Tautulli. If not, see <http://www.gnu.org/licenses/>. # along with Tautulli. If not, see <http://www.gnu.org/licenses/>.
from logging import handlers from logging import handlers
import argparse
import logging import logging
import os import os
import psutil import psutil
import re
import requests import requests
import shutil import shutil
import subprocess import subprocess
import sys import sys
import tempfile import tempfile
import xml.etree.ElementTree as ET
SCRIPT_PATH = os.path.dirname(os.path.abspath(__file__)) SCRIPT_PATH = os.path.dirname(os.path.abspath(__file__))
@ -140,7 +143,7 @@ def update_tautulli():
logger.info('Running %s', download_file) logger.info('Running %s', download_file)
try: try:
subprocess.call([file_path, '/S', '/NORUN'], creationflags=CREATE_NO_WINDOW) subprocess.call([file_path, '/S', '/NORUN', '/D=' + SCRIPT_PATH], creationflags=CREATE_NO_WINDOW)
status = 0 status = 0
except Exception as e: except Exception as e:
logger.exception('Failed to install Tautulli: %s', e) logger.exception('Failed to install Tautulli: %s', e)
@ -158,12 +161,34 @@ def update_tautulli():
if __name__ == '__main__': if __name__ == '__main__':
logger = init_logger() parser = argparse.ArgumentParser()
parser.add_argument('--xml', action='store_true')
opts = parser.parse_args()
try: if opts.xml:
status_code = update_tautulli() xml_path = os.path.join(SCRIPT_PATH, 'TautulliUpdateTask.xml')
except Exception as exc: tree = ET.parse(xml_path)
status_code = exc task = tree.getroot()
logger.debug('Update function returned status code %s', status_code)
sys.exit(status_code) match = re.match(r'{(.*)}', task.tag)
namespace = match.group(1)
namespaces = {'': namespace}
ET.register_namespace('', namespace)
for elem in task.iterfind('./Actions/Exec/Command', namespaces=namespaces):
elem.text = os.path.join(SCRIPT_PATH, 'updater.exe')
for elem in task.iterfind('./Actions/Exec/WorkingDirectory', namespaces=namespaces):
elem.text = SCRIPT_PATH
tree.write(xml_path, encoding='UTF-16')
else:
logger = init_logger()
try:
status_code = update_tautulli()
except Exception as exc:
status_code = exc
logger.debug('Update function returned status code %s', status_code)
sys.exit(status_code)