commit d83e382748d018143ac5e9238f1cc38049ba3ec4 · dwn.poxiao-labs.work/atpasser

src/atpasser/data/__init__.py

···

       0

···

       1
       +
       from .wrapper import *

+76

src/atpasser/data/data.py

···

       1
       +
       import base64

     

       2
       +
       from cid import CIDv0, CIDv1, cid, make_cid

     

       3
       +
       import json

     

       4
       +
       

     

       5
       +
       

     

       6
       +
       class Data:

     

       7
       +
           """

     

       8
       +
           A class representing data with "$type" key.

     

       9
       +
       

     

       10
       +
           Attributes:

     

       11
       +
               type (str): The type of the data.

     

       12
       +
               json (str): Original object in JSON format.

     

       13
       +
           """

     

       14
       +
       

     

       15
       +
           def __init__(self, dataType: str, json: str = "{}") -> None:

     

       16
       +
               """

     

       17
       +
               Initalizes data object.

     

       18
       +
       

     

       19
       +
               Parameters:

     

       20
       +
                   type (str): The type of the data.

     

       21
       +
                   json (str): Original object in JSON format.

     

       22
       +
               """

     

       23
       +
               self.type = dataType

     

       24
       +
               self.json = json

     

       25
       +
       

     

       26
       +
           def data(self):

     

       27
       +
               """

     

       28
       +
               Loads data as a Python-friendly format.

     

       29
       +
       

     

       30
       +
               Returns:

     

       31
       +
                   dict: Converted data from JSON object.

     

       32
       +
               """

     

       33
       +
               return json.loads(self.json, object_hook=dataHook)

     

       34
       +
       

     

       35
       +
       

     

       36
       +
       def dataHook(data: dict):

     

       37
       +
           """

     

       38
       +
           Treated as `JSONDecoder`'s `object_hook`

     

       39
       +
       

     

       40
       +
           Parameters:

     

       41
       +
               data: data in format that `JSONDecoder` like ;)

     

       42
       +
           """

     

       43
       +
           if "$bytes" in data:

     

       44
       +
               return base64.b64decode(data["$bytes"])

     

       45
       +
           elif "$link" in data:

     

       46
       +
               return make_cid(data["$link"])

     

       47
       +
           elif "$type" in data:

     

       48
       +
               dataType = data["$type"]

     

       49
       +
               del data["$type"]

     

       50
       +
               return Data(dataType, json.dumps(data))

     

       51
       +
           else:

     

       52
       +
               return data

     

       53
       +
       

     

       54
       +
       

     

       55
       +
       def _convertDataToFakeJSON(data):

     

       56
       +
           if isinstance(data, bytes):

     

       57
       +
               return {"$bytes": base64.b64encode(data)}

     

       58
       +
           elif isinstance(data, (CIDv0, CIDv1)):

     

       59
       +
               return {"link": data.encode()}

     

       60
       +
           elif isinstance(data, dict):

     

       61
       +
               for item in data:

     

       62
       +
                   data[item] = _convertDataToFakeJSON(data[item])

     

       63
       +
           elif isinstance(data, (tuple, list, set)):

     

       64
       +
               return [_convertDataToFakeJSON(item) for item in data]

     

       65
       +
           else:

     

       66
       +
               return data

     

       67
       +
       

     

       68
       +
       

     

       69
       +
       class DataEncoder(json.JSONEncoder):

     

       70
       +
           """

     

       71
       +
           A superset of `JSONEncoder` to support ATProto data.

     

       72
       +
           """

     

       73
       +
       

     

       74
       +
           def default(self, o):

     

       75
       +
               result = _convertDataToFakeJSON(o)

     

       76
       +
               return super().default(result)

+52

src/atpasser/data/wrapper.py

