Преобразование файла JSON в Python объекты - PullRequest
Новая
       4

Преобразование файла JSON в Python объекты

0 голосов
/ 16 июня 2020

У меня есть файл JSON, который я хочу взять и поместить в объекты python. Он состоит из двух частей, персонала и активов, и я хочу загрузить их в две отдельные части. Вот образец файла JSON: 

{
"staff": [
    {
        "id": "DA7153",
        "name": [
            "Fran\u00c3\u00a7ois",
            "Ullman"
        ],
        "department": {
            "name": "Admin"
        },
        "server_admin": "true"
    },
    {
        "id": "DA7356",
        "name": [
            "Bob",
            "Johnson"
        ],
        "department": {
            "name": "Admin"
        },
        "server_admin": "false"
    },
],
"assets": [
    {
        "asset_name": "ENGAGED SLOTH",
        "asset_type": "File",
        "owner": "DA8333",
        "details": {
            "security": {
                "cia": [
                    "HIGH",
                    "INTERMEDIATE",
                    "LOW"
                ],
                "data_categories": {
                    "Personal": "true",
                    "Personal Sensitive": "true",
                    "Customer Sensitive": "true"
                }
            },
            "retention": 2
        },
        "file_type": "Document",
        "server": {
            "server_name": "ISOLATED UGUISU",
            "ip": [
                10,
                234,
                148,
                52
            ]
        }
    },
    {
        "asset_name": "ISOLATED VIPER",
        "asset_type": "File",
        "owner": "DA8262",
        "details": {
            "security": {
                "cia": [
                    "LOW",
                    "HIGH",
                    "LOW"
                ],
                "data_categories": {
                    "Personal": "false",
                    "Personal Sensitive": "false",
                    "Customer Sensitive": "true"
                }
            },
            "retention": 2
        },
    },
]

Я попытался создать класс для персонала, но всякий раз, когда я это делаю, получаю сообщение об ошибке «TypeError: dict ожидал не более 1 аргумента, получил 3»

Код, который я использую, выглядит так: 

import json

with open('Admin_sample.json') as f:
    admin_json = json.load(f)

class staffmem(admin_json):
    def __init__(self, id, name, department, server_admin):
        self.id = id
        self.name = name
        self.deparment = department[name]
        self.server_admin = server_admin

    def staffid(self):
        return self.id

print(staffmem.staffid)

Я просто не могу с этим разобраться. Любая помощь будет принята с благодарностью.

Спасибо.

1 Ответ

0 голосов
/ 16 июня 2020

Следующее должно быть хорошей отправной точкой, но вам нужно исправить несколько вещей. Обратите внимание, что я использую get() везде, чтобы обеспечить «безопасное» значение по умолчанию, если ключи не существуют: 

import json

class StaffMember:
    def __init__(self, json_entry):
        self.name = ",".join(json_entry.get("name"))
        self.id = json_entry.get("id")
        self.dept = json_entry.get("department", {}).get("name")
        self.server_admin = (
            True
            if json_entry.get("server_admin", "false").lower() == "true"
            else False
        )

# Get the data
with open("/tmp/test.data") as f:
    data = json.load(f)

# For every entry in the data["staff"] create object and index them by ID
all_staff = {}
for json_entry in data.get("staff", []):
    tmp = StaffMember(json_entry)
    all_staff[tmp.id] = tmp


print(all_staff)
print(all_staff['DA7153'].name)

Вывод: 

$ python3 /tmp/test.py
{'DA7153': <__main__.StaffMember object at 0x1097b2d50>, 'DA7356': <__main__.StaffMember object at 0x1097b2d90>}
François,Ullman

Возможные улучшения:

  • Обработка Unicode
  • Добавить геттеры / сеттеры
  • Вместо передачи json dict в ctor рассмотрите возможность добавления метода from_json() stati c для создания вашего объекта
  • Обработка ошибок при пропущенных значениях
  • Рассмотрите возможность использования класса данных в py3, если этот объект используется только для / в основном хранения данных
  • Рассмотрите подход namedtuple из комментариев, если вы не собираетесь для изменения объекта (только для чтения)

Примечания:

  • Указанный вами json неверен - вам нужно будет его исправить
  • Ваш синтаксис неверен в вашем примере, и соглашение об именах не так много pythoni c (подробнее здесь
...