Перевірте, чи існує ключ, і повторіть масив JSON за допомогою Python


130

У мене є маса даних JSON з публікацій у Facebook, як-от нижче:

{"from": {"id": "8", "name": "Mary Pinter"}, "message": "How ARE you?", "comments": {"count": 0}, "updated_time": "2012-05-01", "created_time": "2012-05-01", "to": {"data": [{"id": "1543", "name": "Honey Pinter"}]}, "type": "status", "id": "id_7"}

Дані JSON напівструктуровані, і все не однакове. Нижче мій код:

import json 

str = '{"from": {"id": "8", "name": "Mary Pinter"}, "message": "How ARE you?", "comments": {"count": 0}, "updated_time": "2012-05-01", "created_time": "2012-05-01", "to": {"data": [{"id": "1543", "name": "Honey Pinter"}]}, "type": "status", "id": "id_7"}'
data = json.loads(str)

post_id = data['id']
post_type = data['type']
print(post_id)
print(post_type)

created_time = data['created_time']
updated_time = data['updated_time']
print(created_time)
print(updated_time)

if data.get('application'):
    app_id = data['application'].get('id', 0)
    print(app_id)
else:
    print('null')

#if data.get('to'):
#... This is the part I am not sure how to do
# Since it is in the form "to": {"data":[{"id":...}]}

Я хочу, щоб код надрукував to_id, як 1543 інше надрукувати "null"

Я не впевнений, як це зробити.

Відповіді:


162
import json

jsonData = """{"from": {"id": "8", "name": "Mary Pinter"}, "message": "How ARE you?", "comments": {"count": 0}, "updated_time": "2012-05-01", "created_time": "2012-05-01", "to": {"data": [{"id": "1543", "name": "Honey Pinter"}]}, "type": "status", "id": "id_7"}"""

def getTargetIds(jsonData):
    data = json.loads(jsonData)
    if 'to' not in data:
        raise ValueError("No target in given data")
    if 'data' not in data['to']:
        raise ValueError("No data for target")

    for dest in data['to']['data']:
        if 'id' not in dest:
            continue
        targetId = dest['id']
        print("to_id:", targetId)

Вихід:

In [9]: getTargetIds(s)
to_id: 1543

6
Чому це явні inперевірки і raiseякщо вони відсутні? Просто перейдіть до нього, не перевіряючи, і ви отримаєте абсолютно таку ж поведінку (за винятком KeyErrorзамість а ValueError).
abarnert

100

Якщо все, що вам потрібно, це перевірити, чи існує ключ чи ні

h = {'a': 1}
'b' in h # returns False

Якщо ви хочете перевірити, чи є значення для ключа

h.get('b') # returns None

Поверніть значення за замовчуванням, якщо фактичне значення відсутнє

h.get('b', 'Default value')

поверне 'null', а не 'Default value', як очікувалося для b у випадку {'a': 1, 'b': null}
MikeL

16

Доброю практикою є створення допоміжних методів корисних програм для таких речей, щоб кожен раз, коли вам потрібно змінити логіку перевірки атрибутів, він знаходився б в одному місці, і код буде читабельнішим для послідовників.

Наприклад, створити допоміжний метод (або клас JsonUtilsзі статичними методами) у json_utils.py:

def get_attribute(data, attribute, default_value):
    return data.get(attribute) or default_value

а потім використовувати його у своєму проекті:

from json_utils import get_attribute

def my_cool_iteration_func(data):

    data_to = get_attribute(data, 'to', None)
    if not data_to:
        return

    data_to_data = get_attribute(data_to, 'data', [])
    for item in data_to_data:
        print('The id is: %s' % get_attribute(item, 'id', 'null'))

ВАЖЛИВА ПРИМІТКА:

Є причина, яку я використовую, data.get(attribute) or default_valueа не просто data.get(attribute, default_value):

{'my_key': None}.get('my_key', 'nothing') # returns None
{'my_key': None}.get('my_key') or 'nothing' # returns 'nothing'

У моїх програмах отримання атрибута зі значенням 'null' - це те саме, що не отримувати атрибут взагалі. Якщо ваше використання відрізняється, вам потрібно це змінити.


4
jsonData = """{"from": {"id": "8", "name": "Mary Pinter"}, "message": "How ARE you?", "comments": {"count": 0}, "updated_time": "2012-05-01", "created_time": "2012-05-01", "to": {"data": [{"id": "1543", "name": "Honey Pinter"}, {"name": "Joe Schmoe"}]}, "type": "status", "id": "id_7"}"""

def getTargetIds(jsonData):
    data = json.loads(jsonData)
    for dest in data['to']['data']:
        print("to_id:", dest.get('id', 'null'))

Спробуй це:

>>> getTargetIds(jsonData)
to_id: 1543
to_id: null

Або, якщо ви просто хочете пропустити значення, які відсутні, ідентифікатори замість друку 'null':

def getTargetIds(jsonData):
    data = json.loads(jsonData)
    for dest in data['to']['data']:
        if 'id' in to_id:
            print("to_id:", dest['id'])

Так:

>>> getTargetIds(jsonData)
to_id: 1543

Звичайно, в реальному житті ви, мабуть, не хочете printкожен ідентифікатор, а зберігати їх і робити з ними щось, але це вже інше питання.



4

Я для цієї мети написав крихітну функцію. Не соромтеся переробляти,

def is_json_key_present(json, key):
    try:
        buf = json[key]
    except KeyError:
        return False

    return True
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.