benedict.serializers.ini

Complexity

Total Complexity

16

Size/Duplication

Total Lines	71
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	INISerializer.encode()	0	8	2
A	INISerializer._try_parse()	0	12	4
A	INISerializer.decode()	0	11	3
B	INISerializer._preprocessed_dict()	0	20	6
A	INISerializer.__init__()	0	2	1

# -*- coding: utf-8 -*-

from __future__ import absolute_import

import io
import configparser

from benedict.core import clone
from benedict.serializers.abstract import AbstractSerializer


class INISerializer(AbstractSerializer):

    def __init__(self):
        super(INISerializer, self).__init__()

    @staticmethod
    def _try_parse(parser: configparser.ConfigParser, section, key):
        try:
            return parser.getint(section, key)
        except ValueError:
            try:
                return parser.getfloat(section, key)
            except ValueError:
                try:
                    return parser.getboolean(section, key)
                except ValueError:
                    return parser.get(section, key)

    def decode(self, s, **kwargs):
        parser = configparser.ConfigParser()
        kwargs.setdefault("source", "<string>")
        parser.read_string(s, **kwargs)
        data = {}
        for s in parser.sections():
            data[s] = {}
            for key, _ in parser.items(s):
                data[s][key] = INISerializer._try_parse(parser, s, key)

        return data

    def encode(self, d, **kwargs):
        parser = configparser.ConfigParser()
        processed_dict = self._preprocessed_dict(d, **kwargs)
        parser.read_dict(processed_dict)
        with io.StringIO() as data:
            parser.write(data, **kwargs)
            data.seek(0)
            return data.getvalue()

    def _preprocessed_dict(self, d, default_namespace="default"):
        sections = {
            default_namespace: {}
        }
        for key, value in d.items():
            if isinstance(value, dict):
                sections[key] = clone(value)
            else:
                sections[default_namespace][key] = value

        # Stringify all the value
        for section in sections.values():
            for key, val in section.items():
                section[key] = str(val)

        # if the default section is empty, delete it
        if len(sections[default_namespace]) == 0:
            del sections[default_namespace]

        return sections