Alignment/OfflineValidation/python/TkAlAllInOneTool/utils.py - master - githubmirror/cmssw

Alignment/OfflineValidation/python/TkAlAllInOneTool/utils.py

35 строк1 KB

tkello

Updated DMRTrends with fixed luminosity

04 ноя 2022, 05:34

04 ноя 2022, 05:34ec4e9d6

'''
Function get string objects instead of Unicode from JSON using obje_hook
see: https://stackoverflow.com/questions/956867/how-to-get-string-objects-instead-of-unicode-from-json
This is necessary in python versions less than 3.

usage:

def json_load_byteified(file_handle):
    return _byteify(
        json.load(file_handle, object_hook=_byteify),
        ignore_dicts=True
    )

def json_loads_byteified(json_text):
    return _byteify(
        json.loads(json_text, object_hook=_byteify),
        ignore_dicts=True
    )
'''
def _byteify(data, ignore_dicts = False):
    # if this is a unicode string, return its string representation
    if isinstance(data, unicode):
        return data.encode('utf-8')
    # if this is a list of values, return list of byteified values
    if isinstance(data, list):
        return [ _byteify(item, ignore_dicts=True) for item in data ]
    # if this is a dictionary, return dictionary of byteified keys and values
    # but only if we haven't already byteified it
    if isinstance(data, dict) and not ignore_dicts:
        return {
            _byteify(key, ignore_dicts=True): _byteify(value, ignore_dicts=True)
            for key, value in data.iteritems()
        }
    # if it's anything else, return it in its original form
    return data