twblue/src/mysc/autostart.py

from __future__ import unicode_literals
from future import standard_library
standard_library.install_aliases()
from builtins import str
import winreg
import os
import sys
from platform_utils import paths

RUN_REGKEY = r"SOFTWARE\Microsoft\Windows\CurrentVersion\Run"

def is_installed(app_subkey):
    """Checks if the currently running copy is installed or portable variant. Requires the name of the application subkey found under the uninstall section in Windows registry."""

    try:
        key = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\%s" % app_subkey)
        inst_dir = winreg.QueryValueEx(key,"InstallLocation")[0]
    except WindowsError:
        return False
    winreg.CloseKey(key)
    try:
        return os.stat(inst_dir) == os.stat(paths.app_path())
    except WindowsError:
        return False

def getAutoStart(app_name):
    """Queries if the automatic startup should be set for the application or not, depending on it's current state."""

    try:
        key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, RUN_REGKEY)
        val = winreg.QueryValueEx(key, str(app_name))[0]
        return os.stat(val) == os.stat(sys.argv[0])
    except (WindowsError, OSError):
        return False

def setAutoStart(app_name, enable=True):
    """Configures automatic startup for the application, if the enable argument is set to True. If set to False, deletes the application AutoStart value."""
    if getAutoStart(app_name) == enable:
        return
    key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, RUN_REGKEY, 0, winreg.KEY_WRITE)
    if enable:
        winreg.SetValueEx(key, str(app_name), None, winreg.REG_SZ, paths.get_executable())
    else:
        winreg.DeleteValue(key, str(app_name))