Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(735)

Side by Side Diff: infra_libs/ts_mon/common/pb_to_popo.py

Issue 2213143002: Add infra_libs as a bootstrap dependency. (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@master
Patch Set: Removed the ugly import hack Created 4 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 # Copyright 2015 The Chromium Authors. All rights reserved.
2 # Use of this source code is governed by a BSD-style license that can be
3 # found in the LICENSE file.
4
5 import logging
6
7 from google.protobuf.descriptor import FieldDescriptor as fd
8
9 def convert(pb):
10 """Convert protobuf to plain-old-python-object"""
11 obj = {}
12 for field, value in pb.ListFields():
13 if field.label == fd.LABEL_REPEATED:
14 obj[field.name] = list(_get_json_func(field.type)(v) for v in value)
15 else:
16 obj[field.name] = _get_json_func(field.type)(value)
17 return obj
18
19 def _get_json_func(field_type):
20 if field_type in _FD_TO_JSON:
21 return _FD_TO_JSON[field_type]
22 else: # pragma: no cover
23 logging.warning("pb_to_popo doesn't support converting %s", field_type)
24 return unicode
25
26 _FD_TO_JSON = {
27 fd.TYPE_BOOL: bool,
28 fd.TYPE_DOUBLE: float,
29 fd.TYPE_ENUM: int,
30 fd.TYPE_FIXED32: float,
31 fd.TYPE_FIXED64: float,
32 fd.TYPE_FLOAT: float,
33 fd.TYPE_INT32: int,
34 fd.TYPE_INT64: long,
35 fd.TYPE_SFIXED32: float,
36 fd.TYPE_SFIXED64: float,
37 fd.TYPE_SINT32: int,
38 fd.TYPE_SINT64: long,
39 fd.TYPE_STRING: unicode,
40 fd.TYPE_UINT32: int,
41 fd.TYPE_UINT64: long,
42 fd.TYPE_MESSAGE: convert
43 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698