comparing 85210efb71d0c06b561466fe9a45227b3d071c2c and main on dwn.poxiao-labs.work/atpasser

-16

src/atpasser/blob/__init__.py

···

       1
       1
       -
       import cid

     

       2
       2
       -
       import multihash, hashlib

     

       3
       3
       -
       

     

       4
       4
       -
       

     

       5
       5
       -
       def generateCID(file):

     

       6
       6
       -
           hasher = hashlib.new("sha-256")

     

       7
       7
       -
           while True:

     

       8
       8
       -
               chunk = file.read(8192)

     

       9
       9
       -
               if not chunk:

     

       10
       10
       -
                   break

     

       11
       11
       -
               hasher.update(chunk)

     

       12
       12
       -
       

     

       13
       13
       -
           digest = hasher.digest

     

       14
       14
       -
           mh = multihash.encode(digest, "sha-256")

     

       15
       15
       -
       

     

       16
       16
       -
           return cid.CIDv1(codec="raw", multihash=mh)

-1

src/atpasser/data/__init__.py

···

       1
       1
       -
       from ._wrapper import *

-76

src/atpasser/data/_data.py

···

       1
       1
       -
       import base64

     

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

     

       3
       3
       -
       import json

     

       4
       4
       -
       

     

       5
       5
       -
       

     

       6
       6
       -
       class Data:

     

       7
       7
       -
           """

     

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

     

       9
       9
       -
       

     

       10
       10
       -
           Attributes:

     

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

     

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

     

       13
       13
       -
           """

     

       14
       14
       -
       

     

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

     

       16
       16
       -
               """

     

       17
       17
       -
               Initalizes data object.

     

       18
       18
       -
       

     

       19
       19
       -
               Parameters:

     

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

     

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

     

       22
       22
       -
               """

     

       23
       23
       -
               self.type = dataType

     

       24
       24
       -
               self.json = json

     

       25
       25
       -
       

     

       26
       26
       -
           def data(self):

     

       27
       27
       -
               """

     

       28
       28
       -
               Loads data as a Python-friendly format.

     

       29
       29
       -
       

     

       30
       30
       -
               Returns:

     

       31
       31
       -
                   dict: Converted data from JSON object.

     

       32
       32
       -
               """

     

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

     

       34
       34
       -
       

     

       35
       35
       -
       

     

       36
       36
       -
       def dataHook(data: dict):

     

       37
       37
       -
           """

     

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

     

       39
       39
       -
       

     

       40
       40
       -
           Parameters:

     

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

     

       42
       42
       -
           """

     

       43
       43
       -
           if "$bytes" in data:

     

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

     

       45
       45
       -
           elif "$link" in data:

     

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

     

       47
       47
       -
           elif "$type" in data:

     

       48
       48
       -
               dataType = data["$type"]

     

       49
       49
       -
               del data["$type"]

     

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

     

       51
       51
       -
           else:

     

       52
       52
       -
               return data

     

       53
       53
       -
       

     

       54
       54
       -
       

     

       55
       55
       -
       def _convertDataToFakeJSON(data):

     

       56
       56
       -
           if isinstance(data, bytes):

     

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

     

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

     

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

     

       60
       60
       -
           elif isinstance(data, dict):

     

       61
       61
       -
               for item in data:

     

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

     

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

     

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

     

       65
       65
       -
           else:

     

       66
       66
       -
               return data

     

       67
       67
       -
       

     

       68
       68
       -
       

     

       69
       69
       -
       class DataEncoder(json.JSONEncoder):

     

       70
       70
       -
           """

     

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

     

       72
       72
       -
           """

     

       73
       73
       -
       

     

       74
       74
       -
           def default(self, o):

     

       75
       75
       -
               result = _convertDataToFakeJSON(o)

     

       76
       76
       -
               return super().default(result)

-61

src/atpasser/data/_wrapper.py

···

       1
       1
       -
       from json import loads

     

       2
       2
       -
       from typing import Callable, Any

     

       3
       3
       -
       from ._data import *

     

       4
       4
       -
       import functools

     

       5
       5
       -
       

     

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

     

       7
       7
       -
       

     

       8
       8
       -
       

     

       9
       9
       -
       class DataDecoder(json.JSONDecoder):

     

       10
       10
       -
           """

     

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

     

       12
       12
       -
           """

     

       13
       13
       -
       

     

       14
       14
       -
           def __init__(

     

       15
       15
       -
               self,

     

       16
       16
       -
               *,

     

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

     

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

     

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

     

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

     

       21
       21
       -
               strict: bool = True,

     

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

     

       23
       23
       -
           ) -> None:

     

       24
       24
       -
               super().__init__(

     

       25
       25
       -
                   object_hook=object_hook,

     

       26
       26
       -
                   parse_float=parse_float,

     

       27
       27
       -
                   parse_int=parse_int,

     

       28
       28
       -
                   parse_constant=parse_constant,

     

       29
       29
       -
                   strict=strict,

     

       30
       30
       -
                   object_pairs_hook=object_pairs_hook,

     

       31
       31
       -
               )

     

       32
       32
       -
       

     

       33
       33
       -
       

     

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

     

       35
       35
       -
       

     

       36
       36
       -
       

     

       37
       37
       -
       def _dataDecoratorForDump(func):

     

       38
       38
       -
           @functools.wraps(func)

     

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

     

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

     

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

     

       42
       42
       -
       

     

       43
       43
       -
           return wrapper

     

       44
       44
       -
       

     

       45
       45
       -
       

     

       46
       46
       -
       def _dataDecoratorForLoad(func):

     

       47
       47
       -
           @functools.wraps(func)

     

       48
       48
       -
           def wrapper(obj, *args, **kwargs):

     

       49
       49
       -
               kwargs.setdefault("cls", DataDecoder)

     

       50
       50
       -
               return func(obj, *args, **kwargs)

     

       51
       51
       -
       

     

       52
       52
       -
           return wrapper

     

       53
       53
       -
       

     

       54
       54
       -
       

     

       55
       55
       -
       dump = _dataDecoratorForDump(json.dump)

     

       56
       56
       -
       dumps = _dataDecoratorForDump(json.dumps)

     

       57
       57
       -
       load = _dataDecoratorForLoad(json.load)

     

       58
       58
       -
       loads = _dataDecoratorForLoad(json.loads)

     

       59
       59
       -
       """

     

       60
       60
       -
       Wrapper of the JSON functions to support ATProto data.

     

       61
       61
       -
       """

-137

src/atpasser/data/cbor.py

···

       1
       1
       -
       from datetime import tzinfo

     

       2
       2
       -
       import typing

     

       3
       3
       -
       import cbor2

     

       4
       4
       -
       import cid

     

       5
       5
       -
       

     

       6
       6
       -
       from .data import dataHook, Data

     

       7
       7
       -
       

     

       8
       8
       -
       

     

       9
       9
       -
       def tagHook(decoder: cbor2.CBORDecoder, tag: cbor2.CBORTag, shareable_index=None):

     

       10
       10
       -
           """

     

       11
       11
       -
           A simple tag hook for CID support.

     

       12
       12
       -
           """

     

       13
       13
       -
           return cid.from_bytes(tag.value) if tag.tag == 42 else tag

     

       14
       14
       -
       

     

       15
       15
       -
       

     

       16
       16
       -
       class CBOREncoder(cbor2.CBOREncoder):

     

       17
       17
       -
           """

     

       18
       18
       -
           Wrapper of cbor2.CBOREncoder.

     

       19
       19
       -
           """

     

       20
       20
       -
       

     

       21
       21
       -
           def __init__(

     

       22
       22
       -
               self,

     

       23
       23
       -
               fp: typing.IO[bytes],

     

       24
       24
       -
               datetime_as_timestamp: bool = False,

     

       25
       25
       -
               timezone: tzinfo | None = None,

     

       26
       26
       -
               value_sharing: bool = False,

     

       27
       27
       -
               default: (

     

       28
       28
       -
                   typing.Callable[[cbor2.CBOREncoder, typing.Any], typing.Any] | None

     

       29
       29
       -
               ) = None,

     

       30
       30
       -
               canonical: bool = False,

     

       31
       31
       -
               date_as_datetime: bool = False,

     

       32
       32
       -
               string_referencing: bool = False,

     

       33
       33
       -
               indefinite_containers: bool = False,

     

       34
       34
       -
           ):

     

       35
       35
       -
               super().__init__(

     

       36
       36
       -
                   fp,

     

       37
       37
       -
                   datetime_as_timestamp,

     

       38
       38
       -
                   timezone,

     

       39
       39
       -
                   value_sharing,

     

       40
       40
       -
                   default,

     

       41
       41
       -
                   canonical,

     

       42
       42
       -
                   date_as_datetime,

     

       43
       43
       -
                   string_referencing,

     

       44
       44
       -
                   indefinite_containers,

     

       45
       45
       -
               )

     

       46
       46
       -
       

     

       47
       47
       -
           @cbor2.shareable_encoder

     

       48
       48
       -
           def cidOrDataEncoder(self: cbor2.CBOREncoder, value: cid.CIDv0 | cid.CIDv1 | Data):

     

       49
       49
       -
               """

     

       50
       50
       -
               Encode CID or Data to CBOR Tag.

     

       51
       51
       -
               """

     

       52
       52
       -
               if isinstance(value, (cid.CIDv0, cid.CIDv1)):

     

       53
       53
       -
                   self.encode(cbor2.CBORTag(42, value.encode()))

     

       54
       54
       -
               elif isinstance(value, Data):

     

       55
       55
       -
                   self.encode(value.data())

     

       56
       56
       -
       

     

       57
       57
       -
       

     

       58
       58
       -
       def _cborObjectHook(decoder: cbor2.CBORDecoder, value):

     

       59
       59
       -
           return dataHook(value)

     

       60
       60
       -
       

     

       61
       61
       -
       

     

       62
       62
       -
       class CBORDecoder(cbor2.CBORDecoder):

     

       63
       63
       -
           """

     

       64
       64
       -
           Wrapper of cbor2.CBORDecoder.

     

       65
       65
       -
           """

     

       66
       66
       -
       

     

       67
       67
       -
           def __init__(

     

       68
       68
       -
               self,

     

       69
       69
       -
               fp: typing.IO[bytes],

     

       70
       70
       -
               tag_hook: (

     

       71
       71
       -
                   typing.Callable[[cbor2.CBORDecoder, cbor2.CBORTag], typing.Any] | None

     

       72
       72
       -
               ) = tagHook,

     

       73
       73
       -
               object_hook: (

     

       74
       74
       -
                   typing.Callable[

     

       75
       75
       -
                       [cbor2.CBORDecoder, dict[typing.Any, typing.Any]], typing.Any

     

       76
       76
       -
                   ]

     

       77
       77
       -
                   | None

     

       78
       78
       -
               ) = _cborObjectHook,

     

       79
       79
       -
               str_errors: typing.Literal["strict", "error", "replace"] = "strict",

     

       80
       80
       -
           ):

     

       81
       81
       -
               super().__init__(fp, tag_hook, object_hook, str_errors)

     

       82
       82
       -
       

     

       83
       83
       -
       

     

       84
       84
       -
       # Make things for CBOR again.

     

       85
       85
       -
       

     

       86
       86
       -
       from io import BytesIO

     

       87
       87
       -
       

     

       88
       88
       -
       

     

       89
       89
       -
       def dumps(

     

       90
       90
       -
           obj: object,

     

       91
       91
       -
           datetime_as_timestamp: bool = False,

     

       92
       92
       -
           timezone: tzinfo | None = None,

     

       93
       93
       -
           value_sharing: bool = False,

     

       94
       94
       -
           default: typing.Callable[[cbor2.CBOREncoder, typing.Any], typing.Any] | None = None,

     

       95
       95
       -
           canonical: bool = False,

     

       96
       96
       -
           date_as_datetime: bool = False,

     

       97
       97
       -
           string_referencing: bool = False,

     

       98
       98
       -
           indefinite_containers: bool = False,

     

       99
       99
       -
       ) -> bytes:

     

       100
       100
       -
           with BytesIO() as fp:

     

       101
       101
       -
               CBOREncoder(

     

       102
       102
       -
                   fp,

     

       103
       103
       -
                   datetime_as_timestamp=datetime_as_timestamp,

     

       104
       104
       -
                   timezone=timezone,

     

       105
       105
       -
                   value_sharing=value_sharing,

     

       106
       106
       -
                   default=default,

     

       107
       107
       -
                   canonical=canonical,

     

       108
       108
       -
                   date_as_datetime=date_as_datetime,

     

       109
       109
       -
                   string_referencing=string_referencing,

     

       110
       110
       -
                   indefinite_containers=indefinite_containers,

     

       111
       111
       -
               ).encode(obj)

     

       112
       112
       -
               return fp.getvalue()

     

       113
       113
       -
       

     

       114
       114
       -
       

     

       115
       115
       -
       def dump(

     

       116
       116
       -
           obj: object,

     

       117
       117
       -
           fp: typing.IO[bytes],

     

       118
       118
       -
           datetime_as_timestamp: bool = False,

     

       119
       119
       -
           timezone: tzinfo | None = None,

     

       120
       120
       -
           value_sharing: bool = False,

     

       121
       121
       -
           default: typing.Callable[[cbor2.CBOREncoder, typing.Any], typing.Any] | None = None,

     

       122
       122
       -
           canonical: bool = False,

     

       123
       123
       -
           date_as_datetime: bool = False,

     

       124
       124
       -
           string_referencing: bool = False,

     

       125
       125
       -
           indefinite_containers: bool = False,

     

       126
       126
       -
       ) -> None:

     

       127
       127
       -
           CBOREncoder(

     

       128
       128
       -
               fp,

     

       129
       129
       -
               datetime_as_timestamp=datetime_as_timestamp,

     

       130
       130
       -
               timezone=timezone,

     

       131
       131
       -
               value_sharing=value_sharing,

     

       132
       132
       -
               default=default,

     

       133
       133
       -
               canonical=canonical,

     

       134
       134
       -
               date_as_datetime=date_as_datetime,

     

       135
       135
       -
               string_referencing=string_referencing,

     

       136
       136
       -
               indefinite_containers=indefinite_containers,

     

       137
       137
       -
           ).encode(obj)

