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)
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"
