diff options
author | Koren Lev <korenlev@gmail.com> | 2017-07-27 15:04:07 +0000 |
---|---|---|
committer | Gerrit Code Review <gerrit@opnfv.org> | 2017-07-27 15:04:07 +0000 |
commit | 162c03ef301396072c1934e7e7e0c40a841b4fe2 (patch) | |
tree | 7a2a2781949252436450ff5832962785061a774a /app/utils/string_utils.py | |
parent | b88c78e3cf2bef22aa2f1c4d0bf305e303bc15f0 (diff) | |
parent | 7e83d0876ddb84a45e130eeba28bc40ef53c074b (diff) |
Merge "Calipso initial release for OPNFV"
Diffstat (limited to 'app/utils/string_utils.py')
-rw-r--r-- | app/utils/string_utils.py | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/app/utils/string_utils.py b/app/utils/string_utils.py new file mode 100644 index 0000000..1f51992 --- /dev/null +++ b/app/utils/string_utils.py @@ -0,0 +1,59 @@ +############################################################################### +# Copyright (c) 2017 Koren Lev (Cisco Systems), Yaron Yogev (Cisco Systems) # +# and others # +# # +# All rights reserved. This program and the accompanying materials # +# are made available under the terms of the Apache License, Version 2.0 # +# which accompanies this distribution, and is available at # +# http://www.apache.org/licenses/LICENSE-2.0 # +############################################################################### +import json +from datetime import datetime + +from bson import ObjectId + + +def jsonify(obj, prettify=False): + if prettify: + return json.dumps(obj, sort_keys=True, indent=4, separators=(',', ': ')) + else: + return json.dumps(obj) + + +# stringify datetime object +def stringify_datetime(dt): + return dt.strftime("%Y-%m-%dT%H:%M:%S.%f%z") + + +# stringify ObjectId +def stringify_object_id(object_id): + return str(object_id) + + +stringify_map = { + ObjectId: stringify_object_id, + datetime: stringify_datetime +} + + +def stringify_object_values_by_type(obj, object_type): + if isinstance(obj, dict): + for key, value in obj.items(): + if isinstance(value, object_type): + obj[key] = stringify_map[object_type](value) + else: + stringify_object_values_by_type(value, object_type) + elif isinstance(obj, list): + for index, value in enumerate(obj): + if isinstance(value, object_type): + obj[index] = stringify_map[object_type](value) + else: + stringify_object_values_by_type(value, object_type) + + +# convert some values of the specific types of the object into string +# e.g convert all the ObjectId to string +# convert all the datetime object to string +def stringify_object_values_by_types(obj, object_types): + for object_type in object_types: + stringify_object_values_by_type(obj, object_type) |