datclass

python dataclass nested & extra attrs

Extending the official dataclass to support nested and extended fields.

Install

pip install -U datclass
pip install git+ssh://git@github.com/foyoux/datclass.git
pip install git+https://github.com/foyoux/datclass.git

Usage example

Example 1

from dataclasses import dataclass
from typing import List

# Default `dataclass` supports nested and extended fields. Missing fields will be logged.
from datclass import DatClass


# Custom `DatClass`
# from datclass import get_datclass
# Missing fields will not be logged.
# DatClass = get_datclass(log=False)


@dataclass
class User(DatClass):
    name: str
    age: int


@dataclass
class Group(DatClass):
    name: str
    users: List[User]


if __name__ == '__main__':
    user1 = User(name='foo', age=18)
    # Saving a data class to a file.
    user1.to_file('user.json')
    # Adding some control parameters.
    user1.to_file('user.json', indent=4, ignore_none=True, sort_keys=True)

    # Creating a data class from a dict.
    user2 = User(**{'name': 'bar', 'age': 20})
    # To convert a data class to a dictionary, you can support extended fields.
    # You can also use the official asdict function, but it cannot export extended fields.
    dict1 = user2.to_dict()
    # 'ignore_none' is used to ignore values that are None.
    dict2 = user2.to_dict(ignore_none=True)

    # Creating a data class from a string.
    user3 = User.from_str('{"name": "baz", "age": 22}')
    # Convert the data class to a JSON string.
    dict3 = user3.to_str()
    dict4 = user3.to_str(indent=4, ignore_none=True)

    # Creating a data class from a file.
    user4 = User.from_file('user.json')

    # Nested data classes
    grp = Group(name='group1', users=[user1, user2, user3, user4])
    grp.to_file('group.json', indent=4, ignore_none=True, sort_keys=True)

    for user in grp.users:
        print(user.name, user.age)

    # Extending fields
    user = {'name': 'foo', 'age': 18, 'sex': 1}
    user5 = User(**user)
    assert user5.to_dict() == user

Example 2

from dataclasses import dataclass
from typing import ClassVar, Dict

from datclass import DatClass


@dataclass
class User(DatClass):
    name: str
    age: int
    attr_123: str

    # Renaming fields.
    __rename_attrs__: ClassVar[Dict[str, str]] = {
        'Name': 'name',
        'Age': 'age',
        '123#$^%^%*': 'attr_123'
    }

    # This way of writing is also acceptable.
    # __rename_attrs__ = {
    #     'Name': 'name',
    #     'Age': 'age',
    #     '123#$^%^%*': 'attr_123',
    # }


if __name__ == '__main__':
    s = '{"Name": "foo", "Age": 18, "123#$^%^%*": "test rename attrs"}'
    user = User.from_str(s)
    assert user.to_str() == s

Automatically generate `DataClass`

See the datclass command for details.

$ datclass -h

Example 1

Input user.json

{
  "Name": "foo",
  "Age": 18,
  "123#$^%^%*": "test rename attrs"
}

Execute the command

$ datclass -o user.py user.json

Output user.py

from dataclasses import dataclass

from datclass import DatClass


@dataclass
class Object(DatClass):
    a_123: str = None  # rename from '123#$^%^%*'
    age: int = None  # rename from 'Age'
    name: str = None  # rename from 'Name'

    __rename_attrs__ = {
        'Name': 'name',
        'Age': 'age',
        '123#$^%^%*': 'a_123',
    }

欢迎反馈和建议

Feedback and suggestions welcome

https://github.com/foyoux/datclass/issues/new

Metadata

Release files for datclass 0.2.28

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for datclass 0.2.28
File	Size	Uploaded
datclass-0.2.28.tar.gz	13.0 kB	Oct 8, 2024	Details

Built distribution (wheel)

Table of built distributions (wheels) for datclass 0.2.28
File	Interpreter	ABI	Platform	Reset
datclass-0.2.28-py3-none-any.whl 12.6 kB Oct 8, 2024	Python 3	none	any	Details

Total release size: 25.6 kB

Release files / datclass-0.2.28.tar.gz

Download URL	datclass-0.2.28.tar.gz
Size	13.0 kB
Tags	Source
SHA-256 checksum How to use checksums	`1c5643f8caea3439127b1e53c686516e75e814485461e7174ff1f698483082ad`
BLAKE2b-256 checksum How to use checksums	`43b652392616582cf1497ef2589f674911122d75de8f2ef4a04e52583d2e113c`
Upload date	Oct 8, 2024
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/5.1.1 CPython/3.12.7`

Release files / datclass-0.2.28-py3-none-any.whl

Download URL	datclass-0.2.28-py3-none-any.whl
Size	12.6 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`3ff2aa3d20e599a98664a496d978ffa105d3c0765cf02ad13d0e5e2c266ae538`
BLAKE2b-256 checksum How to use checksums	`4332feac5c9ceda0fa544bfe9341498d73e4d664a91b78c3e27a855f26aa6687`
Upload date	Oct 8, 2024
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/5.1.1 CPython/3.12.7`