home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / share / pyshared / softwareproperties / gtk / DialogMirror.py < prev    next >
Encoding:
Python Source  |  2009-03-27  |  15.3 KB  |  375 lines
  1. # dialog_add.py.in - dialog to add a new repository
  2. #  
  3. #  Copyright (c) 2006 FSF Europe
  4. #              
  5. #  Authors: 
  6. #       Sebastian Heinlein <glatzor@ubuntu.com>
  7. #
  8. #  This program is free software; you can redistribute it and/or 
  9. #  modify it under the terms of the GNU General Public License as 
  10. #  published by the Free Software Foundation; either version 2 of the
  11. #  License, or (at your option) any later version.
  12. # 
  13. #  This program is distributed in the hope that it will be useful,
  14. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16. #  GNU General Public License for more details.
  17. # 
  18. #  You should have received a copy of the GNU General Public License
  19. #  along with this program; if not, write to the Free Software
  20. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  21. #  USA
  22.  
  23. import os
  24. import gobject
  25. import gtk
  26. import gtk.glade
  27. from gettext import gettext as _
  28. import threading
  29. import string
  30. import re
  31. from random import randint
  32.  
  33. import dialogs
  34. from softwareproperties.MirrorTest import MirrorTest
  35.  
  36. testing = threading.Event()
  37.  
  38. (COLUMN_PROTO, COLUMN_DIR) = range(2)
  39. (COLUMN_URI, COLUMN_SEPARATOR, COLUMN_CUSTOM, COLUMN_MIRROR) = range(4)
  40.  
  41. from softwareproperties.CountryInformation import CountryInformation
  42.  
  43. def threaded(f):
  44.     ''' Thanks to Ross Burton for this piece of code '''
  45.     def wrapper(*args, **kwargs):
  46.         t = threading.Thread(target=f, args=args, kwargs=kwargs)
  47.         t.setDaemon(True)
  48.         t.start()
  49.     wrapper.__name__ = f.__name__
  50.     return wrapper
  51.  
  52. def sort_mirrors(model, iter1, iter2, data=None):
  53.       ''' sort function for the mirror list:
  54.            - at first show all custom urls
  55.            - secondly the separator
  56.            - third the official mirrors. if available
  57.              sort the countries '''
  58.       #FIXME: cmp seems to prefer ASCI chars
  59.       (url1, sep1, custom1) = model.get(iter1, 0,1,2)
  60.       (url2, sep2, custom2) = model.get(iter2, 0,1,2)
  61.       if custom1 and custom2:
  62.           return cmp(url1,url2)
  63.       elif custom1:
  64.           return -1
  65.       elif custom2:
  66.           return 1
  67.       if sep1:
  68.           return -1
  69.       elif sep2:
  70.           return 1
  71.       return cmp(url1,url2)
  72.  
  73. class DialogMirror:
  74.   def __init__(self, parent, datadir, distro, custom_mirrors):
  75.     """
  76.     Initialize the dialog that allows to choose a custom or official mirror
  77.     """
  78.     def is_separator(model, iter, data=None):
  79.         return model.get_value(iter, COLUMN_SEPARATOR)
  80.  
  81.     self.custom_mirrors = custom_mirrors
  82.  
  83.     self.country_info = CountryInformation()
  84.  
  85.     self.gladexml = gtk.glade.XML("%s/glade/dialogs.glade" %\
  86.                                   datadir)
  87.     self.gladexml.signal_autoconnect(self)
  88.     self.dialog = self.gladexml.get_widget("dialog_mirror")
  89.     self.dialog.set_transient_for(parent)
  90.     self.dialog_test = self.gladexml.get_widget("dialog_mirror_test")
  91.     self.dialog_test.set_transient_for(self.dialog)
  92.     self.distro = distro
  93.     self.treeview = self.gladexml.get_widget("treeview_mirrors")
  94.     self.button_edit = self.gladexml.get_widget("button_mirror_edit")
  95.     self.button_remove = self.gladexml.get_widget("button_mirror_remove")
  96.     self.button_choose = self.gladexml.get_widget("button_mirror_choose")
  97.     self.button_cancel = self.gladexml.get_widget("button_test_cancel")
  98.     self.label_test = self.gladexml.get_widget("label_test_mirror")
  99.     self.progressbar_test = self.gladexml.get_widget("progressbar_test_mirror")
  100.     self.combobox = self.gladexml.get_widget("combobox_mirror_proto")
  101.     self.progress = self.gladexml.get_widget("progressbar_test_mirror")
  102.     self.label_action = self.gladexml.get_widget("label_test_mirror")
  103.  
  104.     # store each proto and its dir
  105.     model_proto = gtk.ListStore(gobject.TYPE_STRING,
  106.                                 gobject.TYPE_STRING)
  107.     self.combobox.set_model(model_proto)
  108.  
  109.     self.model = gtk.TreeStore(gobject.TYPE_STRING,  # COLUMN_URI
  110.                                gobject.TYPE_BOOLEAN, # COLUMN_SEPARATOR
  111.                                gobject.TYPE_BOOLEAN, # COLUMN_CUSTOM
  112.                                gobject.TYPE_PYOBJECT)# COLUMN_MIRROR
  113.     self.treeview.set_row_separator_func(is_separator)
  114.     self.model_sort = gtk.TreeModelSort(self.model)
  115.     self.model_sort.set_default_sort_func(sort_mirrors)
  116.  
  117.     self.distro = distro
  118.  
  119.     self.treeview.set_model(self.model_sort)
  120.     # the cell renderer for the mirror uri
  121.     self.renderer_mirror = gtk.CellRendererText()
  122.     self.renderer_mirror.connect('edited', 
  123.                                  self.on_edited_custom_mirror, 
  124.                                  self.model)
  125.     # the visible column that holds the mirror uris
  126.     self.column_mirror = gtk.TreeViewColumn("URI", 
  127.                                             self.renderer_mirror, 
  128.                                             text=COLUMN_URI)
  129.     self.treeview.append_column(self.column_mirror)
  130.  
  131.     # used to find the corresponding iter of a location
  132.     map_loc = {}
  133.     patriot = None
  134.     model = self.treeview.get_model().get_model()
  135.     # at first add all custom mirrors and a separator
  136.     if len(self.custom_mirrors) > 0:
  137.         for mirror in self.custom_mirrors:
  138.             model.append(None, [mirror, False, True, None])
  139.             self.column_mirror.add_attribute(self.renderer_mirror, 
  140.                                              "editable", 
  141.                                              COLUMN_CUSTOM)
  142.         model.append(None, [None, True, False, None])
  143.     # secondly add all official mirrors
  144.     for hostname in self.distro.source_template.mirror_set.keys():
  145.         mirror = self.distro.source_template.mirror_set[hostname]
  146.         if map_loc.has_key(mirror.location):
  147.             model.append(map_loc[mirror.location],
  148.                          [hostname, False, False, mirror])
  149.         elif mirror.location != None:
  150.             parent = model.append(None, 
  151.                                   [self.country_info.get_country_name(mirror.location), False, False, None])
  152.             if mirror.location == self.country_info.code and patriot == None:
  153.                 patriot = parent
  154.             model.append(parent, [hostname, False, False, mirror]),
  155.             map_loc[mirror.location] = parent
  156.         else:
  157.             model.append(None, [hostname, False, False, mirror])
  158.     # Scroll to the local mirror set
  159.     if patriot != None:
  160.         path_sort = self.model_sort.get_path(self.model_sort.convert_child_iter_to_iter(None, patriot))
  161.         self.treeview.expand_row(path_sort, False)
  162.         self.treeview.set_cursor(path_sort)
  163.         self.treeview.scroll_to_cell(path_sort, use_align=True, row_align=0.5)
  164.  
  165.   def on_edited_custom_mirror(self, cell, path, new_text, model):
  166.     ''' Check if the new mirror uri is faild, if yes change it, if not
  167.         remove the mirror from the list '''
  168.     iter = model.get_iter(path)
  169.     iter_next = model.iter_next(iter)
  170.     if new_text != "":
  171.         model.set_value(iter, COLUMN_URI, new_text)
  172.         # Add a separator if the next mirror is a not a separator or 
  173.         # a custom one
  174.         if iter_next != None and not \
  175.            (model.get_value(iter_next, COLUMN_SEPARATOR) or \
  176.             model.get_value(iter_next, COLUMN_CUSTOM)):
  177.             model.insert(1, [None, True, False])
  178.         self.button_choose.set_sensitive(self.is_valid_mirror(new_text))
  179.     else:
  180.         model.remove(iter)
  181.         # Remove the separator if this was the last custom mirror
  182.         if model.get_value(model.get_iter_first(), COLUMN_SEPARATOR):
  183.             model.remove(model.get_iter_first())
  184.         self.treeview.set_cursor((0,))
  185.     return
  186.  
  187.   def is_valid_mirror(self, uri):
  188.     ''' Check if a given uri is a vaild one '''
  189.     if uri == None:
  190.         return False
  191.     elif re.match("^((ftp)|(http)|(file)|(rsync)|(https))://([a-z]|[A-Z]|[0-9]|:|/|\.|~)+$", uri) == None:
  192.         return False
  193.     else:
  194.         return True
  195.  
  196.   def on_treeview_mirrors_cursor_changed(self, treeview, data=None):
  197.     ''' Check if the currently selected row in the mirror list
  198.         contains a mirror and or is editable '''
  199.     (row, column) = treeview.get_cursor()
  200.     if row == None:
  201.         self.button_remove.set_sensitive(False)
  202.         self.button_edit.set_sensitive(False)
  203.         self.button_choose.set_sensitive(False)
  204.         return
  205.     model = treeview.get_model()
  206.     iter = model.get_iter(row)
  207.     # Update the list of available protocolls
  208.     mirror = model.get_value(iter, COLUMN_MIRROR)
  209.     model_protos = self.combobox.get_model()
  210.     model_protos.clear()
  211.     if mirror != None:
  212.         self.combobox.set_sensitive(True)
  213.         seen_protos = []
  214.         for repo in mirror.repositories:
  215.             # Only add a repository for a protocoll once
  216.             if repo.proto in seen_protos:
  217.                 continue
  218.             seen_protos.append(repo.proto)
  219.             model_protos.append(repo.get_info())
  220.         self.combobox.set_active(0)
  221.         self.button_choose.set_sensitive(True)
  222.     else:
  223.         # Allow to edit and remove custom mirrors
  224.         self.button_remove.set_sensitive(model.get_value(iter, COLUMN_CUSTOM))
  225.         self.button_edit.set_sensitive(model.get_value(iter, COLUMN_CUSTOM))
  226.         self.button_choose.set_sensitive(self.is_valid_mirror(model.get_value(iter, COLUMN_URI)))
  227.         self.combobox.set_sensitive(False)
  228.  
  229.   def on_button_mirror_remove_clicked(self, button, data=None):
  230.     ''' Remove the currently selected mirror '''
  231.     path, column = self.treeview.get_cursor()
  232.     iter = self.treeview.get_model().get_iter(path)
  233.     model = self.treeview.get_model().get_model()
  234.     model.remove(iter)
  235.     # Remove the separator if this was the last custom mirror
  236.     if model.get_value(model.get_iter_first(), COLUMN_SEPARATOR):
  237.         model.remove(model.get_iter_first())
  238.     self.treeview.set_cursor((0,))
  239.  
  240.   def on_button_mirror_add_clicked(self, button, data=None):
  241.     ''' Add a new mirror at the beginning of the list and start
  242.         editing '''
  243.     model = self.treeview.get_model().get_model()
  244.     model.append(None, [_("New mirror"), False, True, None])
  245.     self.treeview.grab_focus()
  246.     self.treeview.set_cursor((0,),
  247.                              focus_column=self.column_mirror, 
  248.                              start_editing=True)
  249.  
  250.   def on_button_mirror_edit_clicked(self, button, data=None):
  251.     ''' Grab the focus and start editing the currently selected mirror '''
  252.     path, column = self.treeview.get_cursor()
  253.     self.treeview.grab_focus()
  254.     self.treeview.set_cursor(path, focus_column=column, start_editing=True)
  255.  
  256.   def on_dialog_mirror_test_delete_event(self, dialog, event, data=None):
  257.     ''' If anybody wants to close the dialog, stop the test before '''
  258.     self.on_button_cancel_test_clicked(None)
  259.     return True
  260.  
  261.   def run(self):
  262.     ''' Run the chooser dialog and return the chosen mirror or None '''
  263.     res = self.dialog.run()
  264.     self.dialog.hide()
  265.  
  266.     (row, column) = self.treeview.get_cursor()
  267.     model = self.treeview.get_model()
  268.     iter = model.get_iter(row)
  269.     mirror = model.get_value(iter, COLUMN_MIRROR)
  270.  
  271.     # FIXME: we should also return the list of custom servers
  272.     if res == gtk.RESPONSE_OK:
  273.         if mirror == None:
  274.             # Return the URL of the selected custom mirror
  275.             return model.get_value(iter, COLUMN_URI)
  276.         else:
  277.             # Return an URL created from the hostname and the selected
  278.             # repository
  279.             model_proto = self.combobox.get_model()
  280.             iter_proto = model_proto.get_iter(self.combobox.get_active())
  281.             proto = model_proto.get_value(iter_proto, COLUMN_PROTO)
  282.             dir = model_proto.get_value(iter_proto, COLUMN_DIR)
  283.             return "%s://%s/%s" % (proto, mirror.hostname, dir)
  284.     else:
  285.         return None
  286.  
  287.   @threaded
  288.   def on_button_test_clicked(self, button, data=None):
  289.     ''' Perform a test to find the best mirror and select it 
  290.         afterwards in the treeview '''
  291.     class MirrorTestGtk(MirrorTest):
  292.         def __init__(self, mirrors, test_file, running, progressbar, label):
  293.             MirrorTest.__init__(self, mirrors, test_file, running)
  294.             self.progress = progressbar
  295.             self.label = label
  296.         def report_action(self, text):
  297.             gtk.gdk.threads_enter()
  298.             self.label.set_label(str("<i>%s</i>" % text))
  299.             gtk.gdk.threads_leave()
  300.         def report_progress(self, current, max, borders=(0,1), mod=(0,0)):
  301.             gtk.gdk.threads_enter()
  302.             self.progress.set_text(_("Completed %s of %s tests") % \
  303.                                    (current + mod[0], max + mod[1]))
  304.             frac = borders[0] + (borders[1] - borders[0]) / max * current
  305.             self.progress.set_fraction(frac)
  306.             gtk.gdk.threads_leave()
  307.         def run_full_test(self):
  308.             # Determinate the 5 top ping servers
  309.             results_ping = self.run_ping_test(max=5,
  310.                                               borders=(0, 0.5),
  311.                                               mod=(0,7))
  312.             # Add two random mirrors to the download test
  313.             size = len(self.mirrors)
  314.         if size > 2:
  315.                 results_ping.append([0, 0, self.mirrors[randint(1,size-1)]])
  316.                 results_ping.append([0, 0, self.mirrors[randint(1,size-1)]])
  317.             results = self.run_download_test(map(lambda r: r[2], results_ping),
  318.                                              borders=(0.5, 1),
  319.                                              mod=(MirrorTest.todo,
  320.                                                   MirrorTest.todo))
  321.             for (t, h) in results:
  322.                 print "mirror: %s - time: %s" % (h.hostname, t)
  323.             if len(results) == 0:
  324.                 return None
  325.             else:
  326.                 return results[0][1].hostname
  327.  
  328.     gtk.gdk.threads_enter()
  329.     self.button_cancel.set_sensitive(True)
  330.     self.dialog_test.show()
  331.     gtk.gdk.threads_leave()
  332.     self.running = threading.Event()
  333.     self.running.set()
  334.     pipe = os.popen("dpkg --print-architecture")
  335.     arch = pipe.read().strip()
  336.     test_file = "dists/%s/%s/binary-%s/Packages.gz" % \
  337.                  (self.distro.source_template.name,
  338.                   self.distro.source_template.components[0].name,
  339.                   arch)
  340.     test = MirrorTestGtk(self.distro.source_template.mirror_set.values(), 
  341.                          test_file,
  342.                          self.running,
  343.                          self.progress,
  344.                          self.label_action)
  345.     test.start()
  346.     rocker = test.run_full_test()
  347.     gtk.gdk.threads_enter()
  348.     testing.clear()
  349.     self.dialog_test.hide()
  350.     # Select the mirror in the list or show an error dialog
  351.     if rocker != None:
  352.         self.model_sort.foreach(self.select_mirror, rocker)
  353.     else:
  354.         dialogs.show_error_dialog(self.dialog, 
  355.                                   _("No suitable download server was found"),
  356.                                   _("Please check your Internet connection."))
  357.     gtk.gdk.threads_leave()
  358.  
  359.   def select_mirror(self, model, path, iter, mirror):
  360.     """Select and expand the path to a matching mirror in the list"""
  361.     if model.get_value(iter, COLUMN_URI) == mirror:
  362.         self.treeview.expand_to_path(path)
  363.         self.treeview.set_cursor(path)
  364.         self.treeview.scroll_to_cell(path, use_align=True, row_align=0.5)
  365.         self.treeview.grab_focus()
  366.         # breaks foreach
  367.         return True
  368.  
  369.   def on_button_cancel_test_clicked(self, button):
  370.     ''' Abort the mirror performance test '''
  371.     self.running.clear()
  372.     self.label_test.set_label("<i>%s</i>" % _("Canceling..."))
  373.     self.button_cancel.set_sensitive(False)
  374.     self.progressbar_test.set_fraction(1)
  375.