Source code for fastpyxl.descriptors

# Copyright (c) 2010-2024 fastpyxl

from .base import (
    Alias,
    ASCII,
    Bool,
    Convertible,
    DateTime,
    Default,
    Descriptor,
    Float,
    Integer,
    Length,
    MatchPattern,
    Max,
    Min,
    MinMax,
    NoneSet,
    Set,
    String,
    Text,
    Tuple,
    Typed,
)
from .sequence import Sequence




class MetaStrict(type):
    def __new__(cls, clsname, bases, methods):
        for k, v in methods.items():
            if isinstance(v, Descriptor):
                v.name = k
        return type.__new__(cls, clsname, bases, methods)





class Strict(metaclass=MetaStrict):
    pass





class MetaSerialisable(type):
    def __new__(cls, clsname, bases, methods):
        attrs = []
        nested = []
        elements = []
        namespaced = []
        for k, v in methods.items():
            if isinstance(v, Descriptor):
                ns = v.namespace
                if ns:
                    namespaced.append((k, "{%s}%s" % (ns, k)))
                if v.nested:
                    nested.append(k)
                    elements.append(k)
                elif isinstance(v, Sequence):
                    elements.append(k)
                elif isinstance(v, Typed):
                    if hasattr(v.expected_type, "to_tree"):
                        elements.append(k)
                    elif isinstance(v.expected_type, tuple):
                        if any((hasattr(el, "to_tree") for el in v.expected_type)):
                            continue
                    else:
                        attrs.append(k)
                else:
                    if not isinstance(v, Alias):
                        attrs.append(k)

        if methods.get("__attrs__") is None:
            methods["__attrs__"] = tuple(attrs)
        methods["__namespaced__"] = tuple(namespaced)
        if methods.get("__nested__") is None:
            methods["__nested__"] = tuple(sorted(nested))
        if methods.get("__elements__") is None:
            methods["__elements__"] = tuple(sorted(elements))
        return MetaStrict.__new__(cls, clsname, bases, methods)