示例#1
0
文件: app.py 项目: sproaty/whyteboard
    def OnInit(self):
        """
        Initialises the application, parses command line arguments, sets the 
        correct locale and sets up logging.
        """
        startup_time = time.time()
        wx.SetDefaultPyEncoding("utf-8")
        self.SetAppName(u"whyteboard")  # used to identify app in $HOME/

        parser = OptionParser(version="Whyteboard %s" % meta.version)
        parser.add_option("-f", "--file", help="load FILE on load")
        parser.add_option("-c", "--conf", help="load configurations from CONF file")
        parser.add_option("--width", type="int", help="set canvas to WIDTH")
        parser.add_option("--height", type="int", help="set canvas to HEIGHT")
        parser.add_option("-u", "--update", action="store_true", help="check for a newer version of whyteboard")
        parser.add_option("-l", "--lang", help="set language. can be a country code or language (e.g. fr, french; nl, dutch)")
        parser.add_option("-d", "--debug", action="store_true", help="debug mode. more information about the program is logged")

        (options, args) = parser.parse_args()
        self.setup_logging(options.debug)
        self._oldhook = sys.excepthook
        sys.excepthook = ExceptionHook

        logger.info("Program starting")
        logger.debug("Received command line options [%s] and args [%s]", options, args)
        preferences_file = options.conf

        Config().init(preferences_file)
        self.set_language(options.lang)

        if options.update:
            self.update()

        self.frame = GUI()
        self.frame.Show(True)

        try:
            _file = options.file or sys.argv[1]
            _file = os.path.abspath(to_unicode(_file))
            logger.debug("Loading file [%s]", _file)
            if os.path.exists(_file):
                self.frame.do_open(_file)
        except IndexError:
            pass

        x = options.width or self.frame.canvas.area[0]
        y = options.height or self.frame.canvas.area[1]
        self.frame.canvas.resize((x, y))

        # If everything goes well..
        wx.CallAfter(self.delete_temp_update_files)

        logger.info("Startup complete, time taken: %.3fms", (time.time() - startup_time))
        return True
