如何基于REST API返回的JSON动态创建Python对象？

嘿，这个需求在日常开发里特别常见，咱们可以通过「请求API获取JSON数据」+「将JSON字段映射到Python类属性」这两步轻松搞定，我给你具体的代码示例和思路：

第一步：获取API的JSON数据 #

首先咱们需要用requests库来调用API获取JSON响应（如果还没安装，先跑pip install requests搞定依赖）。写个通用的请求函数：

import requests

def fetch_api_json(url):
    response = requests.get(url)
    response.raise_for_status()  # 自动处理请求失败的情况
    return response.json()

第二步：将JSON映射到Python类 #

你的Employee和Task类已经用**kwargs初始化，天生适合直接从JSON字典转换，不过Task里的employee是嵌套的Employee对象，得单独处理两种常见场景：

场景1：API返回的Task中employee字段是员工ID #

比如http://example.com/task返回的JSON结构是这样的：

{
  "ID": 1,
  "start_date": "2024-05-20",
  "employee": 1001
}

直接用你现有的类就行，写个转换函数：

def build_employee(employee_json):
    return Employee(**employee_json)

def build_task(task_json):
    # 从task字典里取出员工ID，先创建Employee实例
    task_json['employee'] = Employee(ID=task_json['employee'])
    return Task(**task_json)

场景2：API返回的Task中employee是完整员工对象 #

如果http://example.com/task返回的JSON里employee是完整的员工信息，比如：

{
  "ID": 1,
  "start_date": "2024-05-20",
  "employee": {
    "ID": 1001,
    "first_name": "John",
    "last_name": "Doe"
  }
}

可以给Task类的初始化逻辑加些灵活性，或者在转换函数里处理：

# 方案1：修改Task类的__init__，兼容两种传入方式
class Task(object):
    def __init__(self, **kwargs):
        self.ID = kwargs['ID']
        self.start_date = kwargs['start_date']
        employee_data = kwargs['employee']
        # 如果是字典就直接转Employee，是ID就用ID初始化
        if isinstance(employee_data, dict):
            self.employee = Employee(**employee_data)
        else:
            self.employee = Employee(ID=employee_data)

# 方案2：在转换函数里手动处理嵌套
def build_task(task_json):
    employee_dict = task_json.pop('employee')
    task = Task(**task_json)
    task.employee = build_employee(employee_dict)
    return task

完整使用示例 #

把这些逻辑整合起来，调用API并创建对象：

# 获取员工数据并生成对象
emp_json = fetch_api_json("http://example.com/employee")
employee = build_employee(emp_json)

# 获取任务数据并生成对象
task_json = fetch_api_json("http://example.com/task")
task = build_task(task_json)

# 测试输出
print(f"任务ID：{task.ID}，负责人：{task.employee.first_name} {task.employee.last_name}")

额外优化建议 #

• 给类添加from_json类方法，让代码更简洁规范：

class Employee(object):
    def __init__(self, **kwargs):
        self.ID = kwargs['ID']
        self.first_name = kwargs.get('first_name')  # 用get()替代if判断，更简洁
        self.last_name = kwargs.get('last_name')
    
    @classmethod
    def from_json(cls, json_data):
        return cls(**json_data)

之后直接用Employee.from_json(emp_json)就能创建对象，更符合Python的风格。

• 字段容错：API返回的JSON可能缺失某些字段，用kwargs.get()比if...else写法更优雅，还能给默认值。

内容的提问来源于stack exchange，提问作者BioGeek