38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
import json
|
|
from datetime import date
|
|
from pathlib import Path
|
|
|
|
from yt_dlp.networking.impersonate import ImpersonateTarget
|
|
from yt_dlp.utils import DateRange
|
|
|
|
|
|
class Encoder(json.JSONEncoder):
|
|
"""
|
|
This class is used to serialize objects to JSON.
|
|
The only difference between this and the default JSONEncoder is that this one
|
|
will call the __dict__ method of an object if it exists.
|
|
"""
|
|
|
|
def default(self, o):
|
|
from .ItemDTO import ItemDTO
|
|
|
|
if isinstance(o, DateRange):
|
|
return {"start": str(o.start).replace("-", ""), "end": str(o.end).replace("-", "")}
|
|
|
|
if isinstance(o, (Path, date, ImpersonateTarget, ValueError)):
|
|
return str(o)
|
|
|
|
if isinstance(o, ItemDTO):
|
|
return o.serialize()
|
|
|
|
if isinstance(o, object):
|
|
if hasattr(o, "serialize"):
|
|
return o.serialize()
|
|
|
|
if hasattr(o, "model_dump"):
|
|
return o.model_dump()
|
|
|
|
if hasattr(o, "__dict__"):
|
|
return o.__dict__
|
|
|
|
return json.JSONEncoder.default(self, o)
|