| # Copyright (c) 2014 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import ap_spec |
| import linksys_ap_configurator |
| |
| |
| class LinksysWRT54GS2APConfigurator( |
| linksys_ap_configurator.LinksysAPConfigurator): |
| """Derived class to control Linksys WRT54GS2 router.""" |
| |
| def get_supported_modes(self): |
| return [{'band': ap_spec.BAND_2GHZ, |
| 'modes': [ap_spec.MODE_B, ap_spec.MODE_G, ap_spec.MODE_M]}] |
| |
| |
| def _set_mode(self, mode): |
| mode_mapping = {ap_spec.MODE_B: 'B-Only', ap_spec.MODE_G: 'G-Only', |
| ap_spec.MODE_M: 'Mixed', 'Disabled' : 'Disabled'} |
| mode_name = None |
| if mode in self.get_supported_modes()[0]['modes']: |
| mode_name = mode_mapping.get(mode) |
| else: |
| mode_name = mode |
| if not mode_name: |
| raise RuntimeError('The mode selected %d is not supported by router' |
| ' %s.', hex(mode), self.name) |
| xpath = ('//select[@name="wl_net_mode"]') |
| self.select_item_from_popup_by_xpath(mode_name, xpath) |