-4

tests/__init__.py

···

       1
       1
       -
       if __name__ != "__main__":

     

       2
       2
       -
           raise Exception("name != main")

     

       3
       3
       -
       

     

       4
       4
       -
       import _strings

-179

tests/_strings.py

···

       1
       1
       -
       from atpasser import did, handle, nsid, rKey, uri

     

       2
       2
       -
       

     

       3
       3
       -
       

     

       4
       4
       -
       testStrings, testMethods = {}, {}

     

       5
       5
       -
       

     

       6
       6
       -
       

     

       7
       7
       -
       testStrings[

     

       8
       8
       -
           "did"

     

       9
       9
       -
       

     

       10
       10
       -
       ] = """did:plc:z72i7hdynmk6r22z27h6tvur

     

       11
       11
       -
       

     

       12
       12
       -
       did:web:blueskyweb.xyz

     

       13
       13
       -
       

     

       14
       14
       -
       did:method:val:two

     

       15
       15
       -
       

     

       16
       16
       -
       did:m:v

     

       17
       17
       -
       

     

       18
       18
       -
       did:method::::val

     

       19
       19
       -
       

     

       20
       20
       -
       did:method:-:_:.

     

       21
       21
       -
       

     

       22
       22
       -
       did:key:zQ3shZc2QzApp2oymGvQbzP8eKheVshBHbU4ZYjeXqwSKEn6N

     

       23
       23
       -
       

     

       24
       24
       -
       did:METHOD:val

     

       25
       25
       -
       

     

       26
       26
       -
       did:m123:val

     

       27
       27
       -
       

     

       28
       28
       -
       DID:method:val

     

       29
       29
       -
       did:method:

     

       30
       30
       -
       

     

       31
       31
       -
       did:method:val/two

     

       32
       32
       -
       

     

       33
       33
       -
       did:method:val?two

     

       34
       34
       -
       

     

       35
       35
       -
       did:method:val#two"""

     

       36
       36
       -
       

     

       37
       37
       -
       testMethods["did"] = did.DID

     

       38
       38
       -
       

     

       39
       39
       -
       

     

       40
       40
       -
       testStrings[

     

       41
       41
       -
           "handle"

     

       42
       42
       -
       

     

       43
       43
       -
       ] = """jay.bsky.social

     

       44
       44
       -
       

     

       45
       45
       -
       8.cn

     

       46
       46
       -
       

     

       47
       47
       -
       name.t--t

     

       48
       48
       -
       

     

       49
       49
       -
       XX.LCS.MIT.EDU

     

       50
       50
       -
       a.co

     

       51
       51
       -
       

     

       52
       52
       -
       xn--notarealidn.com

     

       53
       53
       -
       

     

       54
       54
       -
       xn--fiqa61au8b7zsevnm8ak20mc4a87e.xn--fiqs8s

     

       55
       55
       -
       

     

       56
       56
       -
       xn--ls8h.test

     

       57
       57
       -
       example.t

     

       58
       58
       -
       

     

       59
       59
       -
       jo@hn.test

     

       60
       60
       -
       

     

       61
       61
       -
       💩.tes

     

       62
       62
       -
       t

     

       63
       63
       -
       john..test

     

       64
       64
       -
       

     

       65
       65
       -
       xn--bcher-.tld

     

       66
       66
       -
       

     

       67
       67
       -
       john.0

     

       68
       68
       -
       

     

       69
       69
       -
       cn.8

     

       70
       70
       -
       

     

       71
       71
       -
       www.masełkowski.pl.com

     

       72
       72
       -
       

     

       73
       73
       -
       org

     

       74
       74
       -
       

     

       75
       75
       -
       name.org.

     

       76
       76
       -
       

     

       77
       77
       -
       2gzyxa5ihm7nsggfxnu52rck2vv4rvmdlkiu3zzui5du4xyclen53wid.onion

     

       78
       78
       -
       laptop.local

     

       79
       79
       -
       

     

       80
       80
       -
       blah.arpa"""

     

       81
       81
       -
       

     

       82
       82
       -
       testMethods["handle"] = handle.Handle

     

       83
       83
       -
       

     

       84
       84
       -
       

     

       85
       85
       -
       testStrings[

     

       86
       86
       -
           "nsid"

     

       87
       87
       -
       

     

       88
       88
       -
       ] = """com.example.fooBar

     

       89
       89
       -
       

     

       90
       90
       -
       net.users.bob.ping

     

       91
       91
       -
       

     

       92
       92
       -
       a-0.b-1.c

     

       93
       93
       -
       

     

       94
       94
       -
       a.b.c

     

       95
       95
       -
       

     

       96
       96
       -
       com.example.fooBarV2

     

       97
       97
       -
       

     

       98
       98
       -
       cn.8.lex.stuff

     

       99
       99
       -
       

     

       100
       100
       -
       com.exa💩ple.thin

     

       101
       101
       -
       com.example

     

       102
       102
       -
       

     

       103
       103
       -
       com.example.3"""

     

       104
       104
       -
       

     

       105
       105
       -
       testMethods["nsid"] = nsid.NSID

     

       106
       106
       -
       

     

       107
       107
       -
       

     

       108
       108
       -
       testStrings[

     

       109
       109
       -
       

     

       110
       110
       -
           "rkey"

     

       111
       111
       -
       

     

       112
       112
       -
       ] = """3jui7kd54zh2y

     

       113
       113
       -
       self

     

       114
       114
       -
       example.com

     

       115
       115
       -
       

     

       116
       116
       -
       ~1.2-3_

     

       117
       117
       -
       

     

       118
       118
       -
       dHJ1ZQ

     

       119
       119
       -
       pre:fix

     

       120
       120
       -
       

     

       121
       121
       -
       _

     

       122
       122
       -
       

     

       123
       123
       -
       alpha/beta

     

       124
       124
       -
       .

     

       125
       125
       -
       ..

     

       126
       126
       -
       

     

       127
       127
       -
       #extra

     

       128
       128
       -
       

     

       129
       129
       -
       @handle

     

       130
       130
       -
       

     

       131
       131
       -
       any space

     

       132
       132
       -
       

     

       133
       133
       -
       any+space

     

       134
       134
       -
       

     

       135
       135
       -
       number[3]

     

       136
       136
       -
       

     

       137
       137
       -
       number(3)

     

       138
       138
       -
       

     

       139
       139
       -
       "quote"

     

       140
       140
       -
       

     

       141
       141
       -
       dHJ1ZQ=="""

     

       142
       142
       -
       

     

       143
       143
       -
       testMethods["rkey"] = rKey.RKey

     

       144
       144
       -
       

     

       145
       145
       -
       

     

       146
       146
       -
       testStrings[

     

       147
       147
       -
           "uri"

     

       148
       148
       -
       

     

       149
       149
       -
       ] = """at://foo.com/com.example.foo/123

     

       150
       150
       -
       

     

       151
       151
       -
       at://foo.com/example/123

     

       152
       152
       -
       

     

       153
       153
       -
       at://computer

     

       154
       154
       -
       

     

       155
       155
       -
       at://example.com:3000

     

       156
       156
       -
       

     

       157
       157
       -
       at://foo.com/

     

       158
       158
       -
       

     

       159
       159
       -
       at://user:pass@foo.com"""

     

       160
       160
       -
       

     

       161
       161
       -
       testMethods["uri"] = uri.URI

     

       162
       162
       -
       

     

       163
       163
       -
       

     

       164
       164
       -
       for item in testMethods:

     

       165
       165
       -
       

     

       166
       166
       -
           print(f"START TEST {item}")

     

       167
       167
       -
       

     

       168
       168
       -
           for value in testStrings[item].splitlines():

     

       169
       169
       -
       

     

       170
       170
       -
               print(f"Value: {value}")

     

       171
       171
       -
       

     

       172
       172
       -
               try:

     

       173
       173
       -
       

     

       174
       174
       -
                   print(f"str(): {str(testMethods[item](value))}")

     

       175
       175
       -
       

     

       176
       176
       -
               except Exception as e:

     

       177
       177
       -
       

     

       178
       178
       -
                   print(f"× {e}")

     

       179
       179
       -

