twblue/src/main.py

# -*- coding: utf-8 -*-
import platform
from win32com.client import GetObject

""" there are lots of things not implemented for Gtk+ yet.
    We've started this effort 1 Apr 2015 so it isn't fully functional. We will remove the ifs statements when are no needed"""

system = platform.system()
if system == "Windows":
    import sys
    import os
    #redirect the original stdout and stderr
    stdout=sys.stdout
    stderr=sys.stderr
    sys.stdout = open(os.path.join(os.getenv("temp"), "stdout.log"), "w")
    sys.stderr = open(os.path.join(os.getenv("temp"), "stderr.log"), "w")
import languageHandler
import paths
#check if TWBlue is installed (Windows only)
# ToDo: Remove this soon as this is done already when importing the paths module.
if os.path.exists(os.path.join(paths.app_path(), "Uninstall.exe")):
    paths.mode="installed"
import commandline
import config
import output
import logging
import application
import keys
from mysc.thread_utils import call_threaded
import fixes
import widgetUtils
import webbrowser
from wxUI import commonMessageDialogs
if system == "Windows":
    from logger import logger
    from update import updater
    stdout_temp=sys.stdout
    stderr_temp=sys.stderr
#if it's a binary version
    if hasattr(sys, 'frozen'):
        sys.stderr = open(os.path.join(paths.logs_path(), "stderr.log"), 'w')
        sys.stdout = open(os.path.join(paths.logs_path(), "stdout.log"), 'w')
    else:
        sys.stdout=stdout
        sys.stderr=stderr
        # We are running from source, let's prepare vlc module for that situation
        if system=="Windows":
            arch="x86"
            if platform.architecture()[0][:2] == "64":
                arch="x64"
            os.environ['PYTHON_VLC_MODULE_PATH']=os.path.abspath(os.path.join(paths.app_path(), "..", "windows-dependencies", arch))
            os.environ['PYTHON_VLC_LIB_PATH']=os.path.abspath(os.path.join(paths.app_path(), "..", "windows-dependencies", arch, "libvlc.dll"))
    #the final log files have been opened succesfully, let's close the temporary files
    stdout_temp.close()
    stderr_temp.close()
    #finally, remove the temporary files. TW Blue doesn't need them anymore, and we will get more free space on the harddrive
    os.remove(stdout_temp.name)
    os.remove(stderr_temp.name)
import sound
if system == "Linux":
    from gi.repository import Gdk, GObject, GLib

log = logging.getLogger("main")

def setup():
    log.debug("Starting " + application.name + " %s" % (application.version,))
    config.setup()
    proxy_setup()
    log.debug("Using %s %s" % (platform.system(), platform.architecture()[0]))
    log.debug("Application path is %s" % (paths.app_path(),))
    log.debug("config path  is %s" % (paths.config_path(),))
    sound.setup()
    languageHandler.setLanguage(config.app["app-settings"]["language"])
    fixes.setup() 
    output.setup()
    keys.setup()
    from controller import settings
    from controller import mainController
    from sessionmanager import sessionManager
    app = widgetUtils.mainLoopObject()
    check_pid()
    if system == "Windows":
        if config.app["app-settings"]["donation_dialog_displayed"] == False:
            donation()
        if config.app['app-settings']['check_for_updates']:
            updater.do_update()
    sm = sessionManager.sessionManagerController()
    sm.fill_list()
    if len(sm.sessions) == 0: sm.show()
    else:
        sm.do_ok()
    if hasattr(sm.view, "destroy"):
        sm.view.destroy()
    del sm
    r = mainController.Controller()
    r.view.show()
    r.do_work()
    r.check_invisible_at_startup()
    if system == "Windows":
        call_threaded(r.start)
    elif system == "Linux":
        GLib.idle_add(r.start)
    app.run()

def proxy_setup():
    if config.app["proxy"]["server"] != "" and config.app["proxy"]["type"] > 0:
        log.debug("Loading proxy settings")
        proxy_url = config.app["proxy"]["server"] + ":" + str(config.app["proxy"]["port"])
        if config.app["proxy"]["user"] != "" and config.app["proxy"]["password"] != "":
            proxy_url = config.app["proxy"]["user"] + ":" + config.app["proxy"]["password"] + "@" + proxy_url
        elif config.app["proxy"]["user"] != "" and config.proxyTypes[config.app["proxy"]["type"]] in ["socks4", "socks4a"]:
            proxy_url = config.app["proxy"]["user"] + "@" + proxy_url
        proxy_url = config.proxyTypes[config.app["proxy"]["type"]] + "://" + proxy_url
        os.environ["HTTP_PROXY"] = proxy_url
        os.environ["HTTPS_PROXY"] = proxy_url

def donation():
    dlg = commonMessageDialogs.donation()
    if dlg == widgetUtils.YES:
        webbrowser.open_new_tab(_("https://twblue.es/donate"))
    config.app["app-settings"]["donation_dialog_displayed"] = True

def is_running(pid):
    "Check if the process with ID pid is running. Adapted from https://stackoverflow.com/a/568589"
    WMI = GetObject('winmgmts:')
    processes = WMI.InstancesOf('Win32_Process')
    return [process.Properties_('ProcessID').Value for process in processes if process.Properties_('ProcessID').Value == pid]

def check_pid():
    "Insures that only one copy of the application is running at a time."
    pidpath = os.path.join(os.getenv("temp"), "{}.pid".format(application.name))
    if os.path.exists(pidpath):
        with open(pidpath) as fin:
            pid = int(fin.read())
        if is_running(pid):
            # Display warning dialog
            commonMessageDialogs.common_error(_(u"{0} is already running. Close the other instance before starting this one. If you're sure that {0} isn't running, try deleting the file at {1}. If you're unsure of how to do this, contact the {0} developers.").format(application.name, pidpath))
            sys.exit(1)
        else:
            commonMessageDialogs.dead_pid()
    # Write the new PID
    with open(pidpath,"w") as cam:
        cam.write(str(os.getpid()))

setup()