Index: third_party/google/protobuf/internal/enum_type_wrapper.py |
diff --git a/third_party/google/protobuf/internal/enum_type_wrapper.py b/third_party/google/protobuf/internal/enum_type_wrapper.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..7b28645a7b7acce1371e1c856aa5f7d7f4874e15 |
--- /dev/null |
+++ b/third_party/google/protobuf/internal/enum_type_wrapper.py |
@@ -0,0 +1,89 @@ |
+# Protocol Buffers - Google's data interchange format |
+# Copyright 2008 Google Inc. All rights reserved. |
+# http://code.google.com/p/protobuf/ |
+# |
+# Redistribution and use in source and binary forms, with or without |
+# modification, are permitted provided that the following conditions are |
+# met: |
+# |
+# * Redistributions of source code must retain the above copyright |
+# notice, this list of conditions and the following disclaimer. |
+# * Redistributions in binary form must reproduce the above |
+# copyright notice, this list of conditions and the following disclaimer |
+# in the documentation and/or other materials provided with the |
+# distribution. |
+# * Neither the name of Google Inc. nor the names of its |
+# contributors may be used to endorse or promote products derived from |
+# this software without specific prior written permission. |
+# |
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
+# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
+# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
+# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
+ |
+"""A simple wrapper around enum types to expose utility functions. |
+ |
+Instances are created as properties with the same name as the enum they wrap |
+on proto classes. For usage, see: |
+ reflection_test.py |
+""" |
+ |
+__author__ = 'rabsatt@google.com (Kevin Rabsatt)' |
+ |
+ |
+class EnumTypeWrapper(object): |
+ """A utility for finding the names of enum values.""" |
+ |
+ DESCRIPTOR = None |
+ |
+ def __init__(self, enum_type): |
+ """Inits EnumTypeWrapper with an EnumDescriptor.""" |
+ self._enum_type = enum_type |
+ self.DESCRIPTOR = enum_type; |
+ |
+ def Name(self, number): |
+ """Returns a string containing the name of an enum value.""" |
+ if number in self._enum_type.values_by_number: |
+ return self._enum_type.values_by_number[number].name |
+ raise ValueError('Enum %s has no name defined for value %d' % ( |
+ self._enum_type.name, number)) |
+ |
+ def Value(self, name): |
+ """Returns the value coresponding to the given enum name.""" |
+ if name in self._enum_type.values_by_name: |
+ return self._enum_type.values_by_name[name].number |
+ raise ValueError('Enum %s has no value defined for name %s' % ( |
+ self._enum_type.name, name)) |
+ |
+ def keys(self): |
+ """Return a list of the string names in the enum. |
+ |
+ These are returned in the order they were defined in the .proto file. |
+ """ |
+ |
+ return [value_descriptor.name |
+ for value_descriptor in self._enum_type.values] |
+ |
+ def values(self): |
+ """Return a list of the integer values in the enum. |
+ |
+ These are returned in the order they were defined in the .proto file. |
+ """ |
+ |
+ return [value_descriptor.number |
+ for value_descriptor in self._enum_type.values] |
+ |
+ def items(self): |
+ """Return a list of the (name, value) pairs of the enum. |
+ |
+ These are returned in the order they were defined in the .proto file. |
+ """ |
+ return [(value_descriptor.name, value_descriptor.number) |
+ for value_descriptor in self._enum_type.values] |