Skip to main content

This project helps you to import config file writen by YAML to Python data class.

Project description

YAML Data Class Config

Test CodeQL Code Coverage Maintainability Dependabot Python versions PyPI - Downloads X URL

This project helps you to import config file writen by YAML to Python Data Classes.

📖 Migration Guide

Upgrading from version 1.x? See the Migration Guide for step-by-step instructions to migrate to version 2.x.

Advantage

  1. Type safe import from YAML to Data Classes
  2. Global access and easy unit testing

1. Type safe import from YAML to Data classes

When using pyyaml to import YAML, values be dict and list objects. Using dict or list object will cause such confuses:

  • Reference non exist properties for unexpected instance type
  • Typo of index or key name

To prevent these confuse, one of good way is to use object as model, and python has a good module Data Classes for this purpose.

2. Global access and easy unit testing

You will want to refer config as global because it's troublesome to pass config value as argument over and over like a bucket brigade.

However, when unit testing, if YAML file was loaded automatically on importing global definition, you will face problem that you can't replace config YAML file with the one for unit testing. YAML Data Class Config can divide timings between definition global instance and loading YAML file so you can replace YAML file for unit testing.

Modern Python Features

This library leverages modern Python features for better type safety and developer experience:

  • Modern Union Syntax: Uses str | None instead of Union[str, None] (Python 3.10+)
  • Future Annotations: Uses from __future__ import annotations for improved type hint evaluation
  • Keyword-Only Arguments: Functions use * to enforce keyword-only arguments for better API clarity
  • Type Safety: Full mypy compliance with strict mode enabled
  • Runtime Type Checking: Proper handling of type annotations at runtime for dataclass serialization

Quickstart

1. Install

pip install yamldataclassconfig

2. Prepare config YAML file

Put config.yml YAML Data class Config loads config.yml on Python execution directory by default.

property_a: 1
property_b: '2'
part_config:
  property_c: '2019-06-25 13:33:30'

3. Create config class

Anywhere is OK, for example, I prefer to place on myproduct/config.py

from dataclasses import dataclass
from dataclasses import field
from datetime import datetime

from dataclasses_json import DataClassJsonMixin
from marshmallow import fields

from yamldataclassconfig.config import YamlDataClassConfig


@dataclass
class PartConfig(DataClassJsonMixin):
    property_c: datetime = field(
        metadata={
            "dataclasses_json": {
                "encoder": datetime.isoformat,
                "decoder": datetime.fromisoformat,
                "mm_field": fields.DateTime(format="iso"),
            },
        },
    )


@dataclass
class Config(YamlDataClassConfig):
    property_a: int
    property_b: str
    part_config: PartConfig = field(
        metadata={"dataclasses_json": {"mm_field": PartConfig}},
    )

4. Define as global

Also, anywhere is OK, for example, I prefer to place on myproduct/__init__.py

from myproduct.config import Config

CONFIG: Config = Config.create()

5. Call load before reference config value

from myproduct import CONFIG


def main() -> None:
    CONFIG.load()
    print(CONFIG.property_a)
    print(CONFIG.property_b)
    if CONFIG.part_config is not None:
        print(CONFIG.part_config.property_c)


if __name__ == '__main__':
    main()

How do I...

Fix path to yaml file independent on the Python execution directory?

override FILE_PATH property.

Ex:

from dataclasses import dataclass
from dataclasses import field
from pathlib import Path

from yamldataclassconfig import build_path
from yamldataclassconfig.config import YamlDataClassConfig


@dataclass
class Config(YamlDataClassConfig):
    some_property: str
    # ...

    FILE_PATH: str = field(
        init=False,
        default=build_path(Path(__file__).parent / "config.yml"),
    )

Switch target YAML config file to the one for unit testing?

When setup on unit testing, you can call Config.load() with argument.

Note: The path_is_absolute parameter must now be passed as a keyword argument due to API improvements in version 1.5.0.

Case when unittest:

from pathlib import Path
import unittest

from yourproduct import CONFIG

class ConfigurableTestCase(unittest.TestCase):
    def setUp(self) -> None:
        CONFIG.load(Path('path/to/yaml'))

Case when pytest:

from pathlib import Path
from typing import Generator

import pytest

from yourproduct import CONFIG

@pytest.fixture
def yaml_config() -> Generator[None, None, None]:
    CONFIG.load(Path('path/to/yaml'))
    yield

def test_something(yaml_config: None) -> None:
    """test something"""

Use path to YAML config file as same as production when test?

fixturefilehandler can replace config.yml with tests/config.yml.dist easily. Please call all DeployerFactory.create with YamlConfigFilePathBuilder instance argument to create ConfigDeployer. Then, set target directory which config.yml should be placed into path_target_directory.

Case when unittest:

from pathlib import Path
import unittest

from fixturefilehandler.factories import DeployerFactory
from fixturefilehandler.file_paths import YamlConfigFilePathBuilder

from yourproduct import CONFIG


ConfigDeployer = DeployerFactory.create(YamlConfigFilePathBuilder(path_target_directory=Path(__file__).parent.parent))


class ConfigurableTestCase(unittest.TestCase):
    def setUp(self) -> None:
        ConfigDeployer.setup()
        CONFIG.load()

    def doCleanups(self) -> None:
        ConfigDeployer.teardown()

Case when pytest:

from pathlib import Path
from typing import Generator

import pytest
from fixturefilehandler.factories import DeployerFactory
from fixturefilehandler.file_paths import YamlConfigFilePathBuilder

from yourproduct import CONFIG


ConfigDeployer = DeployerFactory.create(YamlConfigFilePathBuilder(path_target_directory=Path(__file__).parent.parent))


@pytest.fixture
def yaml_config() -> Generator[None, None, None]:
    ConfigDeployer.setup()
    CONFIG.load()
    yield
    ConfigDeployer.teardown()


def test_something(yaml_config: None) -> None:
    """test something"""

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

yamldataclassconfig-2.0.4.tar.gz (29.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

yamldataclassconfig-2.0.4-py3-none-any.whl (16.3 kB view details)

Uploaded Python 3

File details

Details for the file yamldataclassconfig-2.0.4.tar.gz.

File metadata

  • Download URL: yamldataclassconfig-2.0.4.tar.gz
  • Upload date:
  • Size: 29.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for yamldataclassconfig-2.0.4.tar.gz
Algorithm Hash digest
SHA256 29527434a4396391db4c0c69bc998b37fa93422ccbab36e69a9e73049981b019
MD5 29f038c62f5368e76e6269e10d436043
BLAKE2b-256 acd922da9d3af8f6385cb1309395938efe92e029467ffad11d76fe5770de3578

See more details on using hashes here.

File details

Details for the file yamldataclassconfig-2.0.4-py3-none-any.whl.

File metadata

File hashes

Hashes for yamldataclassconfig-2.0.4-py3-none-any.whl
Algorithm Hash digest
SHA256 4a1ee34d8f091ca254ca5ab39df40c66b5b393ecce7e4c8b0514207d567a5c63
MD5 96be6b35b169e00a034ba1999a242ef9
BLAKE2b-256 487f4b916729ec99614b91c28f98a1780b6c0fda47f585b0d77c94afa4c9789b

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page