+41

src/atpasser/model/typed.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       from pydantic import field_serializer

     

       3
       3
       +
       from .base import DataModel

     

       4
       4
       +
       

     

       5
       5
       +
       class TypedDataModel(DataModel):

     

       6
       6
       +
           """

     

       7
       7
       +
           Model for AT Protocol data with type information.

     

       8
       8
       +
       

     

       9
       9
       +
           Includes support for $type field that specifies Lexicon schema.

     

       10
       10
       +
           """

     

       11
       11
       +
       

     

       12
       12
       +
           type: str | None = None

     

       13
       13
       +
           """Lexicon schema type identifier"""

     

       14
       14
       +
       

     

       15
       15
       +
           def __init__(self, **data: Any) -> None:

     

       16
       16
       +
               """

     

       17
       17
       +
               Initialize typed data model with automatic $type handling.

     

       18
       18
       +
       

     

       19
       19
       +
               Args:

     

       20
       20
       +
                   **data: Data including optional $type field

     

       21
       21
       +
               """

     

       22
       22
       +
               # Extract $type if present

     

       23
       23
       +
               dataType = data.pop("$type", None)

     

       24
       24
       +
               if dataType:

     

       25
       25
       +
                   data["type"] = dataType

     

       26
       26
       +
               super().__init__(**data)

     

       27
       27
       +
       

     

       28
       28
       +
           @field_serializer("type")

     

       29
       29
       +
           def serializeType(self, v: str | None) -> dict[str, str] | None:

     

       30
       30
       +
               """

     

       31
       31
       +
               Serialize type field to $type object.

     

       32
       32
       +
       

     

       33
       33
       +
               Args:

     

       34
       34
       +
                   v: Type value to serialize

     

       35
       35
       +
       

     

       36
       36
       +
               Returns:

     

       37
       37
       +
                   $type object if type is not None

     

       38
       38
       +
               """

     

       39
       39
       +
               if v is not None:

     

       40
       40
       +
                   return {"$type": v}

     

       41
       41
       +
               return None

+36

src/atpasser/model/exceptions.py

···

       1
       1
       +
       class AtprotoModelError(Exception):

     

       2
       2
       +
           """Base exception for all AT Protocol model errors"""

     

       3
       3
       +
           pass

     

       4
       4
       +
       

     

       5
       5
       +
       class ValidationError(AtprotoModelError):

     

       6
       6
       +
           """Raised when data validation fails"""

     

       7
       7
       +
           def __init__(self, field: str, message: str):

     

       8
       8
       +
               self.field = field

     

       9
       9
       +
               self.message = message

     

       10
       10
       +
               super().__init__(f"Validation error for field '{field}': {message}")

     

       11
       11
       +
       

     

       12
       12
       +
       class SerializationError(AtprotoModelError):

     

       13
       13
       +
           """Raised when data serialization fails"""

     

       14
       14
       +
           def __init__(self, field: str, message: str):

     

       15
       15
       +
               self.field = field

     

       16
       16
       +
               self.message = message

     

       17
       17
       +
               super().__init__(f"Serialization error for field '{field}': {message}")

     

       18
       18
       +
       

     

       19
       19
       +
       class DeserializationError(AtprotoModelError):

     

       20
       20
       +
           """Raised when data deserialization fails"""

     

       21
       21
       +
           def __init__(self, field: str, message: str):

     

       22
       22
       +
               self.field = field

     

       23
       23
       +
               self.message = message

     

       24
       24
       +
               super().__init__(f"Deserialization error for field '{field}': {message}")

     

       25
       25
       +
       

     

       26
       26
       +
       class InvalidCIDError(AtprotoModelError):

     

       27
       27
       +
           """Raised when CID validation fails"""

     

       28
       28
       +
           pass

     

       29
       29
       +
       

     

       30
       30
       +
       class InvalidBlobError(AtprotoModelError):

     

       31
       31
       +
           """Raised when blob validation fails"""

     

       32
       32
       +
           pass

     

       33
       33
       +
       

     

       34
       34
       +
       class TypeMismatchError(AtprotoModelError):

     

       35
       35
       +
           """Raised when type validation fails"""

     

       36
       36
       +
           pass

src/atpasser/model/__init__.py

