Mã hóa JSON tùy chỉnh

Mã hóa JSON tùy chỉnh là một tính năng mạnh mẽ trong module json của Python, cho phép bạn tuần tự hóa các đối tượng Python phức tạp mà JSON không hỗ trợ sẵn. Tính năng này đặc biệt hữu ích khi bạn làm việc với các lớp hoặc đối tượng tùy chỉnh không có biểu diễn JSON trực tiếp.

Lớp JSONEncoder

Để tạo một bộ mã hóa JSON tùy chỉnh, bạn cần tạo lớp con từ lớp json.JSONEncoder và ghi đè phương thức default() của lớp đó:

import json

class CustomEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, YourCustomClass):
            return obj.to_dict()  # Chuyển đối tượng của bạn thành từ điển
        return json.JSONEncoder.default(self, obj)

Sử dụng bộ mã hóa tùy chỉnh

Sau khi định nghĩa bộ mã hóa tùy chỉnh, bạn có thể dùng nó với json.dumps() hoặc json.dump():

json_string = json.dumps(your_object, cls=CustomEncoder)

Ví dụ: mã hóa một lớp tùy chỉnh

Cùng xem một ví dụ thực tế:

import json
from datetime import datetime

class Person:
    def __init__(self, name, birthdate):
        self.name = name
        self.birthdate = birthdate

class CustomEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, Person):
            return {"name": obj.name, "birthdate": obj.birthdate.isoformat()}
        elif isinstance(obj, datetime):
            return obj.isoformat()
        return json.JSONEncoder.default(self, obj)

person = Person("Alice", datetime(1990, 5, 15))
json_string = json.dumps(person, cls=CustomEncoder, indent=2)
print(json_string)

Ví dụ này minh họa cách mã hóa lớp tùy chỉnh Person và xử lý đối tượng datetime bên trong lớp đó.

Điểm chính

• Phương thức default() được gọi cho các đối tượng không thể tuần tự hóa bằng cách tích hợp sẵn.
• Luôn cung cấp cách quay về bộ mã hóa mặc định khi gặp các kiểu dữ liệu không xác định.
• Bộ mã hóa tùy chỉnh hữu ích khi xử lý các cấu trúc dữ liệu phức tạp, lớp tùy chỉnh và kiểu dữ liệu Python không tiêu chuẩn.

Bằng cách dùng bộ mã hóa JSON tùy chỉnh, bạn có thể tích hợp liền mạch các đối tượng Python phức tạp vào quy trình làm việc với JSON, giúp việc tuần tự hóa dữ liệu linh hoạt và mạnh mẽ hơn.