diff options
-rw-r--r-- | src/ppm/providerdb.py | 7 | ||||
-rwxr-xr-x | src/prepaid-manager-applet.py | 7 |
2 files changed, 13 insertions, 1 deletions
diff --git a/src/ppm/providerdb.py b/src/ppm/providerdb.py index e2b097d..c1aebec 100644 --- a/src/ppm/providerdb.py +++ b/src/ppm/providerdb.py @@ -133,6 +133,13 @@ class ProviderDB(object): for r in searcher(self.tree): yield r.attrib['code'] + def get_country_by_code(self, code): + """Given a country code return it's name""" + try: + return self.countries[code] + except KeyError: + return None + def get_countries(self): for code in self.get_country_codes(): try: diff --git a/src/prepaid-manager-applet.py b/src/prepaid-manager-applet.py index b1ac318..b1f4f6f 100755 --- a/src/prepaid-manager-applet.py +++ b/src/prepaid-manager-applet.py @@ -227,6 +227,9 @@ class PPMController(GObject.GObject): def get_provider_providers(self, country_code): return self.providerdb.get_providers_by_code(country_code) + def get_country_by_code(self, code): + return self.providerdb.get_country_by_code(code) + def on_mm_request_started(self, obj, mm_proxy): logging.debug("Started modem request: %s", mm_proxy.request) self.view.show_modem_response() @@ -672,7 +675,9 @@ class PPMProviderAssistant(PPMObject): self.providers_initialized = self.country_code self._fill_provider_liststore_by_country_code(self.country_code) elif self.assistant.get_current_page() == self.PAGE_CONFIRM: - self.label_country.set_text(self.country_code) + country = self.controller.get_country_by_code(self.country_code) + label = country if country else self.country_code + self.label_country.set_text(label) self.label_provider.set_text(self.provider) def on_treeview_countries_changed(self, obj): |