codekingpro/portable-devtools

1# Protocol Buffers - Google's data interchange format2# Copyright 2008 Google Inc.  All rights reserved.3#4# Use of this source code is governed by a BSD-style5# license that can be found in the LICENSE file or at6# https://developers.google.com/open-source/licenses/bsd7 8"""Contains Unknown Fields APIs.9 10Simple usage example:11  unknown_field_set = UnknownFieldSet(message)12  for unknown_field in unknown_field_set:13    wire_type = unknown_field.wire_type14    field_number = unknown_field.field_number15    data = unknown_field.data16"""17 18 19from google.protobuf.internal import api_implementation20 21if api_implementation._c_module is not None:  # pylint: disable=protected-access22  UnknownFieldSet = api_implementation._c_module.UnknownFieldSet  # pylint: disable=protected-access23else:24  from google.protobuf.internal import decoder  # pylint: disable=g-import-not-at-top25  from google.protobuf.internal import wire_format  # pylint: disable=g-import-not-at-top26 27  class UnknownField:28    """A parsed unknown field."""29 30    # Disallows assignment to other attributes.31    __slots__ = ['_field_number', '_wire_type', '_data']32 33    def __init__(self, field_number, wire_type, data):34      self._field_number = field_number35      self._wire_type = wire_type36      self._data = data37      return38 39    @property40    def field_number(self):41      return self._field_number42 43    @property44    def wire_type(self):45      return self._wire_type46 47    @property48    def data(self):49      return self._data50 51  class UnknownFieldSet:52    """UnknownField container."""53 54    # Disallows assignment to other attributes.55    __slots__ = ['_values']56 57    def __init__(self, msg):58 59      def InternalAdd(field_number, wire_type, data):60        unknown_field = UnknownField(field_number, wire_type, data)61        self._values.append(unknown_field)62 63      self._values = []64      msg_des = msg.DESCRIPTOR65      # pylint: disable=protected-access66      unknown_fields = msg._unknown_fields67      if (msg_des.has_options and68          msg_des.GetOptions().message_set_wire_format):69        local_decoder = decoder.UnknownMessageSetItemDecoder()70        for _, buffer in unknown_fields:71          (field_number, data) = local_decoder(memoryview(buffer))72          InternalAdd(field_number, wire_format.WIRETYPE_LENGTH_DELIMITED, data)73      else:74        for tag_bytes, buffer in unknown_fields:75          field_number, wire_type = decoder.DecodeTag(tag_bytes)76          if field_number == 0:77            raise RuntimeError('Field number 0 is illegal.')78          (data, _) = decoder._DecodeUnknownField(79              memoryview(buffer), 0, len(buffer), field_number, wire_type80          )81          InternalAdd(field_number, wire_type, data)82 83    def __getitem__(self, index):84      size = len(self._values)85      if index < 0:86        index += size87      if index < 0 or index >= size:88        raise IndexError('index %d out of range'.index)89 90      return self._values[index]91 92    def __len__(self):93      return len(self._values)94 95    def __iter__(self):96      return iter(self._values)97