···

       1
       +
       from json import loads

     

       2
       +
       from typing import Callable, Any

     

       3
       +
       from .data import *

     

       4
       +
       import functools

     

       5
       +
       

     

       6
       +
       # Pyright did the whole job. Thank it.

     

       7
       +
       

     

       8
       +
       

     

       9
       +
       class DataDecoder(json.JSONDecoder):

     

       10
       +
           """

     

       11
       +
           A superset of `JSONDecoder` to support ATProto data.

     

       12
       +
           """

     

       13
       +
       

     

       14
       +
           def __init__(

     

       15
       +
               self,

     

       16
       +
               *,

     

       17
       +
               object_hook: Callable[[dict[str, Any]], Any] | None = dataHook,

     

       18
       +
               parse_float: Callable[[str], Any] | None = None,

     

       19
       +
               parse_int: Callable[[str], Any] | None = None,

     

       20
       +
               parse_constant: Callable[[str], Any] | None = None,

     

       21
       +
               strict: bool = True,

     

       22
       +
               object_pairs_hook: Callable[[list[tuple[str, Any]]], Any] | None = None,

     

       23
       +
           ) -> None:

     

       24
       +
               super().__init__(

     

       25
       +
                   object_hook=object_hook,

     

       26
       +
                   parse_float=parse_float,

     

       27
       +
                   parse_int=parse_int,

     

       28
       +
                   parse_constant=parse_constant,

     

       29
       +
                   strict=strict,

     

       30
       +
                   object_pairs_hook=object_pairs_hook,

     

       31
       +
               )

     

       32
       +
       

     

       33
       +
       

     

       34
       +
       # Screw it. I have to make 4 `json`-like functions.

     

       35
       +
       

     

       36
       +
       

     

       37
       +
       def _dataDecorator(func):

     

       38
       +
           @functools.wraps(func)

     

       39
       +
           def wrapper(obj, *args, **kwargs):

     

       40
       +
               kwargs.setdefault("cls", DataEncoder)

     

       41
       +
               return func(obj, *args, **kwargs)

     

       42
       +
       

     

       43
       +
           return wrapper

     

       44
       +
       

     

       45
       +
       

     

       46
       +
       dump = _dataDecorator(json.dump)

     

       47
       +
       dumps = _dataDecorator(json.dumps)

     

       48
       +
       load = _dataDecorator(json.load)

     

       49
       +
       loads = _dataDecorator(json.loads)

     

       50
       +
       """

     

       51
       +
       Wrapper of the JSON functions to support ATProto data.

     

       52
       +
       """

-179

src/atpasser/dataFormatter/__init__.py

