| #!/usr/bin/env python |
| # |
| # Copyright 2016 the original author or authors. |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # |
| |
| """ |
| Device level CLI commands |
| """ |
| from optparse import make_option |
| from cmd2 import Cmd, options |
| from simplejson import dumps |
| |
| from cli.table import print_pb_as_table, print_pb_list_as_table |
| from cli.utils import print_flows, pb2dict |
| from voltha.protos import third_party |
| |
| _ = third_party |
| from voltha.protos import voltha_pb2 |
| from voltha.protos.device_pb2 import PmConfigs, PmConfig, PmGroupConfig |
| from google.protobuf.json_format import MessageToDict |
| |
| # Since proto3 won't send fields that are set to 0/false/"" any object that |
| # might have those values set in them needs to be replicated here such that the |
| # fields can be adequately |
| |
| class DeviceCli(Cmd): |
| |
| def __init__(self, get_channel, device_id): |
| Cmd.__init__(self) |
| self.get_channel = get_channel |
| self.device_id = device_id |
| self.prompt = '(' + self.colorize( |
| self.colorize('device {}'.format(device_id), 'red'), 'bold') + ') ' |
| self.pm_config_last = None |
| self.pm_config_dirty = False |
| |
| def cmdloop(self): |
| self._cmdloop() |
| |
| def get_device(self, depth=0): |
| stub = voltha_pb2.VolthaLocalServiceStub(self.get_channel()) |
| res = stub.GetDevice(voltha_pb2.ID(id=self.device_id), |
| metadata=(('get-depth', str(depth)), )) |
| return res |
| |
| do_exit = Cmd.do_quit |
| |
| def do_show(self, line): |
| """Show detailed device information""" |
| print_pb_as_table('Device {}'.format(self.device_id), |
| self.get_device(depth=-1)) |
| |
| def do_ports(self, line): |
| """Show ports of device""" |
| device = self.get_device(depth=-1) |
| omit_fields = { |
| } |
| print_pb_list_as_table('Device ports:', device.ports, |
| omit_fields, self.poutput) |
| |
| @options([ |
| make_option('-f', '--default_freq', action="store", dest='default_freq', |
| type='long', default=None), |
| make_option('-e', '--enable', action='append', dest='enable', |
| default=None), |
| make_option('-d', '--disable', action='append', dest='disable', |
| default=None), |
| ]) |
| def do_perf_config(self, line, opts): |
| print(line) |
| """Show and set the performance monitoring configuration of the device""" |
| |
| device = self.get_device(depth=-1) |
| if not self.pm_config_last: |
| self.pm_config_last = device.pm_configs |
| |
| # Ensure that a valid sub-command was provided |
| if line.strip() not in {"set", "show", "commit", "reset", ""}: |
| self.poutput(self.colorize('Error: ', 'red') + \ |
| self.colorize(self.colorize(line.strip(), 'blue'), |
| 'bold') + ' is not recognized') |
| return |
| |
| # Ensure no options are provided when requesting to view the config |
| if line.strip() == "show" or line.strip() == "": |
| if opts.default_freq or opts.enable or opts.disable: |
| self.poutput(opts.disable) |
| self.poutput(self.colorize('Error: ', 'red') + 'use ' + \ |
| self.colorize(self.colorize('"set"', 'blue'), |
| 'bold') + ' to change settings') |
| return |
| |
| if line.strip() == "set": # Set the supplied values |
| # The defualt frequency |
| if opts.default_freq: |
| self.pm_config_last.default_freq = opts.default_freq |
| self.pm_config_dirty = True |
| |
| if self.pm_config_last.grouped: |
| for g in self.pm_config_last.groups: |
| if opts.enable: |
| if g.group_name in opts.enable: |
| g.enabled = True |
| self.pm_config_dirty = True |
| for g in self.pm_config_last.groups: |
| if opts.disable: |
| if g.group_name in opts.disable: |
| g.enabled = False |
| self.pm_config_dirty = True |
| else: |
| for m in self.pm_config_last.metrics: |
| if opts.enable: |
| if m.name in opts.enable: |
| m.enabled = True |
| self.pm_config_dirty = True |
| for m in self.pm_config_last.metrics: |
| if opts.disable: |
| if m.name in opts.disable: |
| m.enabled = False |
| self.pm_config_dirty = True |
| #TODO: Add frequency overrides. |
| |
| elif line.strip() == "commit" and self.pm_config_dirty: |
| stub = voltha_pb2.VolthaLocalServiceStub(self.get_channel()) |
| stub.UpdateDevicePmConfigs(self.pm_config_last) |
| self.pm_config_last = self.get_device(depth=-1).pm_configs |
| self.pm_config_dirty = False |
| elif line.strip() == "reset" and self.pm_config_dirty: |
| self.pm_config_last = self.get_device(depth=-1).pm_configs |
| self.pm_config_dirty = False |
| |
| omit_fields = {'groups', 'metrics', 'id'} |
| print_pb_as_table('PM Config:', self.pm_config_last, omit_fields, |
| self.poutput) |
| if self.pm_config_last.grouped: |
| #self.poutput("Supported metric groups:") |
| for g in self.pm_config_last.groups: |
| if self.pm_config_last.freq_override: |
| omit_fields = {'metrics'} |
| else: |
| omit_fields = {'group_freq','metrics'} |
| print_pb_as_table('', g, omit_fields, self.poutput) |
| if g.enabled: |
| state = 'enabled' |
| else: |
| state = 'disabled' |
| print_pb_list_as_table( |
| 'Metric group {} is {}'.format(g.name,state), |
| g.metrics, {'enabled', 'sample_freq'}, self.poutput, |
| dividers=100) |
| else: |
| if self.pm_config_last.freq_override: |
| omit_fields = {} |
| else: |
| omit_fields = {'sample_freq'} |
| print_pb_list_as_table('Supported metrics:', self.pm_config_last.metrics, |
| omit_fields, self.poutput, dividers=100) |
| |
| |
| def do_flows(self, line): |
| """Show flow table for device""" |
| device = pb2dict(self.get_device(-1)) |
| print_flows( |
| 'Device', |
| self.device_id, |
| type=device['type'], |
| flows=device['flows']['items'], |
| groups=device['flow_groups']['items'] |
| ) |
| |