Spaces:
Runtime error
Runtime error
# Protocol Buffers - Google's data interchange format | |
# Copyright 2008 Google Inc. All rights reserved. | |
# | |
# Use of this source code is governed by a BSD-style | |
# license that can be found in the LICENSE file or at | |
# https://developers.google.com/open-source/licenses/bsd | |
"""Contains Unknown Fields APIs. | |
Simple usage example: | |
unknown_field_set = UnknownFieldSet(message) | |
for unknown_field in unknown_field_set: | |
wire_type = unknown_field.wire_type | |
field_number = unknown_field.field_number | |
data = unknown_field.data | |
""" | |
from google.protobuf.internal import api_implementation | |
if api_implementation._c_module is not None: # pylint: disable=protected-access | |
UnknownFieldSet = api_implementation._c_module.UnknownFieldSet # pylint: disable=protected-access | |
else: | |
from google.protobuf.internal import decoder # pylint: disable=g-import-not-at-top | |
from google.protobuf.internal import wire_format # pylint: disable=g-import-not-at-top | |
class UnknownField: | |
"""A parsed unknown field.""" | |
# Disallows assignment to other attributes. | |
__slots__ = ['_field_number', '_wire_type', '_data'] | |
def __init__(self, field_number, wire_type, data): | |
self._field_number = field_number | |
self._wire_type = wire_type | |
self._data = data | |
return | |
def field_number(self): | |
return self._field_number | |
def wire_type(self): | |
return self._wire_type | |
def data(self): | |
return self._data | |
class UnknownFieldSet: | |
"""UnknownField container.""" | |
# Disallows assignment to other attributes. | |
__slots__ = ['_values'] | |
def __init__(self, msg): | |
def InternalAdd(field_number, wire_type, data): | |
unknown_field = UnknownField(field_number, wire_type, data) | |
self._values.append(unknown_field) | |
self._values = [] | |
msg_des = msg.DESCRIPTOR | |
# pylint: disable=protected-access | |
unknown_fields = msg._unknown_fields | |
if (msg_des.has_options and | |
msg_des.GetOptions().message_set_wire_format): | |
local_decoder = decoder.UnknownMessageSetItemDecoder() | |
for _, buffer in unknown_fields: | |
(field_number, data) = local_decoder(memoryview(buffer)) | |
InternalAdd(field_number, wire_format.WIRETYPE_LENGTH_DELIMITED, data) | |
else: | |
for tag_bytes, buffer in unknown_fields: | |
# pylint: disable=protected-access | |
(tag, _) = decoder._DecodeVarint(tag_bytes, 0) | |
field_number, wire_type = wire_format.UnpackTag(tag) | |
if field_number == 0: | |
raise RuntimeError('Field number 0 is illegal.') | |
(data, _) = decoder._DecodeUnknownField( | |
memoryview(buffer), 0, wire_type) | |
InternalAdd(field_number, wire_type, data) | |
def __getitem__(self, index): | |
size = len(self._values) | |
if index < 0: | |
index += size | |
if index < 0 or index >= size: | |
raise IndexError('index %d out of range'.index) | |
return self._values[index] | |
def __len__(self): | |
return len(self._values) | |
def __iter__(self): | |
return iter(self._values) | |