···

       38
       38
        
           ProcedureModel,

     

       39
       39
        
           SubscriptionModel

     

       40
       40
        
       )

     

       41
       41
       +
       from .converter import LexiconConverter

     

       41
       42
        
       

     

       42
       43
        
       __all__ = [

     

       43
       44
        
           "DataModel",

     
···

       61
       62
        
           "QueryModel",

     

       62
       63
        
           "ProcedureModel",

     

       63
       64
        
           "SubscriptionModel",

     

       65
       65
       +
           # Converter

     

       66
       66
       +
           "LexiconConverter",

     

       64
       67
        
           # Exceptions

     

       65
       68
        
           "AtprotoModelError",

     

       66
       69
        
           "ValidationError",

-5

src/atpasser/model/base.py

···

       1
       1
        
       import base64

     

       2
       2
       -
       import re

     

       3
       3
       -
       from datetime import datetime

     

       4
       2
        
       from typing import Any

     

       5
       5
       -
       from collections.abc import Mapping

     

       6
       3
        
       from cid.cid import CIDv1, make_cid

     

       7
       4
        
       from pydantic import BaseModel, field_serializer, field_validator, ConfigDict

     

       8
       8
       -
       from pydantic.fields import FieldInfo

     

       9
       5
        
       from .exceptions import (

     

       10
       10
       -
           ValidationError,

     

       11
       6
        
           SerializationError,

     

       12
       7
        
           DeserializationError,

     

       13
       8
        
           InvalidCIDError

+5 -5

src/atpasser/model/blob.py

···

       1
       1
        
       from typing import Any

     

       2
       2
        
       from pydantic import field_validator, ConfigDict

     

       3
       3
        
       from .base import DataModel

     

       4
       4
       -
       from .exceptions import ValidationError, InvalidBlobError

     

       4
       4
       +
       from .exceptions import ValidationError

     

       5
       5
        
       

     

       6
       6
        
       class BlobModel(DataModel):

     

       7
       7
        
           """

     
···

       34
       34
        
                   Validated size

     

       35
       35
        
       

     

       36
       36
        
               Raises:

     

       37
       37
       -
                   ValueError: If size is not positive

     

       37
       37
       +
                   ValidationError: If size is not positive

     

       38
       38
        
               """

     

       39
       39
        
               if v <= 0:

     

       40
       40
       -
                   raise ValueError("Blob size must be positive and non-zero")

     

       40
       40
       +
                   raise ValidationError(field="size", message="must be positive and non-zero")

     

       41
       41
        
               return v

     

       42
       42
        
       

     

       43
       43
        
           @field_validator("mimeType")

     
···

       52
       52
        
                   Validated MIME type

     

       53
       53
        
       

     

       54
       54
        
               Raises:

     

       55
       55
       -
                   ValueError: If MIME type is empty

     

       55
       55
       +
                   ValidationError: If MIME type is empty

     

       56
       56
        
               """

     

       57
       57
        
               if not v:

     

       58
       58
       -
                   raise ValueError("MIME type cannot be empty")

     

       58
       58
       +
                   raise ValidationError(field="mimeType", message="cannot be empty")

     

       59
       59
        
               return v

src/atpasser/model/converter.py

···

       15
       15
        
           UnknownModel, RecordModel, QueryModel,

     

       16
       16
        
           ProcedureModel, SubscriptionModel

     

       17
       17
        
       )

     

       18
       18
       +
       from .types.binary import BytesModel, CidLinkModel

     

       19
       19
       +
       from .blob import BlobModel

     

       18
       20
        
       

     

       19
       21
        
       class LexiconConverter:

     

       20
       22
        
           """

     
···

       31
       33
        
               "integer": IntegerModel,

     

       32
       34
        
               "string": StringModel,

     

       33
       35
        
               

     

       36
       36
       +
               # Binary types

     

       37
       37
       +
               "bytes": BytesModel,

     

       38
       38
       +
               "cid-link": CidLinkModel,

     

       39
       39
       +
               "blob": BlobModel,

     

       40
       40
       +
               

     

       34
       41
        
               # Complex types

     

       35
       42
        
               "array": ArrayModel,

     

       36
       43
        
               "object": ObjectModel,

+214

src/atpasser/model/types/complex.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       from pydantic import field_validator

     

       3
       3
       +
       from ..base import DataModel

     

       4
       4
       +
       

     

       5
       5
       +
       class ArrayModel(DataModel):

     

       6
       6
       +
           """

     

       7
       7
       +
           Model for AT Protocol array type.

     

       8
       8
       +
           

     

       9
       9
       +
           Represents an array of elements with support for item schema definition,

     

       10
       10
       +
           minimum/maximum length constraints as specified in Lexicon.

     

       11
       11
       +
           """

     

       12
       12
       +
           

     

       13
       13
       +
           items: Any

     

       14
       14
       +
           """Schema definition for array elements"""

     

       15
       15
       +
           

     

       16
       16
       +
           minLength: int | None = None

     

       17
       17
       +
           """Minimum number of elements"""

     

       18
       18
       +
           

     

       19
       19
       +
           maxLength: int | None = None

     

       20
       20
       +
           """Maximum number of elements"""

     

       21
       21
       +
           

     

       22
       22
       +
           value: list[Any]

     

       23
       23
       +
           """Array values"""

     

       24
       24
       +
           

     

       25
       25
       +
           def __init__(self, **data: Any) -> None:

     

       26
       26
       +
               """

     

       27
       27
       +
               Initialize array model with validation.

     

       28
       28
       +
               

     

       29
       29
       +
               Args:

     

       30
       30
       +
                   **data: Input data containing array values

     

       31
       31
       +
                   

     

       32
       32
       +
               Raises:

     

       33
       33
       +
                   ValueError: If array violates constraints

     

       34
       34
       +
               """

     

       35
       35
       +
               super().__init__(**data)

     

       36
       36
       +
               

     

       37
       37
       +
           @field_validator("value", mode="before")

     

       38
       38
       +
           def validate_array(cls, v: Any) -> list[Any]:

     

       39
       39
       +
               """

     

       40
       40
       +
               Validate array structure and elements.

     

       41
       41
       +
               

     

       42
       42
       +
               Args:

     

       43
       43
       +
                   v: Value to validate

     

       44
       44
       +
                   

     

       45
       45
       +
               Returns:

     

       46
       46
       +
                   Validated array

     

       47
       47
       +
                   

     

       48
       48
       +
               Raises:

     

       49
       49
       +
                   ValueError: If array violates constraints

     

       50
       50
       +
               """

     

       51
       51
       +
               if not isinstance(v, list):

     

       52
       52
       +
                   raise ValueError("Value must be an array")

     

       53
       53
       +
                   

     

       54
       54
       +
               # Validate length constraints

     

       55
       55
       +
               if cls.minLength is not None and len(v) < cls.minLength:

     

       56
       56
       +
                   raise ValueError(f"Array must have at least {cls.minLength} items")

     

       57
       57
       +
                   

     

       58
       58
       +
               if cls.maxLength is not None and len(v) > cls.maxLength:

     

       59
       59
       +
                   raise ValueError(f"Array must have at most {cls.maxLength} items")

     

       60
       60
       +
                   

     

       61
       61
       +
               return v

     

       62
       62
       +
       

     

       63
       63
       +
       class ObjectModel(DataModel):

     

       64
       64
       +
           """

     

       65
       65
       +
           Model for AT Protocol object type.

     

       66
       66
       +
           

     

       67
       67
       +
           Represents a generic object schema with properties definitions,

     

       68
       68
       +
           required fields and nullable fields as specified in Lexicon.

     

       69
       69
       +
           """

     

       70
       70
       +
           

     

       71
       71
       +
           properties: dict[str, Any]

     

       72
       72
       +
           """Map of property names to their schema definitions"""

     

       73
       73
       +
           

     

       74
       74
       +
           required: list[str] | None = None

     

       75
       75
       +
           """List of required property names"""

     

       76
       76
       +
           

     

       77
       77
       +
           nullable: list[str] | None = None

     

       78
       78
       +
           """List of properties that can be null"""

     

       79
       79
       +
           

     

       80
       80
       +
           value: dict[str, Any]

     

       81
       81
       +
           """Object property values"""

     

       82
       82
       +
           

     

       83
       83
       +
           def __init__(self, **data: Any) -> None:

     

       84
       84
       +
               """

     

       85
       85
       +
               Initialize object model with validation.

     

       86
       86
       +
               

     

       87
       87
       +
               Args:

     

       88
       88
       +
                   **data: Input data containing object properties

     

       89
       89
       +
                   

     

       90
       90
       +
               Raises:

     

       91
       91
       +
                   ValueError: If object violates constraints

     

       92
       92
       +
               """

     

       93
       93
       +
               super().__init__(**data)

     

       94
       94
       +
               

     

       95
       95
       +
           @field_validator("value", mode="before")

     

       96
       96
       +
           def validate_object(cls, v: Any) -> dict[str, Any]:

     

       97
       97
       +
               """

     

       98
       98
       +
               Validate object structure and properties.

     

       99
       99
       +
               

     

       100
       100
       +
               Args:

     

       101
       101
       +
                   v: Value to validate

     

       102
       102
       +
                   

     

       103
       103
       +
               Returns:

     

       104
       104
       +
                   Validated object

     

       105
       105
       +
                   

     

       106
       106
       +
               Raises:

     

       107
       107
       +
                   ValueError: If object violates constraints

     

       108
       108
       +
               """

     

       109
       109
       +
               if not isinstance(v, dict):

     

       110
       110
       +
                   raise ValueError("Value must be an object")

     

       111
       111
       +
                   

     

       112
       112
       +
               # Validate required fields

     

       113
       113
       +
               if cls.required:

     

       114
       114
       +
                   for field in cls.required:

     

       115
       115
       +
                       if field not in v:

     

       116
       116
       +
                           raise ValueError(f"Missing required field: {field}")

     

       117
       117
       +
                           

     

       118
       118
       +
               # Validate nullable fields

     

       119
       119
       +
               if cls.nullable:

     

       120
       120
       +
                   for field, value in v.items():

     

       121
       121
       +
                       if field not in cls.nullable and value is None:

     

       122
       122
       +
                           raise ValueError(f"Field {field} cannot be null")

     

       123
       123
       +
                           

     

       124
       124
       +
               return v

     

       125
       125
       +
       

     

       126
       126
       +
       class ParamsModel(DataModel):

     

       127
       127
       +
           """

     

       128
       128
       +
           Model for AT Protocol params type.

     

       129
       129
       +
           

     

       130
       130
       +
           Specialized for HTTP query parameters with support for boolean,

     

       131
       131
       +
           integer, string and unknown types as specified in Lexicon.

     

       132
       132
       +
           """

     

       133
       133
       +
           

     

       134
       134
       +
           required: list[str] | None = None

     

       135
       135
       +
           """List of required parameter names"""

     

       136
       136
       +
           

     

       137
       137
       +
           properties: dict[str, Any]

     

       138
       138
       +
           """Map of parameter names to their schema definitions"""

     

       139
       139
       +
           

     

       140
       140
       +
           value: dict[str, Any]

     

       141
       141
       +
           """Parameter values

     

       142
       142
       +
           

     

       143
       143
       +
           Supported types:

     

       144
       144
       +
           - boolean

     

       145
       145
       +
           - integer

     

       146
       146
       +
           - string

     

       147
       147
       +
           - array (of boolean/integer/string/unknown)

     

       148
       148
       +
           - unknown (object)

     

       149
       149
       +
           """

     

       150
       150
       +
           

     

       151
       151
       +
           def __init__(self, **data: Any) -> None:

     

       152
       152
       +
               """

     

       153
       153
       +
               Initialize params model with validation.

     

       154
       154
       +
               

     

       155
       155
       +
               Args:

     

       156
       156
       +
                   **data: Input data containing parameter values

     

       157
       157
       +
                   

     

       158
       158
       +
               Raises:

     

       159
       159
       +
                   ValueError: If parameters violate constraints

     

       160
       160
       +
               """

     

       161
       161
       +
               super().__init__(**data)

     

       162
       162
       +
               

     

       163
       163
       +
           @field_validator("value", mode="before")

     

       164
       164
       +
           def validate_params(cls, v: Any) -> dict[str, Any]:

     

       165
       165
       +
               """

     

       166
       166
       +
               Validate parameters structure and values.

     

       167
       167
       +
               

     

       168
       168
       +
               Args:

     

       169
       169
       +
                   v: Value to validate

     

       170
       170
       +
                   

     

       171
       171
       +
               Returns:

     

       172
       172
       +
                   Validated parameters

     

       173
       173
       +
                   

     

       174
       174
       +
               Raises:

     

       175
       175
       +
                   ValueError: If parameters violate constraints

     

       176
       176
       +
               """

     

       177
       177
       +
               if not isinstance(v, dict):

     

       178
       178
       +
                   raise ValueError("Value must be a dictionary of parameters")

     

       179
       179
       +
                   

     

       180
       180
       +
               # Validate required parameters

     

       181
       181
       +
               if cls.required:

     

       182
       182
       +
                   for param in cls.required:

     

       183
       183
       +
                       if param not in v:

     

       184
       184
       +
                           raise ValueError(f"Missing required parameter: {param}")

     

       185
       185
       +
                           

     

       186
       186
       +
               # Validate parameter types

     

       187
       187
       +
               for param, value in v.items():

     

       188
       188
       +
                   if param in cls.properties:

     

       189
       189
       +
                       prop_type = cls.properties[param].get("type")

     

       190
       190
       +
                       if prop_type == "boolean" and not isinstance(value, bool):

     

       191
       191
       +
                           raise ValueError(f"Parameter {param} must be boolean")

     

       192
       192
       +
                       elif prop_type == "integer" and not isinstance(value, int):

     

       193
       193
       +
                           raise ValueError(f"Parameter {param} must be integer")

     

       194
       194
       +
                       elif prop_type == "string" and not isinstance(value, str):

     

       195
       195
       +
                           raise ValueError(f"Parameter {param} must be string")

     

       196
       196
       +
                       elif prop_type == "array":

     

       197
       197
       +
                           if not isinstance(value, list):

     

       198
       198
       +
                               raise ValueError(f"Parameter {param} must be array")

     

       199
       199
       +
                           # Validate array items if schema is specified

     

       200
       200
       +
                           if "items" in cls.properties[param]:

     

       201
       201
       +
                               item_type = cls.properties[param]["items"].get("type")

     

       202
       202
       +
                               for item in value:

     

       203
       203
       +
                                   if item_type == "boolean" and not isinstance(item, bool):

     

       204
       204
       +
                                       raise ValueError(f"Array item in {param} must be boolean")

     

       205
       205
       +
                                   elif item_type == "integer" and not isinstance(item, int):

     

       206
       206
       +
                                       raise ValueError(f"Array item in {param} must be integer")

     

       207
       207
       +
                                   elif item_type == "string" and not isinstance(item, str):

     

       208
       208
       +
                                       raise ValueError(f"Array item in {param} must be string")

     

       209
       209
       +
                                   elif item_type == "unknown" and not isinstance(item, dict):

     

       210
       210
       +
                                       raise ValueError(f"Array item in {param} must be object")

     

       211
       211
       +
                       elif prop_type == "unknown" and not isinstance(value, dict):

     

       212
       212
       +
                           raise ValueError(f"Parameter {param} must be object")

     

       213
       213
       +
                           

     

       214
       214
       +
               return v

+172

src/atpasser/model/types/primitive.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       from pydantic import field_validator

     

       3
       3
       +
       from ..base import DataModel

     

       4
       4
       +
       

     

       5
       5
       +
       class NullModel(DataModel):

     

       6
       6
       +
           """

     

       7
       7
       +
           Model for AT Protocol null type.

     

       8
       8
       +
           

     

       9
       9
       +
           Represents a null value in AT Protocol data model. This model ensures proper

     

       10
       10
       +
           serialization and validation of null values according to Lexicon specification.

     

       11
       11
       +
           """

     

       12
       12
       +
           

     

       13
       13
       +
           value: None = None

     

       14
       14
       +
           """Always None for null type"""

     

       15
       15
       +
           

     

       16
       16
       +
           def __init__(self, **data: Any) -> None:

     

       17
       17
       +
               """

     

       18
       18
       +
               Initialize null model with validation.

     

       19
       19
       +
               

     

       20
       20
       +
               Args:

     

       21
       21
       +
                   **data: Input data (must be empty or contain only None values)

     

       22
       22
       +
                   

     

       23
       23
       +
               Raises:

     

       24
       24
       +
                   ValueError: If non-null value is provided

     

       25
       25
       +
               """

     

       26
       26
       +
               if data and any(v is not None for v in data.values()):

     

       27
       27
       +
                   raise ValueError("NullModel only accepts None values")

     

       28
       28
       +
               super().__init__(**data)

     

       29
       29
       +
           

     

       30
       30
       +
           @field_validator("*", mode="before")

     

       31
       31
       +
           def validate_null(cls, v: Any) -> None:

     

       32
       32
       +
               """

     

       33
       33
       +
               Validate that value is null.

     

       34
       34
       +
               

     

       35
       35
       +
               Args:

     

       36
       36
       +
                   v: Value to validate

     

       37
       37
       +
                   

     

       38
       38
       +
               Returns:

     

       39
       39
       +
                   None if validation succeeds

     

       40
       40
       +
                   

     

       41
       41
       +
               Raises:

     

       42
       42
       +
                   ValueError: If value is not null

     

       43
       43
       +
               """

     

       44
       44
       +
               if v is not None:

     

       45
       45
       +
                   raise ValueError("NullModel only accepts None values")

     

       46
       46
       +
               return None

     

       47
       47
       +
       

     

       48
       48
       +
       class BooleanModel(DataModel):

     

       49
       49
       +
           """

     

       50
       50
       +
           Model for AT Protocol boolean type.

     

       51
       51
       +
           

     

       52
       52
       +
           Represents a boolean value in AT Protocol data model with support for

     

       53
       53
       +
           default values and constants as specified in Lexicon.

     

       54
       54
       +
           """

     

       55
       55
       +
           

     

       56
       56
       +
           value: bool

     

       57
       57
       +
           """Boolean value"""

     

       58
       58
       +
           

     

       59
       59
       +
           default: bool | None = None

     

       60
       60
       +
           """Default value if not provided"""

     

       61
       61
       +
           

     

       62
       62
       +
           const: bool | None = None

     

       63
       63
       +
           """Fixed constant value if specified"""

     

       64
       64
       +
           

     

       65
       65
       +
           def __init__(self, **data: Any) -> None:

     

       66
       66
       +
               """

     

       67
       67
       +
               Initialize boolean model with validation.

     

       68
       68
       +
               

     

       69
       69
       +
               Args:

     

       70
       70
       +
                   **data: Input data containing boolean value

     

       71
       71
       +
                   

     

       72
       72
       +
               Raises:

     

       73
       73
       +
                   ValueError: If value doesn't match const or is not boolean

     

       74
       74
       +
               """

     

       75
       75
       +
               super().__init__(**data)

     

       76
       76
       +
               if self.const is not None and self.value != self.const:

     

       77
       77
       +
                   raise ValueError(f"Boolean value must be {self.const}")

     

       78
       78
       +
           

     

       79
       79
       +
           @field_validator("value", mode="before")

     

       80
       80
       +
           def validate_boolean(cls, v: Any) -> bool:

     

       81
       81
       +
               """

     

       82
       82
       +
               Validate and convert input to boolean.

     

       83
       83
       +
               

     

       84
       84
       +
               Args:

     

       85
       85
       +
                   v: Value to validate

     

       86
       86
       +
                   

     

       87
       87
       +
               Returns:

     

       88
       88
       +
                   Validated boolean value

     

       89
       89
       +
                   

     

       90
       90
       +
               Raises:

     

       91
       91
       +
                   ValueError: If value cannot be converted to boolean

     

       92
       92
       +
               """

     

       93
       93
       +
               if isinstance(v, bool):

     

       94
       94
       +
                   return v

     

       95
       95
       +
               if isinstance(v, str):

     

       96
       96
       +
                   if v.lower() in ("true", "1"):

     

       97
       97
       +
                       return True

     

       98
       98
       +
                   if v.lower() in ("false", "0"):

     

       99
       99
       +
                       return False

     

       100
       100
       +
               raise ValueError("Value must be a boolean")

     

       101
       101
       +
       

     

       102
       102
       +
       class IntegerModel(DataModel):

     

       103
       103
       +
           """

     

       104
       104
       +
           Model for AT Protocol integer type.

     

       105
       105
       +
           

     

       106
       106
       +
           Represents a signed integer number with support for minimum/maximum values,

     

       107
       107
       +
           enumeration sets, default values and constraints as specified in Lexicon.

     

       108
       108
       +
           """

     

       109
       109
       +
           

     

       110
       110
       +
           value: int

     

       111
       111
       +
           """Integer value"""

     

       112
       112
       +
           

     

       113
       113
       +
           minimum: int | None = None

     

       114
       114
       +
           """Minimum acceptable value"""

     

       115
       115
       +
           

     

       116
       116
       +
           maximum: int | None = None

     

       117
       117
       +
           """Maximum acceptable value"""

     

       118
       118
       +
           

     

       119
       119
       +
           enum: list[int] | None = None

     

       120
       120
       +
           """Closed set of allowed values"""

     

       121
       121
       +
           

     

       122
       122
       +
           default: int | None = None

     

       123
       123
       +
           """Default value if not provided"""

     

       124
       124
       +
           

     

       125
       125
       +
           const: int | None = None

     

       126
       126
       +
           """Fixed constant value if specified"""

     

       127
       127
       +
           

     

       128
       128
       +
           def __init__(self, **data: Any) -> None:

     

       129
       129
       +
               """

     

       130
       130
       +
               Initialize integer model with validation.

     

       131
       131
       +
               

     

       132
       132
       +
               Args:

     

       133
       133
       +
                   **data: Input data containing integer value

     

       134
       134
       +
                   

     

       135
       135
       +
               Raises:

     

       136
       136
       +
                   ValueError: If value violates constraints

     

       137
       137
       +
               """

     

       138
       138
       +
               super().__init__(**data)

     

       139
       139
       +
               if self.const is not None and self.value != self.const:

     

       140
       140
       +
                   raise ValueError(f"Integer value must be {self.const}")

     

       141
       141
       +
           

     

       142
       142
       +
           @field_validator("value", mode="before")

     

       143
       143
       +
           def validate_integer(cls, v: Any) -> int:

     

       144
       144
       +
               """

     

       145
       145
       +
               Validate and convert input to integer.

     

       146
       146
       +
               

     

       147
       147
       +
               Args:

     

       148
       148
       +
                   v: Value to validate

     

       149
       149
       +
                   

     

       150
       150
       +
               Returns:

     

       151
       151
       +
                   Validated integer value

     

       152
       152
       +
                   

     

       153
       153
       +
               Raises:

     

       154
       154
       +
                   ValueError: If value violates constraints

     

       155
       155
       +
               """

     

       156
       156
       +
               if not isinstance(v, int):

     

       157
       157
       +
                   try:

     

       158
       158
       +
                       v = int(v)

     

       159
       159
       +
                   except (TypeError, ValueError):

     

       160
       160
       +
                       raise ValueError("Value must be an integer")

     

       161
       161
       +
               

     

       162
       162
       +
               # Validate against instance attributes

     

       163
       163
       +
               if cls.enum and v not in cls.enum:

     

       164
       164
       +
                   raise ValueError(f"Value must be one of {cls.enum}")

     

       165
       165
       +
                   

     

       166
       166
       +
               if cls.minimum is not None and v < cls.minimum:

     

       167
       167
       +
                   raise ValueError(f"Value must be >= {cls.minimum}")

     

       168
       168
       +
                   

     

       169
       169
       +
               if cls.maximum is not None and v > cls.maximum:

     

       170
       170
       +
                   raise ValueError(f"Value must be <= {cls.maximum}")

     

       171
       171
       +
                   

     

       172
       172
       +
               return v

+131

src/atpasser/model/types/reference.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       from pydantic import field_validator

     

       3
       3
       +
       from ..base import DataModel

     

       4
       4
       +
       

     

       5
       5
       +
       class TokenModel(DataModel):

     

       6
       6
       +
           """

     

       7
       7
       +
           Model for AT Protocol token type.

     

       8
       8
       +
           

     

       9
       9
       +
           Represents empty data values which exist only to be referenced by name.

     

       10
       10
       +
           Tokens encode as string data with the string being the fully-qualified

     

       11
       11
       +
           reference to the token itself (NSID followed by optional fragment).

     

       12
       12
       +
           """

     

       13
       13
       +
           

     

       14
       14
       +
           name: str

     

       15
       15
       +
           """Token name/identifier"""

     

       16
       16
       +
           

     

       17
       17
       +
           description: str | None = None

     

       18
       18
       +
           """Description clarifying the meaning of the token"""

     

       19
       19
       +
           

     

       20
       20
       +
           def __init__(self, **data: Any) -> None:

     

       21
       21
       +
               """

     

       22
       22
       +
               Initialize token model.

     

       23
       23
       +
               

     

       24
       24
       +
               Args:

     

       25
       25
       +
                   **data: Input data containing token name

     

       26
       26
       +
               """

     

       27
       27
       +
               super().__init__(**data)

     

       28
       28
       +
               

     

       29
       29
       +
           @field_validator("name")

     

       30
       30
       +
           def validate_name(cls, v: str) -> str:

     

       31
       31
       +
               """

     

       32
       32
       +
               Validate token name format.

     

       33
       33
       +
               

     

       34
       34
       +
               Args:

     

       35
       35
       +
                   v: Name to validate

     

       36
       36
       +
                   

     

       37
       37
       +
               Returns:

     

       38
       38
       +
                   Validated name

     

       39
       39
       +
                   

     

       40
       40
       +
               Raises:

     

       41
       41
       +
                   ValueError: If name contains whitespace

     

       42
       42
       +
               """

     

       43
       43
       +
               if any(c.isspace() for c in v):

     

       44
       44
       +
                   raise ValueError("Token name must not contain whitespace")

     

       45
       45
       +
               return v

     

       46
       46
       +
       

     

       47
       47
       +
       class RefModel(DataModel):

     

       48
       48
       +
           """

     

       49
       49
       +
           Model for AT Protocol ref type.

     

       50
       50
       +
           

     

       51
       51
       +
           Represents a reference to another schema definition, either globally

     

       52
       52
       +
           (using NSID) or locally (using #-delimited name).

     

       53
       53
       +
           """

     

       54
       54
       +
           

     

       55
       55
       +
           ref: str

     

       56
       56
       +
           """Reference to schema definition (NSID or #name)"""

     

       57
       57
       +
           

     

       58
       58
       +
           description: str | None = None

     

       59
       59
       +
           """Description of the reference"""

     

       60
       60
       +
           

     

       61
       61
       +
           def __init__(self, **data: Any) -> None:

     

       62
       62
       +
               """

     

       63
       63
       +
               Initialize reference model.

     

       64
       64
       +
               

     

       65
       65
       +
               Args:

     

       66
       66
       +
                   **data: Input data containing reference

     

       67
       67
       +
               """

     

       68
       68
       +
               super().__init__(**data)

     

       69
       69
       +
               

     

       70
       70
       +
           @field_validator("ref")

     

       71
       71
       +
           def validate_ref(cls, v: str) -> str:

     

       72
       72
       +
               """

     

       73
       73
       +
               Validate reference format.

     

       74
       74
       +
               

     

       75
       75
       +
               Args:

     

       76
       76
       +
                   v: Reference to validate

     

       77
       77
       +
                   

     

       78
       78
       +
               Returns:

     

       79
       79
       +
                   Validated reference

     

       80
       80
       +
                   

     

       81
       81
       +
               Raises:

     

       82
       82
       +
                   ValueError: If reference is empty or invalid

     

       83
       83
       +
               """

     

       84
       84
       +
               if not v:

     

       85
       85
       +
                   raise ValueError("Reference cannot be empty")

     

       86
       86
       +
               return v

     

       87
       87
       +
       

     

       88
       88
       +
       class UnionModel(DataModel):

     

       89
       89
       +
           """

     

       90
       90
       +
           Model for AT Protocol union type.

     

       91
       91
       +
           

     

       92
       92
       +
           Represents that multiple possible types could be present at a location.

     

       93
       93
       +
           The references follow the same syntax as `ref`, allowing references to

     

       94
       94
       +
           both global or local schema definitions.

     

       95
       95
       +
           """

     

       96
       96
       +
           

     

       97
       97
       +
           refs: list[str]

     

       98
       98
       +
           """References to schema definitions"""

     

       99
       99
       +
           

     

       100
       100
       +
           closed: bool = False

     

       101
       101
       +
           """Indicates if union is open (can be extended) or closed"""

     

       102
       102
       +
           

     

       103
       103
       +
           description: str | None = None

     

       104
       104
       +
           """Description of the union"""

     

       105
       105
       +
           

     

       106
       106
       +
           def __init__(self, **data: Any) -> None:

     

       107
       107
       +
               """

     

       108
       108
       +
               Initialize union model.

     

       109
       109
       +
               

     

       110
       110
       +
               Args:

     

       111
       111
       +
                   **data: Input data containing union references

     

       112
       112
       +
               """

     

       113
       113
       +
               super().__init__(**data)

     

       114
       114
       +
               

     

       115
       115
       +
           @field_validator("refs")

     

       116
       116
       +
           def validate_refs(cls, v: list[str]) -> list[str]:

     

       117
       117
       +
               """

     

       118
       118
       +
               Validate union references.

     

       119
       119
       +
               

     

       120
       120
       +
               Args:

     

       121
       121
       +
                   v: References to validate

     

       122
       122
       +
                   

     

       123
       123
       +
               Returns:

     

       124
       124
       +
                   Validated references

     

       125
       125
       +
                   

     

       126
       126
       +
               Raises:

     

       127
       127
       +
                   ValueError: If references list is empty for closed union

     

       128
       128
       +
               """

     

       129
       129
       +
               if cls.closed and not v:

     

       130
       130
       +
                   raise ValueError("Closed union must have at least one reference")

     

       131
       131
       +
               return v

+323

src/atpasser/model/types/special.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       from pydantic import field_validator

     

       3
       3
       +
       from ..base import DataModel

     

       4
       4
       +
       

     

       5
       5
       +
       class UnknownModel(DataModel):

     

       6
       6
       +
           """

     

       7
       7
       +
           Model for AT Protocol unknown type.

     

       8
       8
       +
           

     

       9
       9
       +
           Indicates that any data object could appear at this location,

     

       10
       10
       +
           with no specific validation. The top-level data must be an object.

     

       11
       11
       +
           """

     

       12
       12
       +
           

     

       13
       13
       +
           description: str | None = None

     

       14
       14
       +
           """Description of the unknown type usage"""

     

       15
       15
       +
           

     

       16
       16
       +
           def __init__(self, **data: Any) -> None:

     

       17
       17
       +
               """

     

       18
       18
       +
               Initialize unknown model.

     

       19
       19
       +
               

     

       20
       20
       +
               Args:

     

       21
       21
       +
                   **data: Input data containing unknown object

     

       22
       22
       +
               """

     

       23
       23
       +
               super().__init__(**data)

     

       24
       24
       +
               

     

       25
       25
       +
           @field_validator("*", mode="before")

     

       26
       26
       +
           def validate_unknown(cls, v: Any) -> Any:

     

       27
       27
       +
               """

     

       28
       28
       +
               Validate unknown data is an object.

     

       29
       29
       +
               

     

       30
       30
       +
               Args:

     

       31
       31
       +
                   v: Value to validate

     

       32
       32
       +
                   

     

       33
       33
       +
               Returns:

     

       34
       34
       +
                   Validated value

     

       35
       35
       +
                   

     

       36
       36
       +
               Raises:

     

       37
       37
       +
                   ValueError: If value is not an object

     

       38
       38
       +
               """

     

       39
       39
       +
               if not isinstance(v, dict):

     

       40
       40
       +
                   raise ValueError("Unknown type must be an object")

     

       41
       41
       +
               return v

     

       42
       42
       +
       

     

       43
       43
       +
       class RecordModel(DataModel):

     

       44
       44
       +
           """

     

       45
       45
       +
           Model for AT Protocol record type.

     

       46
       46
       +
           

     

       47
       47
       +
           Describes an object that can be stored in a repository record.

     

       48
       48
       +
           Records must include a $type field indicating their schema.

     

       49
       49
       +
           """

     

       50
       50
       +
           

     

       51
       51
       +
           key: str

     

       52
       52
       +
           """Specifies the Record Key type"""

     

       53
       53
       +
           

     

       54
       54
       +
           record: dict[str, Any]

     

       55
       55
       +
           """Schema definition with type 'object'"""

     

       56
       56
       +
           

     

       57
       57
       +
           type: str

     

       58
       58
       +
           """Lexicon schema type identifier"""

     

       59
       59
       +
           

     

       60
       60
       +
           def __init__(self, **data: Any) -> None:

     

       61
       61
       +
               """

     

       62
       62
       +
               Initialize record model with validation.

     

       63
       63
       +
               

     

       64
       64
       +
               Args:

     

       65
       65
       +
                   **data: Input data containing record values

     

       66
       66
       +
                   

     

       67
       67
       +
               Raises:

     

       68
       68
       +
                   ValueError: If record is missing required fields

     

       69
       69
       +
               """

     

       70
       70
       +
               # Extract $type if present

     

       71
       71
       +
               data_type = data.pop("$type", None)

     

       72
       72
       +
               if data_type:

     

       73
       73
       +
                   data["type"] = data_type

     

       74
       74
       +
               super().__init__(**data)

     

       75
       75
       +
               

     

       76
       76
       +
           @field_validator("type")

     

       77
       77
       +
           def validate_type(cls, v: str) -> str:

     

       78
       78
       +
               """

     

       79
       79
       +
               Validate record type field.

     

       80
       80
       +
               

     

       81
       81
       +
               Args:

     

       82
       82
       +
                   v: Type value to validate

     

       83
       83
       +
                   

     

       84
       84
       +
               Returns:

     

       85
       85
       +
                   Validated type

     

       86
       86
       +
                   

     

       87
       87
       +
               Raises:

     

       88
       88
       +
                   ValueError: If type is empty

     

       89
       89
       +
               """

     

       90
       90
       +
               if not v:

     

       91
       91
       +
                   raise ValueError("Record must have a type")

     

       92
       92
       +
               return v

     

       93
       93
       +
               

     

       94
       94
       +
           @field_validator("record", mode="before")

     

       95
       95
       +
           def validate_record(cls, v: Any) -> dict[str, Any]:

     

       96
       96
       +
               """

     

       97
       97
       +
               Validate record structure.

     

       98
       98
       +
               

     

       99
       99
       +
               Args:

     

       100
       100
       +
                   v: Record value to validate

     

       101
       101
       +
                   

     

       102
       102
       +
               Returns:

     

       103
       103
       +
                   Validated record

     

       104
       104
       +
                   

     

       105
       105
       +
               Raises:

     

       106
       106
       +
                   ValueError: If record is not an object

     

       107
       107
       +
               """

     

       108
       108
       +
               if not isinstance(v, dict):

     

       109
       109
       +
                   raise ValueError("Record must be an object")

     

       110
       110
       +
               return v

     

       111
       111
       +
       

     

       112
       112
       +
       class QueryModel(DataModel):

     

       113
       113
       +
           """

     

       114
       114
       +
           Model for AT Protocol query type.

     

       115
       115
       +
           

     

       116
       116
       +
           Describes an XRPC Query endpoint (HTTP GET) with support for

     

       117
       117
       +
           parameters, output schema and error responses.

     

       118
       118
       +
           """

     

       119
       119
       +
           

     

       120
       120
       +
           parameters: dict[str, Any] | None = None

     

       121
       121
       +
           """HTTP query parameters schema"""

     

       122
       122
       +
           

     

       123
       123
       +
           output: dict[str, Any] | None = None

     

       124
       124
       +
           """HTTP response body schema"""

     

       125
       125
       +
           

     

       126
       126
       +
           errors: list[dict[str, str]] | None = None

     

       127
       127
       +
           """Possible error responses"""

     

       128
       128
       +
           

     

       129
       129
       +
           def __init__(self, **data: Any) -> None:

     

       130
       130
       +
               """

     

       131
       131
       +
               Initialize query model with validation.

     

       132
       132
       +
               

     

       133
       133
       +
               Args:

     

       134
       134
       +
                   **data: Input data containing query definition

     

       135
       135
       +
               """

     

       136
       136
       +
               super().__init__(**data)

     

       137
       137
       +
               

     

       138
       138
       +
           @field_validator("output")

     

       139
       139
       +
           def validate_output(cls, v: dict[str, Any] | None) -> dict[str, Any] | None:

     

       140
       140
       +
               """

     

       141
       141
       +
               Validate output schema.

     

       142
       142
       +
               

     

       143
       143
       +
               Args:

     

       144
       144
       +
                   v: Output schema to validate

     

       145
       145
       +
                   

     

       146
       146
       +
               Returns:

     

       147
       147
       +
                   Validated output schema

     

       148
       148
       +
                   

     

       149
       149
       +
               Raises:

     

       150
       150
       +
                   ValueError: If output schema is invalid

     

       151
       151
       +
               """

     

       152
       152
       +
               if v and "encoding" not in v:

     

       153
       153
       +
                   raise ValueError("Output must specify encoding")

     

       154
       154
       +
               return v

     

       155
       155
       +
               

     

       156
       156
       +
           @field_validator("errors")

     

       157
       157
       +
           def validate_errors(cls, v: list[dict[str, str]] | None) -> list[dict[str, str]] | None:

     

       158
       158
       +
               """

     

       159
       159
       +
               Validate error definitions.

     

       160
       160
       +
               

     

       161
       161
       +
               Args:

     

       162
       162
       +
                   v: Error definitions to validate

     

       163
       163
       +
                   

     

       164
       164
       +
               Returns:

     

       165
       165
       +
                   Validated error definitions

     

       166
       166
       +
                   

     

       167
       167
       +
               Raises:

     

       168
       168
       +
                   ValueError: If any error definition is invalid

     

       169
       169
       +
               """

     

       170
       170
       +
               if v:

     

       171
       171
       +
                   for error in v:

     

       172
       172
       +
                       if "name" not in error:

     

       173
       173
       +
                           raise ValueError("Error must have a name")

     

       174
       174
       +
               return v

     

       175
       175
       +
       

     

       176
       176
       +
       class ProcedureModel(DataModel):

     

       177
       177
       +
           """

     

       178
       178
       +
           Model for AT Protocol procedure type.

     

       179
       179
       +
           

     

       180
       180
       +
           Describes an XRPC Procedure endpoint (HTTP POST) with support for

     

       181
       181
       +
           parameters, input/output schemas and error responses.

     

       182
       182
       +
           """

     

       183
       183
       +
           

     

       184
       184
       +
           parameters: dict[str, Any] | None = None

     

       185
       185
       +
           """HTTP query parameters schema"""

     

       186
       186
       +
           

     

       187
       187
       +
           input: dict[str, Any] | None = None

     

       188
       188
       +
           """HTTP request body schema"""

     

       189
       189
       +
           

     

       190
       190
       +
           output: dict[str, Any] | None = None

     

       191
       191
       +
           """HTTP response body schema"""

     

       192
       192
       +
           

     

       193
       193
       +
           errors: list[dict[str, str]] | None = None

     

       194
       194
       +
           """Possible error responses"""

     

       195
       195
       +
           

     

       196
       196
       +
           def __init__(self, **data: Any) -> None:

     

       197
       197
       +
               """

     

       198
       198
       +
               Initialize procedure model with validation.

     

       199
       199
       +
               

     

       200
       200
       +
               Args:

     

       201
       201
       +
                   **data: Input data containing procedure definition

     

       202
       202
       +
               """

     

       203
       203
       +
               super().__init__(**data)

     

       204
       204
       +
               

     

       205
       205
       +
           @field_validator("input")

     

       206
       206
       +
           def validate_input(cls, v: dict[str, Any] | None) -> dict[str, Any] | None:

     

       207
       207
       +
               """

     

       208
       208
       +
               Validate input schema.

     

       209
       209
       +
               

     

       210
       210
       +
               Args:

     

       211
       211
       +
                   v: Input schema to validate

     

       212
       212
       +
                   

     

       213
       213
       +
               Returns:

     

       214
       214
       +
                   Validated input schema

     

       215
       215
       +
                   

     

       216
       216
       +
               Raises:

     

       217
       217
       +
                   ValueError: If input schema is invalid

     

       218
       218
       +
               """

     

       219
       219
       +
               if v and "encoding" not in v:

     

       220
       220
       +
                   raise ValueError("Input must specify encoding")

     

       221
       221
       +
               return v

     

       222
       222
       +
               

     

       223
       223
       +
           @field_validator("output")

     

       224
       224
       +
           def validate_output(cls, v: dict[str, Any] | None) -> dict[str, Any] | None:

     

       225
       225
       +
               """

     

       226
       226
       +
               Validate output schema.

     

       227
       227
       +
               

     

       228
       228
       +
               Args:

     

       229
       229
       +
                   v: Output schema to validate

     

       230
       230
       +
                   

     

       231
       231
       +
               Returns:

     

       232
       232
       +
                   Validated output schema

     

       233
       233
       +
                   

     

       234
       234
       +
               Raises:

     

       235
       235
       +
                   ValueError: If output schema is invalid

     

       236
       236
       +
               """

     

       237
       237
       +
               if v and "encoding" not in v:

     

       238
       238
       +
                   raise ValueError("Output must specify encoding")

     

       239
       239
       +
               return v

     

       240
       240
       +
               

     

       241
       241
       +
           @field_validator("errors")

     

       242
       242
       +
           def validate_errors(cls, v: list[dict[str, str]] | None) -> list[dict[str, str]] | None:

     

       243
       243
       +
               """

     

       244
       244
       +
               Validate error definitions.

     

       245
       245
       +
               

     

       246
       246
       +
               Args:

     

       247
       247
       +
                   v: Error definitions to validate

     

       248
       248
       +
                   

     

       249
       249
       +
               Returns:

     

       250
       250
       +
                   Validated error definitions

     

       251
       251
       +
                   

     

       252
       252
       +
               Raises:

     

       253
       253
       +
                   ValueError: If any error definition is invalid

     

       254
       254
       +
               """

     

       255
       255
       +
               if v:

     

       256
       256
       +
                   for error in v:

     

       257
       257
       +
                       if "name" not in error:

     

       258
       258
       +
                           raise ValueError("Error must have a name")

     

       259
       259
       +
               return v

     

       260
       260
       +
       

     

       261
       261
       +
       class SubscriptionModel(DataModel):

     

       262
       262
       +
           """

     

       263
       263
       +
           Model for AT Protocol subscription type.

     

       264
       264
       +
           

     

       265
       265
       +
           Describes an Event Stream (WebSocket) with support for parameters,

     

       266
       266
       +
           message schemas and error responses.

     

       267
       267
       +
           """

     

       268
       268
       +
           

     

       269
       269
       +
           parameters: dict[str, Any] | None = None

     

       270
       270
       +
           """HTTP query parameters schema"""

     

       271
       271
       +
           

     

       272
       272
       +
           message: dict[str, Any] | None = None

     

       273
       273
       +
           """Specifies what messages can be"""

     

       274
       274
       +
           

     

       275
       275
       +
           errors: list[dict[str, str]] | None = None

     

       276
       276
       +
           """Possible error responses"""

     

       277
       277
       +
           

     

       278
       278
       +
           def __init__(self, **data: Any) -> None:

     

       279
       279
       +
               """

     

       280
       280
       +
               Initialize subscription model with validation.

     

       281
       281
       +
               

     

       282
       282
       +
               Args:

     

       283
       283
       +
                   **data: Input data containing subscription definition

     

       284
       284
       +
               """

     

       285
       285
       +
               super().__init__(**data)

     

       286
       286
       +
               

     

       287
       287
       +
           @field_validator("message")

     

       288
       288
       +
           def validate_message(cls, v: dict[str, Any] | None) -> dict[str, Any] | None:

     

       289
       289
       +
               """

     

       290
       290
       +
               Validate message schema.

     

       291
       291
       +
               

     

       292
       292
       +
               Args:

     

       293
       293
       +
                   v: Message schema to validate

     

       294
       294
       +
                   

     

       295
       295
       +
               Returns:

     

       296
       296
       +
                   Validated message schema

     

       297
       297
       +
                   

     

       298
       298
       +
               Raises:

     

       299
       299
       +
                   ValueError: If message schema is invalid

     

       300
       300
       +
               """

     

       301
       301
       +
               if v and "schema" not in v:

     

       302
       302
       +
                   raise ValueError("Message must specify schema")

     

       303
       303
       +
               return v

     

       304
       304
       +
               

     

       305
       305
       +
           @field_validator("errors")

     

       306
       306
       +
           def validate_errors(cls, v: list[dict[str, str]] | None) -> list[dict[str, str]] | None:

     

       307
       307
       +
               """

     

       308
       308
       +
               Validate error definitions.

     

       309
       309
       +
               

     

       310
       310
       +
               Args:

     

       311
       311
       +
                   v: Error definitions to validate

     

       312
       312
       +
                   

     

       313
       313
       +
               Returns:

     

       314
       314
       +
                   Validated error definitions

     

       315
       315
       +
                   

     

       316
       316
       +
               Raises:

     

       317
       317
       +
                   ValueError: If any error definition is invalid

     

       318
       318
       +
               """

     

       319
       319
       +
               if v:

     

       320
       320
       +
                   for error in v:

     

       321
       321
       +
                       if "name" not in error:

     

       322
       322
       +
                           raise ValueError("Error must have a name")

     

       323
       323
       +
               return v

+249

src/atpasser/model/types/string.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       import re

     

       3
       3
       +
       from datetime import datetime

     

       4
       4
       +
       from pydantic import field_validator

     

       5
       5
       +
       from ..base import DataModel

     

       6
       6
       +
       

     

       7
       7
       +
       class StringModel(DataModel):

     

       8
       8
       +
           """

     

       9
       9
       +
           Model for AT Protocol string type.

     

       10
       10
       +
           

     

       11
       11
       +
           Represents a Unicode string with support for format restrictions, length limits,

     

       12
       12
       +
           known values, enumeration sets, default values and constants as specified in Lexicon.

     

       13
       13
       +
           """

     

       14
       14
       +
           

     

       15
       15
       +
           value: str

     

       16
       16
       +
           """String value"""

     

       17
       17
       +
           

     

       18
       18
       +
           format: str | None = None

     

       19
       19
       +
           """String format restriction (e.g. 'datetime', 'uri')"""

     

       20
       20
       +
           

     

       21
       21
       +
           maxLength: int | None = None

     

       22
       22
       +
           """Maximum length in UTF-8 bytes"""

     

       23
       23
       +
           

     

       24
       24
       +
           minLength: int | None = None

     

       25
       25
       +
           """Minimum length in UTF-8 bytes"""

     

       26
       26
       +
           

     

       27
       27
       +
           knownValues: list[str] | None = None

     

       28
       28
       +
           """Suggested/common values (not enforced)"""

     

       29
       29
       +
           

     

       30
       30
       +
           enum: list[str] | None = None

     

       31
       31
       +
           """Closed set of allowed values"""

     

       32
       32
       +
           

     

       33
       33
       +
           default: str | None = None

     

       34
       34
       +
           """Default value if not provided"""

     

       35
       35
       +
           

     

       36
       36
       +
           const: str | None = None

     

       37
       37
       +
           """Fixed constant value if specified"""

     

       38
       38
       +
           

     

       39
       39
       +
           def __init__(self, **data: Any) -> None:

     

       40
       40
       +
               """

     

       41
       41
       +
               Initialize string model with validation.

     

       42
       42
       +
               

     

       43
       43
       +
               Args:

     

       44
       44
       +
                   **data: Input data containing string value

     

       45
       45
       +
                   

     

       46
       46
       +
               Raises:

     

       47
       47
       +
                   ValueError: If value violates constraints

     

       48
       48
       +
               """

     

       49
       49
       +
               super().__init__(**data)

     

       50
       50
       +
               if self.const is not None and self.value != self.const:

     

       51
       51
       +
                   raise ValueError(f"String value must be {self.const}")

     

       52
       52
       +
           

     

       53
       53
       +
           @field_validator("value", mode="before")

     

       54
       54
       +
           def validate_string(cls, v: Any) -> str:

     

       55
       55
       +
               """

     

       56
       56
       +
               Validate and convert input to string.

     

       57
       57
       +
               

     

       58
       58
       +
               Args:

     

       59
       59
       +
                   v: Value to validate

     

       60
       60
       +
                   

     

       61
       61
       +
               Returns:

     

       62
       62
       +
                   Validated string value

     

       63
       63
       +
                   

     

       64
       64
       +
               Raises:

     

       65
       65
       +
                   ValueError: If value violates constraints

     

       66
       66
       +
               """

     

       67
       67
       +
               if not isinstance(v, str):

     

       68
       68
       +
                   v = str(v)

     

       69
       69
       +
                   

     

       70
       70
       +
               # Validate length constraints

     

       71
       71
       +
               if cls.minLength is not None and len(v.encode()) < cls.minLength:

     

       72
       72
       +
                   raise ValueError(f"String must be at least {cls.minLength} bytes")

     

       73
       73
       +
                   

     

       74
       74
       +
               if cls.maxLength is not None and len(v.encode()) > cls.maxLength:

     

       75
       75
       +
                   raise ValueError(f"String must be at most {cls.maxLength} bytes")

     

       76
       76
       +
                   

     

       77
       77
       +
               # Validate enum

     

       78
       78
       +
               if cls.enum and v not in cls.enum:

     

       79
       79
       +
                   raise ValueError(f"Value must be one of {cls.enum}")

     

       80
       80
       +
                   

     

       81
       81
       +
               # Validate format if specified

     

       82
       82
       +
               if cls.format:

     

       83
       83
       +
                   if cls.format == "datetime":

     

       84
       84
       +
                       cls._validate_datetime(v)

     

       85
       85
       +
                   elif cls.format == "uri":

     

       86
       86
       +
                       cls._validate_uri(v)

     

       87
       87
       +
                   elif cls.format == "did":

     

       88
       88
       +
                       cls._validate_did(v)

     

       89
       89
       +
                   elif cls.format == "handle":

     

       90
       90
       +
                       cls._validate_handle(v)

     

       91
       91
       +
                   elif cls.format == "at-identifier":

     

       92
       92
       +
                       cls._validate_at_identifier(v)

     

       93
       93
       +
                   elif cls.format == "at-uri":

     

       94
       94
       +
                       cls._validate_at_uri(v)

     

       95
       95
       +
                   elif cls.format == "cid":

     

       96
       96
       +
                       cls._validate_cid(v)

     

       97
       97
       +
                   elif cls.format == "nsid":

     

       98
       98
       +
                       cls._validate_nsid(v)

     

       99
       99
       +
                   elif cls.format == "tid":

     

       100
       100
       +
                       cls._validate_tid(v)

     

       101
       101
       +
                   elif cls.format == "record-key":

     

       102
       102
       +
                       cls._validate_record_key(v)

     

       103
       103
       +
                   elif cls.format == "language":

     

       104
       104
       +
                       cls._validate_language(v)

     

       105
       105
       +
                   

     

       106
       106
       +
               return v

     

       107
       107
       +
           

     

       108
       108
       +
           @classmethod

     

       109
       109
       +
           def _validate_datetime(cls, v: str) -> None:

     

       110
       110
       +
               """Validate RFC 3339 datetime format"""

     

       111
       111
       +
               try:

     

       112
       112
       +
                   datetime.fromisoformat(v.replace("Z", "+00:00"))

     

       113
       113
       +
               except ValueError:

     

       114
       114
       +
                   raise ValueError("Invalid datetime format, must be RFC 3339")

     

       115
       115
       +
           

     

       116
       116
       +
           @classmethod

     

       117
       117
       +
           def _validate_uri(cls, v: str) -> None:

     

       118
       118
       +
               """Validate URI format"""

     

       119
       119
       +
               if len(v) > 8192:  # 8KB max

     

       120
       120
       +
                   raise ValueError("URI too long, max 8KB")

     

       121
       121
       +
               if not re.match(r"^[a-zA-Z][a-zA-Z0-9+.-]*:.+", v):

     

       122
       122
       +
                   raise ValueError("Invalid URI format")

     

       123
       123
       +
           

     

       124
       124
       +
           @classmethod

     

       125
       125
       +
           def _validate_did(cls, v: str) -> None:

     

       126
       126
       +
               """Validate DID format"""

     

       127
       127
       +
               if len(v) > 2048:

     

       128
       128
       +
                   raise ValueError("DID too long, max 2048 chars")

     

       129
       129
       +
               if not re.match(r"^did:[a-z]+:[a-zA-Z0-9._:%-]*[a-zA-Z0-9._-]$", v):

     

       130
       130
       +
                   raise ValueError("Invalid URI format")

     

       131
       131
       +
           

     

       132
       132
       +
           @classmethod

     

       133
       133
       +
           def _validate_handle(cls, v: str) -> None:

     

       134
       134
       +
               """Validate handle format"""

     

       135
       135
       +
               if not re.match(r"^([a-zA-Z0-9]([a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?\.)+[a-zA-Z]([a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?$", v):

     

       136
       136
       +
                   raise ValueError("Handle contains invalid characters")

     

       137
       137
       +
               if len(v) > 253:

     

       138
       138
       +
                   raise ValueError("Handle too long, max 253 chars")

     

       139
       139
       +
       

     

       140
       140
       +
           @classmethod

     

       141
       141
       +
           def _validate_at_identifier(cls, v: str) -> None:

     

       142
       142
       +
               """Validate at-identifier format (DID or handle)"""

     

       143
       143
       +
               try:

     

       144
       144
       +
                   if v.startswith("did:"):

     

       145
       145
       +
                       cls._validate_did(v)

     

       146
       146
       +
                   else:

     

       147
       147
       +
                       cls._validate_handle(v)

     

       148
       148
       +
               except ValueError as e:

     

       149
       149
       +
                   raise ValueError(f"Invalid at-identifier: {e}")

     

       150
       150
       +
       

     

       151
       151
       +
           @classmethod

     

       152
       152
       +
           def _validate_at_uri(cls, v: str) -> None:

     

       153
       153
       +
               """

     

       154
       154
       +
               Validate AT-URI format according to AT Protocol specification.

     

       155
       155
       +
               

     

       156
       156
       +
               Args:

     

       157
       157
       +
                   v: AT-URI string to validate

     

       158
       158
       +
                   

     

       159
       159
       +
               Raises:

     

       160
       160
       +
                   ValueError: If URI violates any of these rules:

     

       161
       161
       +
                       - Must start with 'at://'

     

       162
       162
       +
                       - Max length 8KB

     

       163
       163
       +
                       - No trailing slash

     

       164
       164
       +
                       - Authority must be valid DID or handle

     

       165
       165
       +
                       - Path segments must follow NSID/RKEY rules if present

     

       166
       166
       +
               """

     

       167
       167
       +
               if not v.startswith("at://"):

     

       168
       168
       +
                   raise ValueError("AT-URI must start with 'at://'")

     

       169
       169
       +
               if len(v) > 8192:  # 8KB

     

       170
       170
       +
                   raise ValueError("AT-URI too long, max 8KB")

     

       171
       171
       +
               if v.endswith('/'):

     

       172
       172
       +
                   raise ValueError("AT-URI cannot have trailing slash")

     

       173
       173
       +
                   

     

       174
       174
       +
               # Split into parts

     

       175
       175
       +
               parts = v[5:].split('/')  # Skip 'at://'

     

       176
       176
       +
               authority = parts[0]

     

       177
       177
       +
               

     

       178
       178
       +
               # Validate authority (DID or handle)

     

       179
       179
       +
               if not authority:

     

       180
       180
       +
                   raise ValueError("AT-URI must have authority")

     

       181
       181
       +
                   

     

       182
       182
       +
               if authority.startswith('did:'):

     

       183
       183
       +
                   # Basic DID format check - actual DID validation is done elsewhere

     

       184
       184
       +
                   if len(authority) > 2048:

     

       185
       185
       +
                       raise ValueError("DID too long")

     

       186
       186
       +
                   if ':' not in authority[4:]:

     

       187
       187
       +
                       raise ValueError("Invalid DID format")

     

       188
       188
       +
               else:

     

       189
       189
       +
                   # Handle validation

     

       190
       190
       +
                   if not re.match(r'^[a-z0-9.-]+$', authority):

     

       191
       191
       +
                       raise ValueError("Invalid handle characters")

     

       192
       192
       +
                   if len(authority) > 253:

     

       193
       193
       +
                       raise ValueError("Handle too long")

     

       194
       194
       +
                       

     

       195
       195
       +
               # Validate path segments if present

     

       196
       196
       +
               if len(parts) > 1:

     

       197
       197
       +
                   if len(parts) > 3:

     

       198
       198
       +
                       raise ValueError("AT-URI path too deep")

     

       199
       199
       +
                       

     

       200
       200
       +
                   collection = parts[1]

     

       201
       201
       +
                   if not re.match(r'^[a-zA-Z0-9.-]+$', collection):

     

       202
       202
       +
                       raise ValueError("Invalid collection NSID")

     

       203
       203
       +
                       

     

       204
       204
       +
                   if len(parts) > 2:

     

       205
       205
       +
                       rkey = parts[2]

     

       206
       206
       +
                       if not rkey:

     

       207
       207
       +
                           raise ValueError("Record key cannot be empty")

     

       208
       208
       +
                       if not re.match(r'^[a-zA-Z0-9._:%-~]+$', rkey):

     

       209
       209
       +
                           raise ValueError("Invalid record key characters")

     

       210
       210
       +
       

     

       211
       211
       +
           @classmethod

     

       212
       212
       +
           def _validate_cid(cls, v: str) -> None:

     

       213
       213
       +
               """Validate CID string format"""

     

       214
       214
       +
               if len(v) > 100:

     

       215
       215
       +
                   raise ValueError("CID too long, max 100 chars")

     

       216
       216
       +
               if not re.match(r"^[a-zA-Z0-9]+$", v):

     

       217
       217
       +
                   raise ValueError("CID contains invalid characters")

     

       218
       218
       +
       

     

       219
       219
       +
           @classmethod

     

       220
       220
       +
           def _validate_nsid(cls, v: str) -> None:

     

       221
       221
       +
               """Validate NSID format"""

     

       222
       222
       +
               if len(v) > 317:

     

       223
       223
       +
                   raise ValueError("NSID too long, max 317 chars")

     

       224
       224
       +
               if not re.match(r"^[a-zA-Z]([a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?(\.[a-zA-Z0-9]([a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?)+(\.[a-zA-Z]([a-zA-Z0-9]{0,62})?)$", v):

     

       225
       225
       +
                   raise ValueError("NSID contains invalid characters")

     

       226
       226
       +
       

     

       227
       227
       +
           @classmethod

     

       228
       228
       +
           def _validate_tid(cls, v: str) -> None:

     

       229
       229
       +
               """Validate TID format"""

     

       230
       230
       +
               if len(v) > 13:

     

       231
       231
       +
                   raise ValueError("TID too long, max 13 chars")

     

       232
       232
       +
               if not re.match(r"^[234567abcdefghij][234567abcdefghijklmnopqrstuvwxyz]{12}$", v):

     

       233
       233
       +
                   raise ValueError("TID contains invalid characters")

     

       234
       234
       +
       

     

       235
       235
       +
           @classmethod

     

       236
       236
       +
           def _validate_record_key(cls, v: str) -> None:

     

       237
       237
       +
               """Validate record-key format"""

     

       238
       238
       +
               if len(v) > 512:

     

       239
       239
       +
                   raise ValueError("Record key too long, max 512 chars")

     

       240
       240
       +
               if v == "." or v == "..":

     

       241
       241
       +
                   raise ValueError(f"Record key is {v}, which is not allowed")

     

       242
       242
       +
               if not re.match(r"^[a-zA-Z0-9._:%-~]+$", v):

     

       243
       243
       +
                   raise ValueError("Record key contains invalid characters")

     

       244
       244
       +
       

     

       245
       245
       +
           @classmethod

     

       246
       246
       +
           def _validate_language(cls, v: str) -> None:

     

       247
       247
       +
               """Validate BCP 47 language tag"""

     

       248
       248
       +
               if not re.match(r"^[a-zA-Z]{1,8}(-[a-zA-Z0-9]{1,8})*$", v):

     

       249
       249
       +
                   raise ValueError("Invalid language tag format")

Compare changes