···

       1
       -
       import base64  # Converts between Base64 and byte object

     

       2
       -
       from cid import CIDv0, CIDv1, cid, make_cid  # Converts CID to/from object

     

       3
       -
       import json

     

       4
       -
       

     

       5
       -
       # JSON is a string, however we uses the term JSON for dicts which read

     

       6
       -
       # directly using the json library itself, and data for dicts that are

     

       7
       -
       # friendly to Python by implementing bytes and CID support *or* data with

     

       8
       -
       # a type

     

       9
       -
       

     

       10
       -
       

     

       11
       -
       class Data:

     

       12
       -
           """

     

       13
       -
           Data with a type.

     

       14
       -
           """

     

       15
       -
       

     

       16
       -
           def __init__(self, type: str, content: dict) -> None:

     

       17
       -
               self.type = type

     

       18
       -
               self.content = jsonToData(content)

     

       19
       -
       

     

       20
       -
       

     

       21
       -
       def jsonToData(json):

     

       22
       -
           """

     

       23
       -
           Convert 'JSON' to 'data'

     

       24
       -
           """

     

       25
       -
           if isinstance(json, dict):

     

       26
       -
               if "$bytes" in json:  # bytes shown as `$bytes` object in JSON

     

       27
       -
                   bytesInB64 = json["$bytes"]

     

       28
       -
                   return base64.b64decode(bytesInB64)

     

       29
       -
               elif "$link" in json:  # CID link

     

       30
       -
                   cidInString = json["$link"]

     

       31
       -
                   return make_cid(cidInString)

     

       32
       -
               elif "$type" in json:  # data have their lexicons

     

       33
       -
                   type = json["$type"]

     

       34
       -
                   del json["$type"]  # loop if we don't del

     

       35
       -
                   return Data(type, json)

     

       36
       -
               else:  # general dict

     

       37
       -
                   result = {}

     

       38
       -
                   for item in json:

     

       39
       -
                       result[item] = jsonToData(json[item])

     

       40
       -
                   return result

     

       41
       -
           elif isinstance(json, (list, tuple, set)):  # only lists are valid

     

       42
       -
               return [jsonToData(item) for item in json]

     

       43
       -
           else:

     

       44
       -
               return json

     

       45
       -
       

     

       46
       -
       

     

       47
       -
       def dataToJson(data):

     

       48
       -
           """

     

       49
       -
           Convert 'data' to 'JSON'

     

       50
       -
           """

     

       51
       -
           if isinstance(data, bytes):  # bytes

     

       52
       -
               return {"$bytes": base64.b64encode(data)}

     

       53
       -
           elif isinstance(data, (CIDv0, CIDv1)):  # CID link

     

       54
       -
               return {"$link": str(data)}

     

       55
       -
           elif isinstance(data, dict):  # general dict

     

       56
       -
               result = {}

     

       57
       -
               for item in data:

     

       58
       -
                   result[item] = dataToJson(data[item])

     

       59
       -
               return result

     

       60
       -
           elif isinstance(data, (list, tuple, set)):  # only lists are valid

     

       61
       -
               return [dataToJson(item) for item in data]

     

       62
       -
           else:

     

       63
       -
               return data

     

       64
       -
       

     

       65
       -
       

     

       66
       -
       # We should need a wrapper, converts REAL json TEXT/FILE to so-called data

     

       67
       -
       

     

       68
       -
       

     

       69
       -
       def dump(

     

       70
       -
           obj,

     

       71
       -
           fp,

     

       72
       -
           *w,

     

       73
       -
           skipkeys=False,

     

       74
       -
           ensure_ascii=True,

     

       75
       -
           check_circular=True,

     

       76
       -
           allow_nan=True,

     

       77
       -
           cls=None,

     

       78
       -
           indent=None,

     

       79
       -
           separators=None,

     

       80
       -
           default=None,

     

       81
       -
           sort_keys=False,

     

       82
       -
           **kw

     

       83
       -
       ):

     

       84
       -
           """

     

       85
       -
           json.dump() like function which replaces json to data instead

     

       86
       -
           """

     

       87
       -
           json_result = dataToJson(obj)

     

       88
       -
           json.dump(

     

       89
       -
               json_result,

     

       90
       -
               fp,

     

       91
       -
               *w,

     

       92
       -
               skipkeys=skipkeys,

     

       93
       -
               ensure_ascii=ensure_ascii,

     

       94
       -
               check_circular=check_circular,

     

       95
       -
               allow_nan=allow_nan,

     

       96
       -
               cls=cls,

     

       97
       -
               indent=indent,

     

       98
       -
               separators=separators,

     

       99
       -
               default=default,

     

       100
       -
               sort_keys=sort_keys,

     

       101
       -
               **kw,

     

       102
       -
           )

     

       103
       -
       

     

       104
       -
       

     

       105
       -
       def dumps(

     

       106
       -
           obj,

     

       107
       -
           *w,

     

       108
       -
           skipkeys=False,

     

       109
       -
           ensure_ascii=True,

     

       110
       -
           check_circular=True,

     

       111
       -
           allow_nan=True,

     

       112
       -
           cls=None,

     

       113
       -
           indent=None,

     

       114
       -
           separators=None,

     

       115
       -
           default=None,

     

       116
       -
           sort_keys=False,

     

       117
       -
           **kw

     

       118
       -
       ):

     

       119
       -
           """

     

       120
       -
           json.dumps() like function which replaces json to data instead

     

       121
       -
           """

     

       122
       -
           json_result = dataToJson(obj)

     

       123
       -
           json.dumps(

     

       124
       -
               json_result,

     

       125
       -
               *w,

     

       126
       -
               skipkeys=skipkeys,

     

       127
       -
               ensure_ascii=ensure_ascii,

     

       128
       -
               check_circular=check_circular,

     

       129
       -
               allow_nan=allow_nan,

     

       130
       -
               cls=cls,

     

       131
       -
               indent=indent,

     

       132
       -
               separators=separators,

     

       133
       -
               default=default,

     

       134
       -
               sort_keys=sort_keys,

     

       135
       -
               **kw,

     

       136
       -
           )

     

       137
       -
       

     

       138
       -
       

     

       139
       -
       def load(

     

       140
       -
           fp,

     

       141
       -
           *w,

     

       142
       -
           cls=None,

     

       143
       -
           object_hook=None,

     

       144
       -
           parse_float=None,

     

       145
       -
           parse_int=None,

     

       146
       -
           parse_constant=None,

     

       147
       -
           object_pairs_hook=None,

     

       148
       -
           **kw

     

       149
       -
       ):

     

       150
       -
           """

     

       151
       -
           json.load() like function which replaces json to data instead

     

       152
       -
           """

     

       153
       -
           result = json.load(

     

       154
       -
               fp,

     

       155
       -
               *w,

     

       156
       -
               cls=cls,

     

       157
       -
               object_hook=object_hook,

     

       158
       -
               parse_float=parse_float,

     

       159
       -
               parse_int=parse_int,

     

       160
       -
               parse_constant=parse_constant,

     

       161
       -
               object_pairs_hook=object_pairs_hook,

     

       162
       -
               **kw,

     

       163
       -
           )

     

       164
       -
           return jsonToData(result)

     

       165
       -
       

     

       166
       -
       

     

       167
       -
       def loads(

     

       168
       -
           s,

     

       169
       -
           *w,

     

       170
       -
           cls=None,

     

       171
       -
           object_hook=None,

     

       172
       -
           parse_float=None,

     

       173
       -
           parse_int=None,

     

       174
       -
           parse_constant=None,

     

       175
       -
           object_pairs_hook=None,

     

       176
       -
           **kw

     

       177
       -
       ):

     

       178
       -
           result = json.loads(s,*w,cls=cls,object_hook=object_hook,parse_float=parse_float,parse_int=parse_int,parse_constant=parse_constant,object_pairs_hook=object_pairs_hook,**kw)

     

       179
       -
           return jsonToData(result)