1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
######################################################################
#
# File: b2/_internal/json_encoder.py
#
# Copyright 2020 Backblaze Inc. All Rights Reserved.
#
# License https://www.backblaze.com/using_b2_code.html
#
######################################################################
import json
from enum import Enum
from b2sdk.v2 import Bucket, DownloadVersion, FileIdAndName, FileVersion
class B2CliJsonEncoder(json.JSONEncoder):
"""
Makes it possible to serialize b2sdk objects
(specifically bucket['options'] set and FileVersionInfo/FileIdAndName) to json.
>>> json.dumps(set([1,2,3,'a','b','c']), cls=json_encoder.B2CliJsonEncoder)
'[1, 2, 3, "c", "b", "a"]'
>>>
"""
def default(self, obj):
if isinstance(obj, set):
return list(obj)
elif isinstance(obj, (DownloadVersion, FileVersion, FileIdAndName, Bucket)):
return obj.as_dict()
elif isinstance(obj, Enum):
return obj.value
return super().default(obj)
|