示例#2
0
文件: app.py 项目: sproaty/whyteboard
class WhyteboardApp(wx.App):
    def OnInit(self):
        """
        Initialises the application, parses command line arguments, sets the 
        correct locale and sets up logging.
        """
        startup_time = time.time()
        wx.SetDefaultPyEncoding("utf-8")
        self.SetAppName(u"whyteboard")  # used to identify app in $HOME/

        parser = OptionParser(version="Whyteboard %s" % meta.version)
        parser.add_option("-f", "--file", help="load FILE on load")
        parser.add_option("-c", "--conf", help="load configurations from CONF file")
        parser.add_option("--width", type="int", help="set canvas to WIDTH")
        parser.add_option("--height", type="int", help="set canvas to HEIGHT")
        parser.add_option("-u", "--update", action="store_true", help="check for a newer version of whyteboard")
        parser.add_option("-l", "--lang", help="set language. can be a country code or language (e.g. fr, french; nl, dutch)")
        parser.add_option("-d", "--debug", action="store_true", help="debug mode. more information about the program is logged")

        (options, args) = parser.parse_args()
        self.setup_logging(options.debug)
        self._oldhook = sys.excepthook
        sys.excepthook = ExceptionHook

        logger.info("Program starting")
        logger.debug("Received command line options [%s] and args [%s]", options, args)
        preferences_file = options.conf

        Config().init(preferences_file)
        self.set_language(options.lang)

        if options.update:
            self.update()

        self.frame = GUI()
        self.frame.Show(True)

        try:
            _file = options.file or sys.argv[1]
            _file = os.path.abspath(to_unicode(_file))
            logger.debug("Loading file [%s]", _file)
            if os.path.exists(_file):
                self.frame.do_open(_file)
        except IndexError:
            pass

        x = options.width or self.frame.canvas.area[0]
        y = options.height or self.frame.canvas.area[1]
        self.frame.canvas.resize((x, y))

        # If everything goes well..
        wx.CallAfter(self.delete_temp_update_files)

        logger.info("Startup complete, time taken: %.3fms", (time.time() - startup_time))
        return True

    def __del__(self):
        sys.excepthook = self._oldhook


    def setup_logging(self, debug):
        logfile = os.path.join(get_home_dir(), u"whyteboard.log")
        fh = logging.FileHandler(logfile)
        ch = logging.StreamHandler()
        if debug:
            logger.setLevel(logging.DEBUG)
        formatter = logging.Formatter('%(levelname)s %(asctime)s %(message)s')
        fh.setFormatter(formatter)
        ch.setFormatter(formatter)
        logger.addHandler(fh)
        logger.addHandler(ch)


    def delete_temp_update_files(self):
        tmp_file = os.path.join(get_path(), u"whyteboard-tmp.exe")
        if is_exe() and os.path.exists(tmp_file):
            logger.info("Removing backup EXE [%s] after performing an update", tmp_file)
            os.remove(tmp_file)

    def set_language(self, option_lang=None):
        """
        Sets the user's language.
        """
        set_lang = False
        lang_name = Config().get('language')
        logger.debug("Found language [%s] in config", lang_name)

        if option_lang:
            logger.debug("Attempting to set language from command line: [%s]", option_lang)
            country = wx.Locale.FindLanguageInfo(option_lang)
            if country:
                set_lang = True
                lang_name = country.Description
                self.locale = wx.Locale(country.Language)
                logger.debug("Using command-line set language [%s]", lang_name)
            else:
                logger.warning("Could not parse command-line argument [%s] into a known locale/language", option_lang)

        if not set_lang:
            for x in meta.languages:
                if lang_name.capitalize() == 'Welsh':
                    self.locale = wx.Locale()
                    self.locale.Init(u"Cymraeg", u"cy", u"cy_GB.utf8")
                    break
                elif lang_name == x[0]:
                    logger.debug("Attempting to set language to [%s] from config", lang_name)
                    nolog = wx.LogNull()
                    self.locale = wx.Locale(x[2])

        if not hasattr(self, "locale"):
            logger.debug("No locale set, reverting to system language")
            self.locale = wx.Locale(wx.LANGUAGE_DEFAULT)
            Config().language(wx.Locale.GetLanguageName(wx.LANGUAGE_DEFAULT))
            Config().write()

        if not wx.Locale.IsOk(self.locale):
            logger.warning("Could not set language to [%s]", lang_name)
            wx.MessageBox(u"Error setting language to %s - reverting to English"
                          % lang_name, u"Whyteboard")
            if not set_lang:
                Config().language('English')
                Config().write()
            self.locale = wx.Locale(wx.LANGUAGE_ENGLISH)

        logger.info("Whyteboard is running in [%s]", wx.Locale.GetLanguageName(self.locale.GetLanguage()))
        langdir = os.path.join(get_path(), u'locale')
        logger.debug("Adding locale catalogue [%s]", langdir)
        self.locale.AddCatalogLookupPathPrefix(langdir)
        self.locale.AddCatalog(u"whyteboard")
        self.locale.AddCatalog(u'wxstd')

        # nasty fix for some translated strings not being applied
        meta.languages = meta.define_languages()
        meta.types, meta.dialog_wildcard = meta.define_filetypes()
    
        
    def update(self):
        """
        Prompts the user for confirmation whether to update (without launching the GUI)
        """
        self.updater = Updater()
        download = self.updater.get_latest_version_info()

        if not download:
            print _("Could not connect to server.")
            return
        if not self.updater.can_update():
            print _("You are running the latest version.")
            return

        confirm = raw_input("There is a new version available, %(version)s\nFile: %(filename)s\nSize: %(filesize)s)\n\nDo you want to update? (y/n)\n"
                            % {u'version': download.version,
                               u'filename': download.filename(),
                               u'filesize': download.filesize()})

        print ""
        if confirm.lower() == "y" or confirm.lower() == "yes":
            self.progress = ProgressBar(total=download.raw_filesize())
            
            if not self.updater.download_file(self.update_progress_reporter):
                print _("Could not connect to server.")
                wx.Exit()
                
            self.updater.extract()
            args = self.updater.restart_args()
            os.execvp(*args)
        
               
    def update_progress_reporter(self, count, block, total):
        self.progress.increment_amount(count)
        print self.progress, '\r',
        sys.stdout.flush()
        if count * block >= total:
            print "\n"