Ticket #1630: gsm-provider-db.patch

extensions/cpsection/modemconfiguration/model.py

@@ -15 +15 @@
 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  US
 
 from jarabe.model.network import StorageGsm
+import gtk
+import os
+from xml.etree.cElementTree import ElementTree
+from gettext import gettext as _
+
+PROVIDERS_PATH = '/usr/share/mobile-broadband-provider-info/serviceproviders.xml'
+PROVIDERS_FORMAT_SUPPORTED = '2.0'
+COUNTRY_CODES_PATH = '/usr/share/zoneinfo/iso3166.tab'
 
 storage = StorageGsm()
 
@@ -41 +49 @@
 
 def set_apn(apn):
     storage.set_apn(apn)
+
+def has_providers_db():
+    if not os.path.isfile(COUNTRY_CODES_PATH):
+        return False
+    try:
+        et = ElementTree(file=PROVIDERS_PATH)
+        elem = et.getroot()
+        if elem is None or elem.get('format') != PROVIDERS_FORMAT_SUPPORTED:
+            return False
+        return True
+    except IOError:
+        return False
+
+class CountryListStore(gtk.ListStore):
+    COUNTRY_CODE = os.environ['LANG'][3:5].lower()
+
+    def __init__(self):
+        gtk.ListStore.__init__(self, str, object)
+        codes = {}
+        with open(COUNTRY_CODES_PATH) as f:
+            for line in f:
+                if line.startswith('#'):
+                    continue
+                code, name = line.split('\t')[:2]
+                codes[code.lower()] = name.strip()
+        etree = ElementTree(file=PROVIDERS_PATH).getroot()
+        self._country_idx = None
+        i = 0
+        for elem in etree.findall('.//country'):
+            code = elem.attrib['code']
+            if code == self.COUNTRY_CODE:
+                self._country_idx = i
+            else:
+                i += 1
+            self.append((codes[code], elem))
+
+    def get_row_providers(self, n):
+        return self[n][1]
+
+    def guess_country_row(self):
+        if self._country_idx is not None:
+            return self._country_idx
+        else:
+            return -1
+
+class ProviderListStore(gtk.ListStore):
+    def __init__(self, elem):
+        gtk.ListStore.__init__(self, str, object)
+        for provider_elem in elem.findall('.//provider'):
+            apns = provider_elem.findall('.//apn')
+            if not apns:
+                # Skip carriers with CDMA entries only
+                continue
+            self.append((provider_elem.find('.//name').text, apns))
+
+    def get_row_plans(self, n):
+        return self[n][1]
+
+class PlanListStore(gtk.ListStore):
+    LANG_NS_ATTR = '{http://www.w3.org/XML/1998/namespace}lang'
+    LANG = os.environ['LANG'][:2] # FIXME: Is this the best way?
+    DEFAULT_NUMBER = '*99#'
+
+    def __init__(self, elems):
+        gtk.ListStore.__init__(self, str, object)
+        for apn_elem in elems:
+            plan = {}
+            names = apn_elem.findall('.//name')
+            if names:
+                for name in names:
+                    if name.get(self.LANG_NS_ATTR) is None:
+                        # serviceproviders.xml default value
+                        plan['name'] = name.text
+                    elif name.get(self.LANG_NS_ATTR) == self.LANG:
+                        # Great! We found a name value for our locale!
+                        plan['name'] = name.text
+                        break
+            else:
+                plan['name'] = _('Default')
+            plan['apn'] = apn_elem.get('value')
+            user = apn_elem.find('.//username')
+            if user is not None:
+                plan['username'] = user.text
+            else:
+                plan['username'] = ''
+            passwd = apn_elem.find('.//password')
+            if passwd is not None:
+                plan['password'] = passwd.text
+            else:
+                plan['password'] = ''
+
+            plan['number'] = self.DEFAULT_NUMBER
+
+            self.append((plan['name'], plan))
+
+    def get_row_plan(self, n):
+        return self[n][1]

extensions/cpsection/modemconfiguration/view.py

@@ -59 +59 @@
         self._password_valid = True
         self._number_valid = True
         self._apn_valid = True
+        self._root = gtk.VBox()
+        self.pack_start(self._root)
+
+        if self._model.has_providers_db():
+            self._upper_box = gtk.VBox(spacing=style.DEFAULT_SPACING)
+            self._root.pack_start(self._upper_box, fill=False)
+            self._root.pack_start(gtk.HSeparator(), False)
+
+            #Country
+            box = gtk.HBox(True)
+            box.pack_start(gtk.Label(_('Country')), False)
+            store = self._model.CountryListStore()
+            country_combo = gtk.ComboBox(store)
+            cell = gtk.CellRendererText()
+            cell.props.xalign = 0.5
+            country_combo.pack_start(cell)
+            country_combo.add_attribute(cell, 'text', 0)
+            country_combo.connect('changed', self._country_selected)
+            box.pack_start(country_combo, False)
+            self._upper_box.pack_start(box, False)
+
+            #Provider
+            box = gtk.HBox(True)
+            box.pack_start(gtk.Label(_('Provider')), False)
+            self._prov_combo = gtk.ComboBox()
+            cell = gtk.CellRendererText()
+            cell.props.xalign = 0.5
+            self._prov_combo.pack_start(cell)
+            self._prov_combo.add_attribute(cell, 'text', 0)
+            self._prov_combo.connect('changed', self._provider_selected)
+            box.pack_start(self._prov_combo, False)
+            self._upper_box.pack_start(box, False)
+
+            #Plan
+            box = gtk.HBox(True)
+            box.pack_start(gtk.Label(_('Plan')), False)
+            self._plan_combo = gtk.ComboBox()
+            cell = gtk.CellRendererText()
+            cell.props.xalign = 0.5
+            self._plan_combo.pack_start(cell)
+            self._plan_combo.add_attribute(cell, 'text', 0)
+            self._plan_combo.connect('changed', self._plan_selected)
+            box.pack_start(self._plan_combo, False)
+            self._upper_box.pack_start(box, False)
+
+            country_combo.set_active(store.guess_country_row())
+
+        self._lower_box = gtk.VBox(spacing=style.DEFAULT_SPACING)
+        self._root.pack_start(self._lower_box, fill=False)
 
         self.set_border_width(style.DEFAULT_SPACING)
         self.set_spacing(style.DEFAULT_SPACING)
@@ -69 +118 @@
         self._number_entry = self._setup_field('number')
         self._apn_entry = self._setup_field('apn')
 
+        self._root.show_all()
         self.setup()
 
+    def _country_selected(self, combo):
+        model = combo.get_model()
+        providers = model.get_row_providers(combo.get_active())
+        self._prov_combo.set_model(self._model.ProviderListStore(providers))
+
+    def _provider_selected(self, combo):
+        model = combo.get_model()
+        plans = model.get_row_plans(combo.get_active())
+        self._plan_combo.set_model(self._model.PlanListStore(plans))
+
+    def _plan_selected(self, combo):
+        model = combo.get_model()
+        plan = model.get_row_plan(combo.get_active())
+        self._username_entry.set_text(plan['username'])
+        self._password_entry.set_text(plan['password'])
+        self._number_entry.set_text(plan['number'])
+        self._apn_entry.set_text(plan['apn'])
+
     def _setup_field(self, attr_name):
         box = gtk.HBox(spacing=style.DEFAULT_SPACING)
       
@@ -80 +148 @@
          
         entry = MyEntry(self._model, attr_name)
         box.pack_start(entry, False)
-        box.show_all()
 
-        self.pack_start(box, False)
+        self._lower_box.pack_start(box, False)
         return entry
 
     def setup(self):